2017-03-03 203 views
0

在一个WebSocket中接收消息时,将此消息转发给其他连接的WebSocket的问题。我使用jetty-9.2.20.v20161216。会话管理Jetty Websocket服务器

所以我初始化服务器..

JettyWSServer websocketServer = new JettyWSServer("localhost", 8000, new MySocketHandler(), new QueuedThreadPool(128)); 

public <T extends WebSocketHandler> JettyWSServer(String hostName, int port, T webscoketHandler, QueuedThreadPool threadPool) { 
    this.hostName = hostName; 
    this.port = port; 
    this.handler = webscoketHandler; 
    this.threadPool = threadPool; 
    this.socket = null; 

    //create server 
    this.server = new Server(this.threadPool); 

    //set connector 
    ServerConnector connector = new ServerConnector(server); 
    connector.setHost(this.hostName); 
    connector.setPort(this.port); 
    this.server.addConnector(connector); 

    //set handler 
    this.server.setHandler(this.handler); 

    //set listener 
    setLifecycleListener(); 
} 

MySocketHandler.java

public class MySocketHandler extends WebSocketHandler { 

    private final String TAG = MySocketHandler.class.getSimpleName(); 
    private MySocketCreator creator; 

    @Override 
    public void configure(WebSocketServletFactory webSocketServletFactory) { 
     this.creator = new MySocketCreator(); 
     webSocketServletFactory.setCreator(this.creator); 
    } 


    public Set<ServerSocket> getSockets(){ 
     return this.creator.getSockets(); 
    } 
} 

MySocketCreator.java

public class MySocketCreator implements WebSocketCreator { 

    private static final String TAG = MySocketCreator.class.getSimpleName(); 
    private static Log log = new Log(TAG, true); 

    private Set<ServerSocket> sockets = new HashSet<>(); 
    private Set<Session> guests = new HashSet<>(); 
    private ConcurrentHashMap<ServiceUser, ArrayList<WSDeviceSessionWrapper>> users = new ConcurrentHashMap<>(); 


    @Override 
    public Object createWebSocket(ServletUpgradeRequest servletUpgradeRequest, ServletUpgradeResponse servletUpgradeResponse) { 
     ServerSocket socket = new ServerSocket(statusCallback); 
     sockets.add(socket); 
     return socket; 
    } 


    private OnSessionStatusListener statusCallback = new OnSessionStatusListener() { 
     @Override 
     public void onGuestIn(Session session) { 


      synchronized (this) { 
       guests.add(session); 
       Integer totalAgeReduce = users.values() 
         .stream() 
         .map(wsDeviceSessionWrappers -> { 
          return 1; 
         }) 
         .reduce(
           0, 
           (a, b) -> a + b); 
       log.debug("onGuestIn() " + "Guests: " + guests.size() + " Registered: " + totalAgeReduce); 
      } 
     } 

     @Override 
     public void onUserIn(Session session, ServiceUser user, Device device) { 


      synchronized (this) { 

       if (guests.contains(session)) guests.remove(session); 

       if (!users.containsKey(user)) { 
        users.put(user, new ArrayList<WSDeviceSessionWrapper>()); 
       } 
       users.get(user).add(new WSDeviceSessionWrapper(session, device)); 

       log.debug("onUserIn() " + "Guests: " + guests.size() + " Registered: " + users.size()); 
      } 

     } 

     @Override 
     public void sendResponse(ArrayList<ServiceUser> users, WSResponse response) { 
     log.debug("Send message to [" + (users != null ? users.size() : null) + "] current users " + MySocketCreator.this.users.size()); 
     MySocketCreator.this.users.keySet().forEach(user -> { 
       users.forEach(u -> { 
        if (user.equals(u)) { 

         ArrayList<WSDeviceSessionWrapper> wsDeviceSessionWrappers = MySocketCreator.this.users.get(user); 
         new ArrayList<>(wsDeviceSessionWrappers).forEach(wrapper -> { 
           wrapper.getSession().getRemote().sendStringByFuture(response.toJSON()); 
          } 
         }); 

        } 
       }); 
      }); 


     } 

     @Override 
     public void sendResponse(ServiceUser user, WSResponse response, Device excludeDevice) { 
       MySocketCreator.this.users.get(user).forEach(wrapper -> { 
        wrapper.getSession().getRemote().sendStringByFuture(response.toJSON()); 
       }); 
     } 

     @Override 
     public void onExit(Session session, ServiceUser user, Device device) { 

      synchronized (this) { 

       //remove from guest sessions 
       if (session != null && guests.contains(session)) guests.remove(session); 


       if (user != null && device != null && users.containsKey(user)) { 
        ArrayList<WSDeviceSessionWrapper> wrappers = users.get(user); 
        Iterator<WSDeviceSessionWrapper> iterator = wrappers.iterator(); 
        while (iterator.hasNext()) { 
         WSDeviceSessionWrapper wrapper = iterator.next(); 
         if (wrapper.getSession() == session || wrapper.getSession().equals(session) && wrapper.getDevice() == device || wrapper.getDevice().equals(device)) { 
          //remove session for current device 
          iterator.remove(); 

          //if user does not have session on server 
          //remove him from current server users 
          if (wrappers.size() == 0) { 
           users.remove(user); 
          } 
         } 
        } 
       } 

       Integer totalRegisteredDevices = users.values() 
         .stream() 
         .map(wsDeviceSessionWrappers -> { 
          return 1; 
         }) 
         .reduce(
           0, 
           (a, b) -> a + b); 
       log.debug("onExit() " + "Guests: " + guests.size() + " Registered: " + totalRegisteredDevices); 
      } 


     } 
    }; 

    public Set<ServerSocket> getSockets() { 
     return sockets; 
    } 

} 

这段代码的逻辑:

MySocketCreator的类中,当创建一个新的套接字时,我传递给回调函数。接下来在事件onOpen的套接字中,我调用回调并将它传递给会话,并将会话存储在MySocketCreator类中,在此会话之后,我将与用户和设备相关联。

的问题是,当我尝试通过回调方法

@Override 
    public void sendResponse(ArrayList<ServiceUser> users, WSResponse response) { 
    log.debug("Send message to [" + (users != null ? users.size() : null) + "] current users " + MySocketCreator.this.users.size()); 
     MySocketCreator.this.users.keySet().forEach(user -> { 
      users.forEach(u -> { 
       if (user.equals(u)) { 
        ArrayList<WSDeviceSessionWrapper> wsDeviceSessionWrappers = MySocketCreator.this.users.get(user); 
        new ArrayList<>(wsDeviceSessionWrappers).forEach(wrapper -> { 
          wrapper.getSession().getRemote().sendStringByFuture(response.toJSON()); 
         } 
        }); 

       } 
      }); 
     }); 
    } 

字符串将消息发送给所有用户,从任何websocketwrapper.getSession().getRemote().sendStringByFuture(response.toJSON());

锁的线程和服务器不能再工作。试图取代它

wrapper.getSession().getRemote().sendString(response.toJSON());

抛出异常

java.lang.IllegalStateException: Blocking message pending 10000 for BLOCKING 

org.eclipse.jetty.websocket.api.WebSocketException: RemoteEndpoint unavailable, outgoing connection not open 

这两个选项不会对300间的连接工作/

问题:我怎样才能将消息发送给所有用户?

回答

0

您不能盲目地向所有用户发送消息。

如果某些用户拥塞并且没有像其他用户一样快速读取该怎么办?

如果一个用户以更快的速度生成邮件,那么您的网络连接速度最慢?

在你的情况下,你必须处理Future你尽快回到sendStringByFuture()(或可替代的sendString(String,WriteCallback)),如果是连接速度慢,或失败,你必须从你的用户列表中删除。如果您收到意味着该消息已发送的事件,那么您知道该特定用户/客户端不拥塞且可以自由发送另一条消息。

您发送的循环必须为每个用户/客户端排队消息,而不是作为一个整体,但每个用户/客户端都有自己的队列。

只有在发送时已知该用户未拥塞,您的环路才会发送。

您甚至可能需要删除消息以真正减慢客户端速度,或者在队列变得太大时完全断开它们。

是的,这是很复杂的,这就是为什么有这么多的库建立在websocket之上才能做到这一点。

考虑使用cometd及其websocket传输。

+0

当我使用sendStringByFuture和sendString(String,WriteCallback)并且如果客户端没有收到消息,循环不会向其他用户发送消息。我如何解决它? –