Add CoAP over Websocket interface in cloud

[platform/upstream/iotivity.git] / cloud / stack / src / main / java / org / iotivity / cloud / base / ServerSystem.java

diff --git a/cloud/stack/src/main/java/org/iotivity/cloud/base/ServerSystem.java b/cloud/stack/src/main/java/org/iotivity/cloud/base/ServerSystem.java
index 298fa7a..ddd8481 100644 (file)
--- a/cloud/stack/src/main/java/org/iotivity/cloud/base/ServerSystem.java
+++ b/cloud/stack/src/main/java/org/iotivity/cloud/base/ServerSystem.java
@@ -21,6 +21,11 @@
  */
 package org.iotivity.cloud.base;
 
+import io.netty.channel.ChannelHandler.Sharable;
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.channel.SimpleChannelInboundHandler;
+import io.netty.util.AttributeKey;
+
 import java.util.ArrayList;
 import java.util.List;
 
@@ -41,6 +46,7 @@ import org.iotivity.cloud.base.resource.ResourceManager;
 import org.iotivity.cloud.base.server.CoapServer;
 import org.iotivity.cloud.base.server.HttpServer;
 import org.iotivity.cloud.base.server.Server;
+import org.iotivity.cloud.base.server.WebSocketServer;
 import org.iotivity.cloud.util.Log;
 
 import io.netty.channel.ChannelHandler.Sharable;
@@ -146,7 +152,7 @@ public class ServerSystem extends ResourceManager {
     }
 
     public void addServer(Server server) {
-        if (server instanceof CoapServer) {
+        if (server instanceof CoapServer || server instanceof WebSocketServer) {
             server.addHandler(new PersistentPacketReceiver());
         } else if (server instanceof HttpServer) {
             server.addHandler(new NonPersistentPacketReceiver());