1 files changed, 115 insertions, 0 deletions
diff --git a/src/main/java/org/traccar/TrackerServer.java b/src/main/java/org/traccar/TrackerServer.java
new file mode 100644
index 000000000..3a1e1c4e8
--- /dev/null
+++ b/src/main/java/org/traccar/TrackerServer.java
@@ -0,0 +1,115 @@
+/*
+ * Copyright 2012 - 2018 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.
+ * 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;
+
+import io.netty.bootstrap.AbstractBootstrap;
+import io.netty.bootstrap.Bootstrap;
+import io.netty.bootstrap.ServerBootstrap;
+import io.netty.channel.Channel;
+import io.netty.channel.group.ChannelGroup;
+import io.netty.channel.group.DefaultChannelGroup;
+import io.netty.channel.socket.nio.NioDatagramChannel;
+import io.netty.channel.socket.nio.NioServerSocketChannel;
+import io.netty.util.concurrent.GlobalEventExecutor;
+
+import java.net.InetSocketAddress;
+
+public abstract class TrackerServer {
+
+    private final boolean datagram;
+    private final AbstractBootstrap bootstrap;
+
+    public boolean isDatagram() {
+        return datagram;
+    }
+
+    public TrackerServer(boolean datagram, String protocol) {
+        this.datagram = datagram;
+
+        address = Context.getConfig().getString(protocol + ".address");
+        port = Context.getConfig().getInteger(protocol + ".port");
+
+        BasePipelineFactory pipelineFactory = new BasePipelineFactory(this, protocol) {
+            @Override
+            protected void addProtocolHandlers(PipelineBuilder pipeline) {
+                TrackerServer.this.addProtocolHandlers(pipeline);
+            }
+        };
+
+        if (datagram) {
+
+            this.bootstrap = new Bootstrap()
+                    .group(EventLoopGroupFactory.getWorkerGroup())
+                    .channel(NioDatagramChannel.class)
+                    .handler(pipelineFactory);
+
+        } else {
+
+            this.bootstrap = new ServerBootstrap()
+                    .group(EventLoopGroupFactory.getBossGroup(), EventLoopGroupFactory.getWorkerGroup())
+                    .channel(NioServerSocketChannel.class)
+                    .childHandler(pipelineFactory);
+
+        }
+    }
+
+    protected abstract void addProtocolHandlers(PipelineBuilder pipeline);
+
+    private int port;
+
+    public int getPort() {
+        return port;
+    }
+
+    public void setPort(int port) {
+        this.port = port;
+    }
+
+    private String address;
+
+    public String getAddress() {
+        return address;
+    }
+
+    public void setAddress(String address) {
+        this.address = address;
+    }
+
+    private final ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
+
+    public ChannelGroup getChannelGroup() {
+        return channelGroup;
+    }
+
+    public void start() throws Exception {
+        InetSocketAddress endpoint;
+        if (address == null) {
+            endpoint = new InetSocketAddress(port);
+        } else {
+            endpoint = new InetSocketAddress(address, port);
+        }
+
+        Channel channel = bootstrap.bind(endpoint).sync().channel();
+        if (channel != null) {
+            getChannelGroup().add(channel);
+        }
+    }
+
+    public void stop() {
+        channelGroup.close().awaitUninterruptibly();
+    }
+
+}