aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/traccar/database/StatisticsManager.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/org/traccar/database/StatisticsManager.java')
-rw-r--r--src/main/java/org/traccar/database/StatisticsManager.java79
1 files changed, 51 insertions, 28 deletions
diff --git a/src/main/java/org/traccar/database/StatisticsManager.java b/src/main/java/org/traccar/database/StatisticsManager.java
index e59f8e767..21b59540e 100644
--- a/src/main/java/org/traccar/database/StatisticsManager.java
+++ b/src/main/java/org/traccar/database/StatisticsManager.java
@@ -1,5 +1,5 @@
/*
- * Copyright 2016 - 2019 Anton Tananaev (anton@traccar.org)
+ * Copyright 2016 - 2020 Anton Tananaev (anton@traccar.org)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -15,6 +15,8 @@
*/
package org.traccar.database;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.traccar.config.Config;
@@ -29,7 +31,9 @@ import javax.ws.rs.core.Form;
import java.sql.SQLException;
import java.util.Calendar;
import java.util.Date;
+import java.util.HashMap;
import java.util.HashSet;
+import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicInteger;
@@ -37,16 +41,17 @@ public class StatisticsManager {
private static final Logger LOGGER = LoggerFactory.getLogger(StatisticsManager.class);
- private static final int SPLIT_MODE = Calendar.DAY_OF_MONTH;
+ private static final int SPLIT_MODE = Calendar.MINUTE;
private final Config config;
private final DataManager dataManager;
private final Client client;
+ private final ObjectMapper objectMapper;
- private AtomicInteger lastUpdate = new AtomicInteger(Calendar.getInstance().get(SPLIT_MODE));
+ private final AtomicInteger lastUpdate = new AtomicInteger(Calendar.getInstance().get(SPLIT_MODE));
- private Set<Long> users = new HashSet<>();
- private Set<Long> devices = new HashSet<>();
+ private final Set<Long> users = new HashSet<>();
+ private final Map<Long, String> deviceProtocols = new HashMap<>();
private int requests;
private int messagesReceived;
@@ -57,26 +62,47 @@ public class StatisticsManager {
private int geolocationRequests;
@Inject
- public StatisticsManager(Config config, DataManager dataManager, Client client) {
+ public StatisticsManager(Config config, DataManager dataManager, Client client, ObjectMapper objectMapper) {
this.config = config;
this.dataManager = dataManager;
this.client = client;
+ this.objectMapper = objectMapper;
}
private void checkSplit() {
int currentUpdate = Calendar.getInstance().get(SPLIT_MODE);
if (lastUpdate.getAndSet(currentUpdate) != currentUpdate) {
Statistics statistics = new Statistics();
- statistics.setCaptureTime(new Date());
- statistics.setActiveUsers(users.size());
- statistics.setActiveDevices(devices.size());
- statistics.setRequests(requests);
- statistics.setMessagesReceived(messagesReceived);
- statistics.setMessagesStored(messagesStored);
- statistics.setMailSent(mailSent);
- statistics.setSmsSent(smsSent);
- statistics.setGeocoderRequests(geocoderRequests);
- statistics.setGeolocationRequests(geolocationRequests);
+
+ synchronized (this) {
+ statistics.setCaptureTime(new Date());
+ statistics.setActiveUsers(users.size());
+ statistics.setActiveDevices(deviceProtocols.size());
+ statistics.setRequests(requests);
+ statistics.setMessagesReceived(messagesReceived);
+ statistics.setMessagesStored(messagesStored);
+ statistics.setMailSent(mailSent);
+ statistics.setSmsSent(smsSent);
+ statistics.setGeocoderRequests(geocoderRequests);
+ statistics.setGeolocationRequests(geolocationRequests);
+ if (!deviceProtocols.isEmpty()) {
+ Map<String, Integer> protocols = new HashMap<>();
+ for (String protocol : deviceProtocols.values()) {
+ protocols.compute(protocol, (key, count) -> count != null ? count += 1 : 1);
+ }
+ statistics.setProtocols(protocols);
+ }
+
+ users.clear();
+ deviceProtocols.clear();
+ requests = 0;
+ messagesReceived = 0;
+ messagesStored = 0;
+ mailSent = 0;
+ smsSent = 0;
+ geocoderRequests = 0;
+ geolocationRequests = 0;
+ }
try {
dataManager.addObject(statistics);
@@ -100,19 +126,16 @@ public class StatisticsManager {
form.param("smsSent", String.valueOf(statistics.getSmsSent()));
form.param("geocoderRequests", String.valueOf(statistics.getGeocoderRequests()));
form.param("geolocationRequests", String.valueOf(statistics.getGeolocationRequests()));
+ if (statistics.getProtocols() != null) {
+ try {
+ form.param("protocols", objectMapper.writeValueAsString(statistics.getProtocols()));
+ } catch (JsonProcessingException e) {
+ LOGGER.warn("Failed to serialize protocols", e);
+ }
+ }
client.target(url).request().async().post(Entity.form(form));
}
-
- users.clear();
- devices.clear();
- requests = 0;
- messagesReceived = 0;
- messagesStored = 0;
- mailSent = 0;
- smsSent = 0;
- geocoderRequests = 0;
- geolocationRequests = 0;
}
}
@@ -129,11 +152,11 @@ public class StatisticsManager {
messagesReceived += 1;
}
- public synchronized void registerMessageStored(long deviceId) {
+ public synchronized void registerMessageStored(long deviceId, String protocol) {
checkSplit();
messagesStored += 1;
if (deviceId != 0) {
- devices.add(deviceId);
+ deviceProtocols.put(deviceId, protocol);
}
}