From 727e7bfad908abd899d1114c5405f9bf69b47ebc Mon Sep 17 00:00:00 2001 From: Anton Tananaev Date: Tue, 30 Jun 2015 09:51:23 +1200 Subject: Active devices in connection manager --- src/org/traccar/database/ActiveDevice.java | 17 ++++ src/org/traccar/database/ConnectionManager.java | 111 ++++++++++++++++++++++++ src/org/traccar/database/DataCache.java | 98 --------------------- src/org/traccar/database/DataManager.java | 12 --- 4 files changed, 128 insertions(+), 110 deletions(-) create mode 100644 src/org/traccar/database/ConnectionManager.java delete mode 100644 src/org/traccar/database/DataCache.java (limited to 'src/org/traccar/database') diff --git a/src/org/traccar/database/ActiveDevice.java b/src/org/traccar/database/ActiveDevice.java index 58c5d210d..f94e1b12a 100644 --- a/src/org/traccar/database/ActiveDevice.java +++ b/src/org/traccar/database/ActiveDevice.java @@ -1,3 +1,18 @@ +/* + * Copyright 2015 Anton Tananaev (anton.tananaev@gmail.com) + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ package org.traccar.database; import org.jboss.netty.channel.Channel; @@ -7,6 +22,7 @@ import org.traccar.command.GpsCommand; import java.net.SocketAddress; public class ActiveDevice { + private String uniqueId; private Protocol protocol; private Channel channel; @@ -34,4 +50,5 @@ public class ActiveDevice { public void write(Object message) { getChannel().write(message, remoteAddress); } + } diff --git a/src/org/traccar/database/ConnectionManager.java b/src/org/traccar/database/ConnectionManager.java new file mode 100644 index 000000000..7ef190fcf --- /dev/null +++ b/src/org/traccar/database/ConnectionManager.java @@ -0,0 +1,111 @@ +/* + * Copyright 2015 Anton Tananaev (anton.tananaev@gmail.com) + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.traccar.database; + +import java.net.SocketAddress; +import java.sql.SQLException; +import java.util.Collection; +import java.util.HashMap; +import java.util.HashSet; +import java.util.LinkedList; +import java.util.List; +import java.util.Map; +import java.util.Set; + +import org.jboss.netty.channel.Channel; +import org.traccar.Protocol; +import org.traccar.helper.Log; +import org.traccar.model.Position; + +public class ConnectionManager { + + private Map activeDevices = new HashMap(); + private final Map positions = new HashMap(); + private final Map> listeners = new HashMap>(); + + public void init(DataManager dataManager) { + try { + Collection positions = dataManager.getLatestPositions(); + for (Position position : positions) { + this.positions.put(position.getDeviceId(), position); + } + } catch (SQLException error) { + Log.warning(error); + } + } + + public void setActiveDevice(String uniqueId, Protocol protocol, Channel channel, SocketAddress remoteAddress) { + activeDevices.put(uniqueId, new ActiveDevice(uniqueId, protocol, channel, remoteAddress)); + } + + public ActiveDevice getActiveDevice(String uniqueId) { + return activeDevices.get(uniqueId); + } + + public synchronized void update(Position position) { + long deviceId = position.getDeviceId(); + positions.put(deviceId, position); + if (listeners.containsKey(deviceId)) { + for (DataCacheListener listener : listeners.get(deviceId)) { + listener.onUpdate(position); + } + } + } + + public synchronized Collection getInitialState(Collection devices) { + + List result = new LinkedList(); + + for (long device : devices) { + if (positions.containsKey(device)) { + result.add(positions.get(device)); + } + } + + return result; + } + + public static interface DataCacheListener { + public void onUpdate(Position position); + } + + public void addListener(Collection devices, DataCacheListener listener) { + for (long deviceId : devices) { + addListener(deviceId, listener); + } + } + + public synchronized void addListener(long deviceId, DataCacheListener listener) { + if (!listeners.containsKey(deviceId)) { + listeners.put(deviceId, new HashSet()); + } + listeners.get(deviceId).add(listener); + } + + public void removeListener(Collection devices, DataCacheListener listener) { + for (long deviceId : devices) { + removeListener(deviceId, listener); + } + } + + public synchronized void removeListener(long deviceId, DataCacheListener listener) { + if (!listeners.containsKey(deviceId)) { + listeners.put(deviceId, new HashSet()); + } + listeners.get(deviceId).remove(listener); + } + +} diff --git a/src/org/traccar/database/DataCache.java b/src/org/traccar/database/DataCache.java deleted file mode 100644 index fc29f0e2e..000000000 --- a/src/org/traccar/database/DataCache.java +++ /dev/null @@ -1,98 +0,0 @@ -/* - * Copyright 2015 Anton Tananaev (anton.tananaev@gmail.com) - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ -package org.traccar.database; - -import java.sql.SQLException; -import java.util.Collection; -import java.util.HashMap; -import java.util.HashSet; -import java.util.LinkedList; -import java.util.List; -import java.util.Map; -import java.util.Set; -import org.traccar.helper.Log; -import org.traccar.model.Position; - -public class DataCache { - - private final Map positions = new HashMap(); - private final Map> listeners = new HashMap>(); - - public void init(DataManager dataManager) { - try { - Collection positions = dataManager.getLatestPositions(); - for (Position position : positions) { - this.positions.put(position.getDeviceId(), position); - } - } catch (SQLException error) { - Log.warning(error); - } - } - - public synchronized void update(Position position) { - long deviceId = position.getDeviceId(); - positions.put(deviceId, position); - if (listeners.containsKey(deviceId)) { - for (DataCacheListener listener : listeners.get(deviceId)) { - listener.onUpdate(position); - } - } - } - - public synchronized Collection getInitialState(Collection devices) { - - List result = new LinkedList(); - - for (long device : devices) { - if (positions.containsKey(device)) { - result.add(positions.get(device)); - } - } - - return result; - } - - public static interface DataCacheListener { - public void onUpdate(Position position); - } - - public void addListener(Collection devices, DataCacheListener listener) { - for (long deviceId : devices) { - addListener(deviceId, listener); - } - } - - public synchronized void addListener(long deviceId, DataCacheListener listener) { - if (!listeners.containsKey(deviceId)) { - listeners.put(deviceId, new HashSet()); - } - listeners.get(deviceId).add(listener); - } - - public void removeListener(Collection devices, DataCacheListener listener) { - for (long deviceId : devices) { - removeListener(deviceId, listener); - } - } - - public synchronized void removeListener(long deviceId, DataCacheListener listener) { - if (!listeners.containsKey(deviceId)) { - listeners.put(deviceId, new HashSet()); - } - listeners.get(deviceId).remove(listener); - } - -} diff --git a/src/org/traccar/database/DataManager.java b/src/org/traccar/database/DataManager.java index 62275af3c..07e50ff5d 100644 --- a/src/org/traccar/database/DataManager.java +++ b/src/org/traccar/database/DataManager.java @@ -17,7 +17,6 @@ package org.traccar.database; import com.mchange.v2.c3p0.ComboPooledDataSource; import java.io.File; -import java.net.SocketAddress; import java.net.URL; import java.net.URLClassLoader; import java.sql.Connection; @@ -34,9 +33,7 @@ import java.util.Properties; import javax.naming.InitialContext; import javax.sql.DataSource; -import org.jboss.netty.channel.Channel; import org.traccar.Context; -import org.traccar.Protocol; import org.traccar.helper.DriverDelegate; import org.traccar.helper.Log; import org.traccar.http.AsyncServlet; @@ -56,7 +53,6 @@ public class DataManager { private DataSource dataSource; private final Map devices = new HashMap(); - private Map activeDevices = new HashMap(); private long devicesLastUpdate; private long devicesRefreshDelay; @@ -79,14 +75,6 @@ public class DataManager { return dataSource; } - public void setActiveDevice(String uniqueId, Protocol protocol, Channel channel, SocketAddress remoteAddress) { - this.activeDevices.put(uniqueId, new ActiveDevice(uniqueId, protocol, channel, remoteAddress)); - } - - public ActiveDevice getActiveDevice(String uniqueId) { - return this.activeDevices.get(uniqueId); - } - private void initDatabase(Properties properties) throws Exception { String jndiName = properties.getProperty("database.jndi"); -- cgit v1.2.3