{"id":7415,"date":"2022-09-23T15:42:16","date_gmt":"2022-09-23T07:42:16","guid":{"rendered":"http:\/\/123.57.164.21\/?p=7415"},"modified":"2022-09-23T15:42:16","modified_gmt":"2022-09-23T07:42:16","slug":"websocket%e7%b3%bb%e5%88%97%e5%9f%ba%e4%ba%8espring-boot-starter-websocket%e5%ae%9e%e7%8e%b0","status":"publish","type":"post","link":"https:\/\/92it.top\/?p=7415","title":{"rendered":"websocket\u7cfb\u5217:\u57fa\u4e8espring-boot-starter-websocket\u5b9e\u73b0"},"content":{"rendered":"\n
\"\"<\/figure>\n\n\n\n
\"\"<\/figure><\/div>\n\n\n\n
\"\"<\/figure>\n\n\n\n
<dependencies>\n        <dependency>\n            <groupId>org.springframework.boot<\/groupId>\n            <artifactId>spring-boot-starter-websocket<\/artifactId>\n        <\/dependency>\n\n        <dependency>\n            <groupId>org.springframework.boot<\/groupId>\n            <artifactId>spring-boot-starter-web<\/artifactId>\n        <\/dependency>\n\n        <dependency>\n            <groupId>org.projectlombok<\/groupId>\n            <artifactId>lombok<\/artifactId>\n            <optional>true<\/optional>\n        <\/dependency>\n        <dependency>\n            <groupId>org.springframework.boot<\/groupId>\n            <artifactId>spring-boot-starter-test<\/artifactId>\n            <scope>test<\/scope>\n        <\/dependency>\n    <\/dependencies>\n<\/pre>\n\n\n\n
\"\"<\/figure>\n\n\n\n
@Configuration\n@EnableWebSocket\npublic class WebSocketConfig implements WebSocketConfigurer {\n\n    @Autowired\n    private MyWsHandler myWsHandler;\n\n    @Override\n    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {\n        registry\n                .addHandler(myWsHandler, \"myWs\")\n                \/\/\u5141\u8bb8\u8de8\u57df\n                .setAllowedOrigins(\"*\");\n    }\n}\n<\/pre>\n\n\n\n
\"\"<\/figure>\n\n\n\n
\/**\n * ws\u6d88\u606f\u5904\u7406\u7c7b\n *\/\n@Component\n@Slf4j\npublic class MyWsHandler extends AbstractWebSocketHandler {\n\n    @Override\n    public void afterConnectionEstablished(WebSocketSession session) throws Exception {\n        log.info(\"\u5efa\u7acbws\u8fde\u63a5\");\n        WsSessionManager.add(session.getId(),session);\n    }\n\n    @Override\n    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {\n        log.info(\"\u53d1\u9001\u6587\u672c\u6d88\u606f\");\n        \/\/ \u83b7\u5f97\u5ba2\u6237\u7aef\u4f20\u6765\u7684\u6d88\u606f\n        String payload = message.getPayload();\n        log.info(\"server \u63a5\u6536\u5230\u6d88\u606f \" + payload);\n        session.sendMessage(new TextMessage(\"server \u53d1\u9001\u7ed9\u7684\u6d88\u606f \" + payload + \"\uff0c\u53d1\u9001\u65f6\u95f4:\" + LocalDateTime.now().toString()));\n    }\n\n    @Override\n    protected void handleBinaryMessage(WebSocketSession session, BinaryMessage message) throws Exception {\n        log.info(\"\u53d1\u9001\u4e8c\u8fdb\u5236\u6d88\u606f\");\n    }\n\n    @Override\n    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {\n        log.error(\"\u5f02\u5e38\u5904\u7406\");\n        WsSessionManager.removeAndClose(session.getId());\n    }\n\n    @Override\n    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {\n        log.info(\"\u5173\u95edws\u8fde\u63a5\");\n        WsSessionManager.removeAndClose(session.getId());\n    }\n}\n<\/pre>\n\n\n\n
\"\"<\/figure>\n\n\n\n
\"\"<\/figure><\/div>\n\n\n\n
\"\"<\/figure>\n\n\n\n
\/**\n * \u6d88\u606f\u751f\u6210job\n *\/\n@Slf4j\n@Component\npublic class MessageJob {\n    @Autowired\n    WsService wsService;\n\n    \/**\n     * \u6bcf5s\u53d1\u9001\n     *\/\n    @Scheduled(cron = \"0\/5 * * * * *\")\n    public void run(){\n        try {\n            wsService.broadcastMsg(\"\u81ea\u52a8\u751f\u6210\u6d88\u606f \"  + LocalDateTime.now().toString());\n        } catch (IOException e) {\n            e.printStackTrace();\n        }\n    }\n}\n<\/pre>\n\n\n\n
\"\"<\/figure>\n\n\n\n
\/**\n * ws\u64cd\u4f5c\u76f8\u5173\u670d\u52a1\n *\/\n@Service\n@Slf4j\npublic class WsService {\n\n    \/**\n     * \u53d1\u9001\u6d88\u606f\n     * @param session\n     * @param text\n     * @return\n     * @throws IOException\n     *\/\n    public void sendMsg(WebSocketSession session, String text) throws IOException {\n        session.sendMessage(new TextMessage(text));\n    }\n\n    \/**\n     * \u5e7f\u64ad\u6d88\u606f\n     * @param text\n     * @return\n     * @throws IOException\n     *\/\n    public void broadcastMsg(String text) throws IOException {\n        for (WebSocketSession session: WsSessionManager.SESSION_POOL.values()) {\n            session.sendMessage(new TextMessage(text));\n        }\n    }\n\n}\n<\/pre>\n\n\n\n
\"\"<\/figure>\n\n\n\n
@Slf4j\npublic class WsSessionManager {\n    \/**\n     * \u4fdd\u5b58\u8fde\u63a5 session \u7684\u5730\u65b9\n     *\/\n    public  static ConcurrentHashMap<String, WebSocketSession> SESSION_POOL = new ConcurrentHashMap<>();\n\n    \/**\n     * \u6dfb\u52a0 session\n     *\n     * @param key\n     *\/\n    public static void add(String key, WebSocketSession session) {\n        \/\/ \u6dfb\u52a0 session\n        SESSION_POOL.put(key, session);\n    }\n\n    \/**\n     * \u5220\u9664 session,\u4f1a\u8fd4\u56de\u5220\u9664\u7684 session\n     *\n     * @param key\n     * @return\n     *\/\n    public static WebSocketSession remove(String key) {\n        \/\/ \u5220\u9664 session\n        return SESSION_POOL.remove(key);\n    }\n\n    \/**\n     * \u5220\u9664\u5e76\u540c\u6b65\u5173\u95ed\u8fde\u63a5\n     *\n     * @param key\n     *\/\n    public static void removeAndClose(String key) {\n        WebSocketSession session = remove(key);\n        if (session != null) {\n            try {\n                \/\/ \u5173\u95ed\u8fde\u63a5\n                session.close();\n            } catch (IOException e) {\n                \/\/ todo: \u5173\u95ed\u51fa\u73b0\u5f02\u5e38\u5904\u7406\n                e.printStackTrace();\n            }\n        }\n    }\n\n    \/**\n     * \u83b7\u5f97 session\n     *\n     * @param key\n     * @return\n     *\/\n    public static WebSocketSession get(String key) {\n        \/\/ \u83b7\u5f97 session\n        return SESSION_POOL.get(key);\n    }\n}\n<\/pre>\n\n\n\n
\"\"<\/figure>\n\n\n\n
@SpringBootApplication\n@EnableScheduling\npublic class SpringWsApplication {\n\n    public static void main(String[] args) {\n        SpringApplication.run(SpringWsApplication.class, args);\n    }\n\n}\n<\/pre>\n\n\n\n
\"\"<\/figure>\n\n\n\n
<!DOCTYPE HTML>\n<html>\n<head>\n    <title>My WebSocket<\/title>\n<\/head>\n\n<body>\n<input id=\"text\" type=\"text\" \/>\n<button onclick=\"send()\">Send<\/button>\n<button onclick=\"closeWebSocket()\">Close<\/button>\n<div id=\"message\"><\/div>\n\n\n<\/body>\n\n<script type=\"text\/javascript\">\n\n    let ws = null;\n    \/\/\u5224\u65ad\u5f53\u524d\u6d4f\u89c8\u5668\u662f\u5426\u652f\u6301WebSocket\n    if ('WebSocket' in window) {\n        ws = new WebSocket(\"ws:\/\/localhost:8333\/myWs\");\n    }\n    else {\n        alert('\u5f53\u524d\u6d4f\u89c8\u5668 Not support websocket')\n    }\n\n    \/\/\u8fde\u63a5\u53d1\u751f\u9519\u8bef\u7684\u56de\u8c03\u65b9\u6cd5\n    ws.onerror = function () {\n        setMessageInnerHTML(\"WebSocket\u8fde\u63a5\u53d1\u751f\u9519\u8bef\");\n    };\n\n    \/\/\u8fde\u63a5\u6210\u529f\u5efa\u7acb\u7684\u56de\u8c03\u65b9\u6cd5\n    ws.onopen = function(event) {\n        console.log(\"ws\u8c03\u7528\u8fde\u63a5\u6210\u529f\u56de\u8c03\u65b9\u6cd5\")\n        \/\/ws.send(\"\")\n    }\n    \/\/\u63a5\u6536\u5230\u6d88\u606f\u7684\u56de\u8c03\u65b9\u6cd5\n    ws.onmessage = function(message) {\n        console.log(\"\u63a5\u6536\u6d88\u606f\uff1a\" + message.data);\n        if (typeof(message.data) == 'string') {\n            setMessageInnerHTML(message.data);\n        }\n    }\n    \/\/ws\u8fde\u63a5\u65ad\u5f00\u7684\u56de\u8c03\u65b9\u6cd5\n    ws.onclose = function(e) {\n        console.log(\"ws\u8fde\u63a5\u65ad\u5f00\")\n        \/\/console.log(e)\n        setMessageInnerHTML(\"ws close\");\n    }\n\n    \/\/\u5c06\u6d88\u606f\u663e\u793a\u5728\u7f51\u9875\u4e0a\n    function setMessageInnerHTML(innerHTML) {\n        console.log(innerHTML)\n        document.getElementById('message').innerHTML += '\u63a5\u6536\u7684\u6d88\u606f:' + innerHTML + '<br\/>';\n    }\n\n    \/\/\u5173\u95ed\u8fde\u63a5\n    function closeWebSocket() {\n        ws.close();\n    }\n\n\n    \/\/\u53d1\u9001\u6d88\u606f\n    function send(msg) {\n        if(!msg){\n            msg = document.getElementById('text').value;\n            document.getElementById('message').innerHTML += \"\u53d1\u9001\u7684\u6d88\u606f:\" + msg + '<br\/>';\n            ws.send(msg);\n        }\n    }\n<\/script>\n<\/html>\n\n<\/pre>\n\n\n\n
\"\"<\/figure><\/div>\n\n\n\n
\"\"<\/figure>\n","protected":false},"excerpt":{"rendered":"","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[26,27],"tags":[],"_links":{"self":[{"href":"https:\/\/92it.top\/index.php?rest_route=\/wp\/v2\/posts\/7415"}],"collection":[{"href":"https:\/\/92it.top\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/92it.top\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/92it.top\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/92it.top\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=7415"}],"version-history":[{"count":2,"href":"https:\/\/92it.top\/index.php?rest_route=\/wp\/v2\/posts\/7415\/revisions"}],"predecessor-version":[{"id":7432,"href":"https:\/\/92it.top\/index.php?rest_route=\/wp\/v2\/posts\/7415\/revisions\/7432"}],"wp:attachment":[{"href":"https:\/\/92it.top\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=7415"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/92it.top\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=7415"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/92it.top\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=7415"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}