java推送之spring-websocket

準備

導入依賴

 compile group: 'org.springframework', name: 'spring-messaging', version: '4.3.3.RELEASE'
 compile group: 'org.springframework', name: 'spring-websocket', version: '4.3.3.RELEASE'

方式一

<bean id="websocket" class="net.rock.projects.orp.web.websocket.WebsocketEndPoint" />
<websocket:handlers allowed-origins="*">  
    <websocket:mapping path="/websocket" handler="websocket"/>  
    <websocket:handshake-interceptors>  
    <bean class="com.up.websocket.HandshakeInterceptor"/>  
    </websocket:handshake-interceptors>  
</websocket:handlers>

其中，path對應的路徑就是前段通過ws協議調的接口路徑
創建握手接口（攔截器）

public class HandshakeInterceptor extends HttpSessionHandshakeInterceptor {
    @Override
    public boolean beforeHandshake(ServerHttpRequest request,
                                   ServerHttpResponse response, WebSocketHandler wsHandler,
                                   Map<String, Object> attributes) throws Exception {

        HttpServletRequest servletRequest = ((ServletServerHttpRequest) request).getServletRequest();
        HttpSession session = servletRequest.getSession(false);
        User user = (User) session.getAttribute("token");

        if(user!=null){
            attributes.put("token", user);
        }
        return super.beforeHandshake(request, response, wsHandler, attributes);
    }

    @Override
    public void afterHandshake(ServerHttpRequest request,
                               ServerHttpResponse response, WebSocketHandler wsHandler,
                               Exception ex) {
        super.afterHandshake(request, response, wsHandler, ex);
    }
}

創建處理類

@Component
public class WebsocketEndPoint extends TextWebSocketHandler {

    @Autowired
    IProjectDeliverService iProjectDeliverService;
    @Autowired
    IProjectDeliverReplyService iProjectDeliverReplyService;
    @Autowired
    private IUserService iUserService;

    /**
     * 所有的用戶
     */
    private static final Map<Integer , WebSocketSession > users ;

    static{
        users = new HashMap<>();
    }

    @Override
    protected void handleTextMessage(WebSocketSession session,
                                     TextMessage message) throws Exception {

        super.handleTextMessage(session, message);
        session.sendMessage(message);
    }
    @Override
    public void afterConnectionEstablished(final WebSocketSession session) throws Exception {

        User user = (User) session.getAttributes().get("token");
        if(user != null)
            users.put(user.getId() , session);
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {

        User user = (User) session.getAttributes().get("token");

        // 如果用戶對象還存在內存中,那么不進行退出記錄
        if(user != null){
            iUserService.exit(DateUtils.dateToString(new Date()),user.getId());
            System.out.println(user.getNickName()+" exits from the system ! Connection Closed！");
        }
    }


    /**
     * 發送給指定用戶
     * @param userIds    用戶編號數組,不傳(長度=0) 默認發送給所有用戶
     * @param message   消息內容
     */
    public static void sendToUser( TextMessage message , List<Integer> userIds){
        try {
            // 如果沒有傳用戶編號 , 默認發送給所有用戶
            if(userIds == null || userIds.size() == 0){
                if(users.keySet() != null && users.keySet().size() > 0 ){
                    for (Integer key : users.keySet()) {
                        if(users.get(key).isOpen())
                            users.get(key).sendMessage(message);
                    }
                }
            }else{  // 發送給指定的用戶

                for (Integer id : userIds) {
                    if(users.containsKey(id) && users.get(id).isOpen()){
                        users.get(id).sendMessage(message);
                    }
                }

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

用戶頁面連接(sockjs-0.3.min.js自行下載)

<script src="<%=resourcePath%>js/sockjs-0.3.min.js"></script>
<script>
    var ws = null;
    var url = "ws://<%=basePath.replace("http://","")%>/msgcenter?uid=20125";

    function socketConnection(){
        if (!url) {
            alert('Select whether to use W3C WebSocket or SockJS');
            return;
        }

        ws = new WebSocket(url); //申請一個WebSocket對象

        ws.onopen = function () {
            console.info("連接已經打開。。。。。。");
        };
        ws.onmessage = function (event) {
            console.info("你接受到消息是："+event.data);
                }
            }
        };
        ws.onclose = function (event) {
            console.info("連接已經斷開。。。。。" );
        };
    }
 $(function(){
        socketConnection();
    });
</script>

后臺調用sendToUser給指定用戶發送信息

.................................................................................................

方式二

websocket入口

@Configuration
@EnableWebMvc
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        //setAllowedOrigins("*") 解決跨域問題
        registry.addHandler(systemWebSocketHandler(),"/webSocketServer").addInterceptors(new WebSocketHandshakeInterceptor()).setAllowedOrigins("*");

        registry.addHandler(systemWebSocketHandler(), "/sockjs/webSocketServer").addInterceptors(new WebSocketHandshakeInterceptor()).withSockJS();
    }

    @Bean
    public WebSocketHandler systemWebSocketHandler(){
        return new SystemWebSocketHandler();
    }

}

攔截器,beforeHandshake在調用handler前處理方法。常用在注冊用戶信息，綁定WebSocketSession，在handler里根據用戶信息獲取WebSocketSession發送消息。

public class WebSocketHandshakeInterceptor implements HandshakeInterceptor {
    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {


        if (request instanceof ServletServerHttpRequest) {
            ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
            HttpSession session = servletRequest.getServletRequest().getSession(false);
            if (session != null) {
                //使用userName區分WebSocketHandler，以便定向發送消息
                attributes.put("userId",session.getAttribute("userId"));
            }
        }
        return true;
    }

    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {

    }
}

handle處理器

public class SystemWebSocketHandler implements WebSocketHandler {

    private static final Map<String,WebSocketSession> users;

    static {
        users = new HashMap<>();
    }


    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {

        String userId = (String) session.getAttributes().get("userId");
        if(userId !=null){
            users.put(userId,session);
        }
        session.sendMessage(new TextMessage("hello world"));
        System.out.println("建立連接");
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {

    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        System.out.println("連接被關閉");
        for (Map.Entry<String, WebSocketSession> entry : users.entrySet()) {
            if(entry.getValue().equals(session)){
                users.remove(entry.getKey());
            }
        }
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        System.out.println("連接被關閉");
        users.remove(session);
    }

    @Override
    public boolean supportsPartialMessages() {
        return false;
    }

    /**
     * 給某個用戶發送消息
     *
     * @param userId
     * @param message
     */
    public static void  sendMessageToUser(String userId, TextMessage message) throws IOException {
        Set<Map.Entry<String, WebSocketSession>> entries = users.entrySet();
        for (Map.Entry<String, WebSocketSession> entry : entries) {
            if(entry.getKey().equals(userId))
                entry.getValue().sendMessage(message);
        }
    }
}

用戶建立socket連接(sockjs-0.3.min.js自行下載)

<%@ page contentType="text/html;charset=UTF-8" language="java" %>
<%
  String path = request.getContextPath();
  String basePath = request.getScheme()+"://"+request.getServerName()+":"+request.getServerPort()+path;
  String url = request.getServerName()+":"+request.getServerPort()+path;
%>
<html>
  <head>
    <title>index</title>
  </head>
  <body>
    <h3 align="center">
      消息通知：<span id="msg"></span>
    </h3>
  </body>
</html>
<script src="<%=basePath%>/js/jquery-1.9.1.min.js"></script>
<script src="<%=basePath%>/js/sockjs-0.3.min.js"></script>
<script>
    var websocket;
    if ('WebSocket' in window) {
        websocket = new WebSocket("ws://<%=url%>/webSocketServer?userId=1");
    } else if ('MozWebSocket' in window) {
        websocket = new MozWebSocket("ws://<%=url%>/webSocketServer?userId=1");
    } else {
        websocket = new SockJS("http://<%=url%>/sockjs/webSocketServer?userId=1");
    }
    websocket.onopen = function (evnt) {
        console.info("連接已建立")
    };
    websocket.onmessage = function (evnt) {
        console.info("本次推送內容是："+evnt.data);
        $("#msg").html(evnt.data)
    };
    websocket.onerror = function (evnt) {
        console.info("連接已斷開")
    };
    websocket.onclose = function (evnt) {
        console.info("連接已斷開")
    }

</script>

調用 sendMessageToUser推送信息

/**
     * 消息推送
     * @param msg  消息內容
     * @param userId 推送用戶id
     */
    @RequestMapping(value = "pushMsg",method = RequestMethod.POST)
    @ResponseBody
    public String pushMsg(String userId, String msg) {
        System.out.println("用戶id:"+userId+"    推送的消息："+msg);
        try {
            SystemWebSocketHandler.sendMessageToUser(userId,new TextMessage(msg));
            return new Gson().toJson("success");
        } catch (Exception e) {
            System.out.println(e.getMessage()+"................");
            e.printStackTrace();
            return new Gson().toJson("fail:"+e);
        }
    }

相關鏈接
http://www.bridgeli.cn/archives/262
https://my.oschina.net/ldl123292/blog/304360
http://www.cnblogs.com/nosqlcoco/p/5860730.html

最后編輯于 ：
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明：文章內容（如有圖片或視頻亦包括在內）由作者上傳并發布，文章內容僅代表作者本人觀點，簡書系信息發布平臺，僅提供信息存儲服務。

推薦閱讀更多精彩內容

  • Spring Cloud為開發人員提供了快速構建分布式系統中一些常見模式的工具（例如配置管理，服務發現，斷路器，智...
    卡卡羅2017閱讀 134,923評論 18贊 139
  • Android 自定義View的各種姿勢1 Activity的顯示之ViewRootImpl詳解 Activity...
    passiontim閱讀 173,285評論 25贊 708
  • 1. Java基礎部分 基礎部分的順序：基本語法，類相關的語法，內部類的語法，繼承相關的語法，異常的語法，線程的語...
    子非魚_t_閱讀 31,765評論 18贊 399
  • 版權歸作者所有，任何形式轉載請聯系作者。 作者：楊從周（來自豆瓣） 來源：https://www.douban.c...
    沒房沒車農村戶口閱讀 363評論 0贊 0
  • 感覺快被小臭孩弄成抑郁癥了，他爸爸還有奶奶給他養的壞習慣要讓我來改正，真的耐心不足。不過總的來說今天進步了一些，雖...
    冰糖糖冰閱讀 142評論 1贊 0