aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/traccar/database
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/org/traccar/database')
-rw-r--r--src/main/java/org/traccar/database/DeviceLookupService.java47
1 files changed, 30 insertions, 17 deletions
diff --git a/src/main/java/org/traccar/database/DeviceLookupService.java b/src/main/java/org/traccar/database/DeviceLookupService.java
index 28910c24a..583b2ae35 100644
--- a/src/main/java/org/traccar/database/DeviceLookupService.java
+++ b/src/main/java/org/traccar/database/DeviceLookupService.java
@@ -20,6 +20,8 @@ import io.netty.util.Timer;
import io.netty.util.TimerTask;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
+import org.traccar.config.Config;
+import org.traccar.config.Keys;
import org.traccar.model.Device;
import org.traccar.storage.Storage;
import org.traccar.storage.StorageException;
@@ -45,6 +47,8 @@ public class DeviceLookupService {
private final Storage storage;
private final Timer timer;
+ private final boolean throttlingEnabled;
+
private static class IdentifierInfo {
private long lastQuery;
private long delay;
@@ -70,36 +74,45 @@ public class DeviceLookupService {
private final Map<String, IdentifierInfo> identifierMap = new ConcurrentHashMap<>();
@Inject
- public DeviceLookupService(Storage storage, Timer timer) {
+ public DeviceLookupService(Config config, Storage storage, Timer timer) {
this.storage = storage;
this.timer = timer;
+ throttlingEnabled = config.getBoolean(Keys.DATABASE_THROTTLE_UNKNOWN);
}
private synchronized boolean isThrottled(String uniqueId) {
- IdentifierInfo info = identifierMap.get(uniqueId);
- return info != null && System.currentTimeMillis() < info.lastQuery + info.delay;
+ if (throttlingEnabled) {
+ IdentifierInfo info = identifierMap.get(uniqueId);
+ return info != null && System.currentTimeMillis() < info.lastQuery + info.delay;
+ } else {
+ return false;
+ }
}
private synchronized void lookupSucceeded(String uniqueId) {
- IdentifierInfo info = identifierMap.remove(uniqueId);
- if (info != null) {
- info.timeout.cancel();
+ if (throttlingEnabled) {
+ IdentifierInfo info = identifierMap.remove(uniqueId);
+ if (info != null) {
+ info.timeout.cancel();
+ }
}
}
private synchronized void lookupFailed(String uniqueId) {
- IdentifierInfo info = identifierMap.get(uniqueId);
- if (info != null) {
- info.timeout.cancel();
- info.delay = Math.min(info.delay * 2, THROTTLE_MAX_MS);
- } else {
- info = new IdentifierInfo();
- identifierMap.put(uniqueId, info);
- info.delay = THROTTLE_MIN_MS;
+ if (throttlingEnabled) {
+ IdentifierInfo info = identifierMap.get(uniqueId);
+ if (info != null) {
+ info.timeout.cancel();
+ info.delay = Math.min(info.delay * 2, THROTTLE_MAX_MS);
+ } else {
+ info = new IdentifierInfo();
+ identifierMap.put(uniqueId, info);
+ info.delay = THROTTLE_MIN_MS;
+ }
+ info.lastQuery = System.currentTimeMillis();
+ info.timeout = timer.newTimeout(new IdentifierTask(uniqueId), INFO_TIMEOUT_MS, TimeUnit.MILLISECONDS);
+ LOGGER.debug("Device lookup {} throttled for {} ms", uniqueId, info.delay);
}
- info.lastQuery = System.currentTimeMillis();
- info.timeout = timer.newTimeout(new IdentifierTask(uniqueId), INFO_TIMEOUT_MS, TimeUnit.MILLISECONDS);
- LOGGER.debug("Device lookup {} throttled for {} ms", uniqueId, info.delay);
}
public Device lookup(String[] uniqueIds) {