174. Spring Boot WebSocket:群聊-昵称

需求缘起:

       在上一篇文章已经可以群聊功能了,但是存在一个问题,就是无法显示消息是谁发送了,这样就会导致消息乱七八糟的无法解读。在这版文章中解决昵称显示的问题。

效果展示:

悟空消息:



师傅消息:



八戒消息:



(1)在连接的时候,让用户输入昵称

       这个简单,只需要提供一个input让用户进行输入即可:

    昵称:<input type="text" id="nickname"/>
           <button οnclick="conectWebSocket()">连接WebSocket</button>

(2)连接前的校验

       在用户点击连接判断用户是否输入昵称了:

    var nickname = document.getElementById("nickname").value;
    if(nickname==""){
           alert("请输入昵称");
           return;
    }

    以上这些都是简单了,那问题的关键就是如何把昵称显示在聊天信息中呢?这里提供几种思路,大家可以自己开拓方法。

(3)昵称显示方式1:消息携带

       这种方式就是在发送消息的时候,直接拼接上用户的昵称,这种改动是最小的,也是比较简单的方式,只需要修改前端网页代码即可:

    //获取输入的文本信息进行发送
    var message = document.getElementById('text').value;
    websocket.send(nickname+":"+message);

  但是这种方式有一个弊端,就是服务端要知道用户的昵称就比较费劲了,所以这种方式就是玩玩而已,实际不这么玩。

(4)昵称显示方式2:连接传递

       这个就是在建立连接的时候传递到服务端连接点。

首先修改客户端的连接方式:

websocket = new WebSocket("ws://localhost:8080/websocket/"+nickname); 

 服务端的服务端点发布:

    @ServerEndpoint(value = "/websocket/{nickname}")
    @Component
    public class MyWebSocket {}

服务端获取参数:

       这个在每个方法中都可以获取的,使用@PathParam("nickname") String nickname,比如在连接的时候:

    @OnOpen
    public void onOpen(Session session,@PathParam("nickname") String nickname) {
            this.session = session;
            this.nickname = nickname;
    }

(5)昵称显示方式3:使用httpSession

       这种方式实现起来就比较复杂了,这里只说下大概的一个思路,就是将昵称保存到httpSession对象中,使用httpSession.setAttribute("nickname",nickname)进行存储。

       问题的关键是如何在websocket中获取到httpSession呢,主要是通过添加一个配置类进行实现:

    //配置类  将http中的session传入websocket中
     public class GetHttpSessionConfigurator extends
                ServerEndpointConfig.Configurator {
            @Override
            public void modifyHandshake(ServerEndpointConfig config,
                    HandshakeRequest request, HandshakeResponse response) {
                HttpSession httpSession = (HttpSession) request.getHttpSession();
                config.getUserProperties().put(HttpSession.class.getName(), httpSession);
            }
        }

       那么在WebSocket就可以获取到HttpSession对象了:

    @OnOpen
    public void onOpen(Session session, EndpointConfig config) {
            this.session = session;
            httpSession = (HttpSession) config.getUserProperties().get(
                    HttpSession.class.getName());
    }

这种方式具体可以参考链接的文章,博主并没有具体的进行实现,只是简单看了下文章内容,具体地址如下:

http://blog.csdn.net/z719725611/article/details/52400404

http://blog.csdn.net/yayicho/article/details/52330627

方案总结

       本篇文章最终使用的是方式2的方式进行实现,对于方式2和方式3都有各自的使用场景,大家可以自己进行斟酌。


购买完整视频,请前往:http://www.mark-to-win.com/TeacherV2.html?id=287