Add timeout and update limit for long-polling updates to DefaultBotOptions

Just a little commit to make the long-polling timeout and long-polling limit configurable via the BotOptions. Mainly for Testing
This commit is contained in:
Chase22 2020-09-21 19:31:50 +02:00
parent e6b9fd892f
commit 2009ef411d
2 changed files with 22 additions and 2 deletions

View File

@ -26,6 +26,8 @@ public class DefaultBotOptions implements BotOptions {
private ProxyType proxyType; private ProxyType proxyType;
private String proxyHost; private String proxyHost;
private int proxyPort; private int proxyPort;
private int getUpdatesTimeout;
private int getUpdatesLimit;
public enum ProxyType { public enum ProxyType {
NO_PROXY, NO_PROXY,
@ -39,6 +41,8 @@ public class DefaultBotOptions implements BotOptions {
baseUrl = ApiConstants.BASE_URL; baseUrl = ApiConstants.BASE_URL;
httpContext = HttpClientContext.create(); httpContext = HttpClientContext.create();
proxyType = ProxyType.NO_PROXY; proxyType = ProxyType.NO_PROXY;
getUpdatesLimit = ApiConstants.GETUPDATES_TIMEOUT;
getUpdatesLimit = 100;
} }
@Override @Override
@ -129,4 +133,20 @@ public class DefaultBotOptions implements BotOptions {
public void setProxyPort(int proxyPort) { public void setProxyPort(int proxyPort) {
this.proxyPort = proxyPort; this.proxyPort = proxyPort;
} }
public int getGetUpdatesTimeout() {
return getUpdatesTimeout;
}
public void setGetUpdatesTimeout(int getUpdatesTimeout) {
this.getUpdatesTimeout = getUpdatesTimeout;
}
public int getGetUpdatesLimit() {
return getUpdatesLimit;
}
public void setGetUpdatesLimit(int getUpdatesLimit) {
this.getUpdatesLimit = getUpdatesLimit;
}
} }

View File

@ -236,8 +236,8 @@ public class DefaultBotSession implements BotSession {
private List<Update> getUpdatesFromServer() throws IOException { private List<Update> getUpdatesFromServer() throws IOException {
GetUpdates request = new GetUpdates() GetUpdates request = new GetUpdates()
.setLimit(100) .setLimit(options.getGetUpdatesLimit())
.setTimeout(ApiConstants.GETUPDATES_TIMEOUT) .setTimeout(options.getGetUpdatesTimeout())
.setOffset(lastReceivedUpdate + 1); .setOffset(lastReceivedUpdate + 1);
if (options.getAllowedUpdates() != null) { if (options.getAllowedUpdates() != null) {