服务器端的webSockets连接管理

时间:2013-05-27 05:49:05

标签: java jsp tomcat servlets websocket

我在服务器端使用网络套接字。我设法从应用程序连接到服务器并向每个打开的连接发送消息/通知。我想要的是

- 如何识别连接(即打开websocket),以便用户可以向特定用户发送消息/通知(通过webSockets)。这是我的代码首先是桌面客户端代码。

if( e.getSource() == connect ) {
        try {
            cc = new WebSocketClient( new URI( "http://localhost:8080/webSocket/chatServlet"), (Draft) draft.getSelectedItem() ) {
                @Override
                public void onMessage( String message ) {
                    ta.append( "got: " + message + "\n" );
                    ta.setCaretPosition( ta.getDocument().getLength() );
                }
               @Override
                public void onOpen( ServerHandshake handshake ) {
                    ta.append( "You are connected to ChatServer: " + getURI() + "\n" );
                    ta.setCaretPosition( ta.getDocument().getLength() );
                }

                @Override
                public void onClose( int code, String reason, boolean wasClean ) {
                    ta.append( "You have been disconnected from: " + getURI() + "; Code: " + code + " " + reason + "\n" );
                    System.out.println("the reason for disconnection is ........ "+wasClean);
                                            ta.setCaretPosition( ta.getDocument().getLength() );
                    connect.setEnabled( true );
                    uriField.setEditable( true );
                    draft.setEditable( true );
                    close.setEnabled( false );
                }

                @Override
                public void onError( Exception ex ) {
                    ta.append( "Exception occured ...\n" + ex + "\n" );
                    ta.setCaretPosition( ta.getDocument().getLength() );
                    ex.printStackTrace();
                    connect.setEnabled( true );
                    uriField.setEditable( true );
                    draft.setEditable( true );
                    close.setEnabled( false );
                }
            };

            close.setEnabled( true );
            connect.setEnabled( false );
            uriField.setEditable( false );
            draft.setEditable( false );
            cc.connect();
        } catch ( URISyntaxException ex ) {
            ta.append( uriField.getText() + " is not a valid WebSocket URI\n" );
        }
    } else if( e.getSource() == close ) {
        cc.close();
    }
}

现在这是服务器端代码......

public class SocketListener extends WebSocketServlet {

/**
 * 
 */
private static final long serialVersionUID = 1L;
private static ArrayList<MyStreamBound> mmiList = new ArrayList<MyStreamBound>();
private HttpServletRequest request;
private String clientName;
private String zone;
private String subId;
@Override
protected StreamInbound createWebSocketInbound(String protocol) {
    System.out.println("protocol values are..."+protocol);
    return new MyStreamBound();

}



private class MyStreamBound extends StreamInbound{
    WsOutbound myoutbound;


    public MyStreamBound(){
        super();

    }
    @Override
    public void onOpen(WsOutbound outbound){
         try {


             System.out.println("Open Client."+outbound.toString()+" and value of this "+this.toString());
             this.myoutbound = outbound;
             mmiList.add(this);
             outbound.writeTextMessage(CharBuffer.wrap("Hello!"));

         } catch (IOException e) {
         e.printStackTrace();
         }
     }


    @Override
    protected void onBinaryData(InputStream arg0) throws IOException {
        // TODO Auto-generated method stub

    }

    @Override
    protected void onTextData(Reader recievedData) throws IOException {
        BufferedReader in = new BufferedReader(recievedData);
        String line = null;
        StringBuilder rslt = new StringBuilder();
        while ((line = in.readLine()) != null) {
            rslt.append(line);
        }
        System.out.println(rslt.toString()); 

        for(MyStreamBound mmib: mmiList){
             CharBuffer buffer = CharBuffer.wrap(rslt.toString());
             mmib.myoutbound.writeTextMessage(buffer);
             mmib.myoutbound.flush();

         }

    }

     @Override
     protected void onClose(int status){
         System.out.println("Close Client."+status);
         mmiList.remove(this);
     }


}

}

1 个答案:

答案 0 :(得分:1)

我最近遇到了同样的问题。

你有一些选择......

  1. 在构造函数中创建MyStreamBound获取SessionID,因此mmib中的每个对象都有一个唯一的ID。

    protected StreamInbound createWebSocketInbound(String protocol, HttpServletRequest req) {
        System.out.println("protocol values are..."+protocol);
        return new MyStreamBound(req.getSession().getId());
    }
    
  2. 可能与之前相关联,以使其更安全。连接时,将第一条消息作为握手发送,您可以在其中发送有关用户的信息。您可以将此信息存储为MyStreamBound Object实例中的字段。

  3. 对于这两种情况,您可以根据需要在for语句中验证此值。

    除了你的问题,作为推荐,我使用的语法有问题:

        for(MyStreamBound mmib: mmiList){
             CharBuffer buffer = CharBuffer.wrap(rslt.toString());
             mmib.myoutbound.writeTextMessage(buffer);
             mmib.myoutbound.flush();
    
         }
    

    它从mmiList生成一个Iterator,如果你有多个并发访问(这就是WebSocket的用途),它将导致异常。我这样写(即使看起来不太好):

        for(int i = 0; i < mmiList.size(); i++) {
             MyStreamBound mmib = mmiList.get(i);
             CharBuffer buffer = CharBuffer.wrap(rslt.toString());
             mmib.myoutbound.writeTextMessage(buffer);
             mmib.myoutbound.flush();
    
         }
    

    这样你就不会搞乱创建集合并使代码成为Threadsafe。事实上,我对此感到有些失望,但这就是增强的for循环在Java中的作用。

    循环的正常效率并不高。我不明白为什么mmiList不是Threadsafe Collection,它没有任何意义。您也可以检查这些替代方案:Thread-safe iteration over a collection如果您不想使用普通for循环。