From ef25655365eb97d29295f8e0cb720b86a29c5e26 Mon Sep 17 00:00:00 2001 From: Anton Tananaev Date: Sat, 16 Apr 2016 11:47:03 +1200 Subject: Bind async connection to user id --- src/org/traccar/api/AsyncSocket.java | 10 ++--- src/org/traccar/api/resource/PositionResource.java | 3 +- src/org/traccar/database/ConnectionManager.java | 52 +++++++++------------- src/org/traccar/database/PermissionsManager.java | 14 ++++++ 4 files changed, 42 insertions(+), 37 deletions(-) diff --git a/src/org/traccar/api/AsyncSocket.java b/src/org/traccar/api/AsyncSocket.java index 5353a8d85..35e760613 100644 --- a/src/org/traccar/api/AsyncSocket.java +++ b/src/org/traccar/api/AsyncSocket.java @@ -33,26 +33,26 @@ public class AsyncSocket extends WebSocketAdapter implements ConnectionManager.U private static final String KEY_DEVICES = "devices"; private static final String KEY_POSITIONS = "positions"; - private Collection devices; + private long userId; public AsyncSocket(long userId) { - devices = Context.getPermissionsManager().getDevicePermissions(userId); + this.userId = userId; } @Override public void onWebSocketConnect(Session session) { super.onWebSocketConnect(session); - sendData(KEY_POSITIONS, Context.getConnectionManager().getInitialState(devices)); + sendData(KEY_POSITIONS, Context.getConnectionManager().getInitialState(userId)); - Context.getConnectionManager().addListener(devices, this); + Context.getConnectionManager().addListener(userId, this); } @Override public void onWebSocketClose(int statusCode, String reason) { super.onWebSocketClose(statusCode, reason); - Context.getConnectionManager().removeListener(devices, this); + Context.getConnectionManager().removeListener(userId, this); } @Override diff --git a/src/org/traccar/api/resource/PositionResource.java b/src/org/traccar/api/resource/PositionResource.java index 9c174625c..2da517d4f 100644 --- a/src/org/traccar/api/resource/PositionResource.java +++ b/src/org/traccar/api/resource/PositionResource.java @@ -39,8 +39,7 @@ public class PositionResource extends BaseResource { @QueryParam("deviceId") long deviceId, @QueryParam("from") String from, @QueryParam("to") String to) throws SQLException { if (deviceId == 0) { - return Context.getConnectionManager().getInitialState( - Context.getPermissionsManager().getDevicePermissions(getUserId())); + return Context.getConnectionManager().getInitialState(getUserId()); } else { Context.getPermissionsManager().checkDevice(getUserId(), deviceId); return Context.getDataManager().getPositions( diff --git a/src/org/traccar/database/ConnectionManager.java b/src/org/traccar/database/ConnectionManager.java index e03fd1663..e91aa2d5b 100644 --- a/src/org/traccar/database/ConnectionManager.java +++ b/src/org/traccar/database/ConnectionManager.java @@ -112,9 +112,11 @@ public class ConnectionManager { Log.warning(error); } - if (listeners.containsKey(deviceId)) { - for (UpdateListener listener : listeners.get(deviceId)) { - listener.onUpdateDevice(device); + for (long userId : Context.getPermissionsManager().getDeviceUsers(deviceId)) { + if (listeners.containsKey(userId)) { + for (UpdateListener listener : listeners.get(userId)) { + listener.onUpdateDevice(device); + } } } } @@ -123,9 +125,11 @@ public class ConnectionManager { long deviceId = position.getDeviceId(); positions.put(deviceId, position); - if (listeners.containsKey(deviceId)) { - for (UpdateListener listener : listeners.get(deviceId)) { - listener.onUpdatePosition(position); + for (long userId : Context.getPermissionsManager().getDeviceUsers(deviceId)) { + if (listeners.containsKey(userId)) { + for (UpdateListener listener : listeners.get(userId)) { + listener.onUpdatePosition(position); + } } } } @@ -134,13 +138,13 @@ public class ConnectionManager { return positions.get(deviceId); } - public synchronized Collection getInitialState(Collection devices) { + public synchronized Collection getInitialState(long userId) { List result = new LinkedList<>(); - for (long device : devices) { - if (positions.containsKey(device)) { - result.add(positions.get(device)); + for (long deviceId : Context.getPermissionsManager().getDevicePermissions(userId)) { + if (positions.containsKey(deviceId)) { + result.add(positions.get(deviceId)); } } @@ -152,30 +156,18 @@ public class ConnectionManager { void onUpdatePosition(Position position); } - public void addListener(Collection devices, UpdateListener listener) { - for (long deviceId : devices) { - addListener(deviceId, listener); - } - } - - public synchronized void addListener(long deviceId, UpdateListener listener) { - if (!listeners.containsKey(deviceId)) { - listeners.put(deviceId, new HashSet()); - } - listeners.get(deviceId).add(listener); - } - - public void removeListener(Collection devices, UpdateListener listener) { - for (long deviceId : devices) { - removeListener(deviceId, listener); + public synchronized void addListener(long userId, UpdateListener listener) { + if (!listeners.containsKey(userId)) { + listeners.put(userId, new HashSet()); } + listeners.get(userId).add(listener); } - public synchronized void removeListener(long deviceId, UpdateListener listener) { - if (!listeners.containsKey(deviceId)) { - listeners.put(deviceId, new HashSet()); + public synchronized void removeListener(long userId, UpdateListener listener) { + if (!listeners.containsKey(userId)) { + listeners.put(userId, new HashSet()); } - listeners.get(deviceId).remove(listener); + listeners.get(userId).remove(listener); } } diff --git a/src/org/traccar/database/PermissionsManager.java b/src/org/traccar/database/PermissionsManager.java index dc37bbc84..57307b58a 100644 --- a/src/org/traccar/database/PermissionsManager.java +++ b/src/org/traccar/database/PermissionsManager.java @@ -38,6 +38,7 @@ public class PermissionsManager { private final Map> groupPermissions = new HashMap<>(); private final Map> devicePermissions = new HashMap<>(); + private final Map> deviceUsers = new HashMap<>(); public Set getGroupPermissions(long userId) { if (!groupPermissions.containsKey(userId)) { @@ -53,6 +54,13 @@ public class PermissionsManager { return devicePermissions.get(userId); } + public Set getDeviceUsers(long deviceId) { + if (!deviceUsers.containsKey(deviceId)) { + deviceUsers.put(deviceId, new HashSet()); + } + return deviceUsers.get(deviceId); + } + public PermissionsManager(DataManager dataManager) { this.dataManager = dataManager; refresh(); @@ -88,6 +96,12 @@ public class PermissionsManager { } catch (SQLException error) { Log.warning(error); } + + for (Map.Entry> entry : devicePermissions.entrySet()) { + for (long deviceId : entry.getValue()) { + getDeviceUsers(deviceId).add(entry.getKey()); + } + } } public boolean isAdmin(long userId) { -- cgit v1.2.3