dendi-wang/thingsboard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Performance Improvement for Device State Service

Browse Source
This commit is contained in:
Andrew Shvayka 2019-12-20 15:58:55 +02:00
parent cacdf09eef
commit 95752ff3b9
1 changed files with 23 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
application/src/main/java/org/thingsboard/server/service/state/DefaultDeviceStateService.java
View File

@ -141,6 +141,8 @@ public class DefaultDeviceStateService implements DeviceStateService {
private ListeningScheduledExecutorService queueExecutor; private ListeningScheduledExecutorService queueExecutor;
private ConcurrentMap<TenantId, Set<DeviceId>> tenantDevices = new ConcurrentHashMap<>(); private ConcurrentMap<TenantId, Set<DeviceId>> tenantDevices = new ConcurrentHashMap<>();
private ConcurrentMap<DeviceId, DeviceStateData> deviceStates = new ConcurrentHashMap<>(); private ConcurrentMap<DeviceId, DeviceStateData> deviceStates = new ConcurrentHashMap<>();
private ConcurrentMap<DeviceId, Long> deviceLastReportedActivity = new ConcurrentHashMap<>();
private ConcurrentMap<DeviceId, Long> deviceLastSavedActivity = new ConcurrentHashMap<>();
@PostConstruct @PostConstruct
public void init() { public void init() {
@ -175,6 +177,7 @@ public class DefaultDeviceStateService implements DeviceStateService {
@Override @Override
public void onDeviceActivity(DeviceId deviceId) { public void onDeviceActivity(DeviceId deviceId) {
deviceLastReportedActivity.put(deviceId, System.currentTimeMillis());
queueExecutor.submit(() -> onDeviceActivitySync(deviceId)); queueExecutor.submit(() -> onDeviceActivitySync(deviceId));
} }
@ -245,6 +248,8 @@ public class DefaultDeviceStateService implements DeviceStateService {
tenantDeviceSet.remove(device.getId()); tenantDeviceSet.remove(device.getId());
} }
deviceStates.remove(device.getId()); deviceStates.remove(device.getId());
deviceLastReportedActivity.remove(device.getId());
deviceLastSavedActivity.remove(device.getId());
} }
} }
try { try {
@ -305,6 +310,8 @@ public class DefaultDeviceStateService implements DeviceStateService {
} else { } else {
log.debug("[{}] Device that belongs to other server is detected and removed.", deviceId); log.debug("[{}] Device that belongs to other server is detected and removed.", deviceId);
deviceStates.remove(deviceId); deviceStates.remove(deviceId);
deviceLastReportedActivity.remove(deviceId);
deviceLastSavedActivity.remove(deviceId);
} }
} }
} }
@ -330,17 +337,20 @@ public class DefaultDeviceStateService implements DeviceStateService {
} }
private void onDeviceActivitySync(DeviceId deviceId) { private void onDeviceActivitySync(DeviceId deviceId) {
DeviceStateData stateData = getOrFetchDeviceStateData(deviceId); long lastReportedActivity = deviceLastReportedActivity.getOrDefault(deviceId, 0L);
if (stateData != null) { long lastSavedActivity = deviceLastSavedActivity.getOrDefault(deviceId, 0L);
DeviceState state = stateData.getState(); if (lastReportedActivity > 0 && lastReportedActivity > lastSavedActivity) {
long ts = System.currentTimeMillis(); DeviceStateData stateData = getOrFetchDeviceStateData(deviceId);
stateData.getState().setLastActivityTime(ts); if (stateData != null) {
pushRuleEngineMessage(stateData, ACTIVITY_EVENT); DeviceState state = stateData.getState();
save(deviceId, LAST_ACTIVITY_TIME, ts); stateData.getState().setLastActivityTime(lastReportedActivity);
pushRuleEngineMessage(stateData, ACTIVITY_EVENT);
if (!state.isActive()) { save(deviceId, LAST_ACTIVITY_TIME, lastReportedActivity);
state.setActive(true); deviceLastSavedActivity.put(deviceId, lastReportedActivity);
save(deviceId, ACTIVITY_STATE, state.isActive()); if (!state.isActive()) {
state.setActive(true);
save(deviceId, ACTIVITY_STATE, state.isActive());
}
} }
} }
} }
@ -431,6 +441,8 @@ public class DefaultDeviceStateService implements DeviceStateService {
Optional<ServerAddress> address = routingService.resolveById(deviceId); Optional<ServerAddress> address = routingService.resolveById(deviceId);
if (!address.isPresent()) { if (!address.isPresent()) {
deviceStates.remove(deviceId); deviceStates.remove(deviceId);
deviceLastReportedActivity.remove(deviceId);
deviceLastSavedActivity.remove(deviceId);
Set<DeviceId> deviceIds = tenantDevices.get(tenantId); Set<DeviceId> deviceIds = tenantDevices.get(tenantId);
if (deviceIds != null) { if (deviceIds != null) {
deviceIds.remove(deviceId); deviceIds.remove(deviceId);