java注解實現(xiàn)websocket服務的兩種方式

更新時間：2024年08月03日 08:25:28 作者：humannoid

Java WebSocket是一種基于TCP協(xié)議的雙向全雙工消息傳輸技術,它允許服務器和客戶端之間實時通信,具有低延遲和高效率的特點,下面這篇文章主要給大家介紹了關于java注解實現(xiàn)websocket服務的兩種方式,需要的朋友可以參考下

1.基于java注解實現(xiàn)websocket服務器端

1.1需要的類

1.1.1服務終端類

用java注解來監(jiān)聽連接@ServerEndpoint、連接成功@OnOpen、連接失敗@OnClose、收到消息等狀態(tài)@OnMessage

1.1.2配置類

把spring中的ServerEndpointExporter對象注入進來

2.1代碼示例

2.1.1 maven配置

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>com.heima</groupId>
    <artifactId>ws-demo</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter</artifactId>
           <version>2.7.3</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.16.22</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.springframework.boot/spring-boot-starter-websocket -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
            <version>2.7.14</version>
        </dependency>



    </dependencies>

</project>

2.1.2 WsServerEndpoint類

package com.heima;

import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/***
 * 監(jiān)聽websocket地址  /myWs
 */
@ServerEndpoint("/myWs")
@Component
@Slf4j
@EnableScheduling
public class WsServerEndpoint {

    static Map<String,Session> map = new ConcurrentHashMap<String,Session>();

    /***
     * 連接建立時執(zhí)行的操作
     * @param session
     */
    @OnOpen
    public void onOpen(Session session)
    {
        map.put(session.getId(),session);
        log.info("websocket is open");
    }

    /***
     * 收到客戶端消息執(zhí)行的操作
     * @param text
     */
    @OnMessage
    public String OnMessage(String text)
    {
        log.info("收到了一條信息"+text);
        return "已收到你的信息" ;
    }

    /***
     * 連接關閉時執(zhí)行的操作
     * @param session
     */
    @OnClose
    public void OnClose(Session session)
    {
        map.remove(session.getId());
        log.info("連接關閉時執(zhí)行的操作");
    }

    /***
     * 向客戶端發(fā)送信息
     */
    @Scheduled(fixedRate = 2000)
    public void sendMsg() throws IOException {
        for (String key : map.keySet())
        {
            map.get(key).getBasicRemote().sendText("你好，你好");
        }
    }
}

2.1.3 WebSocketConfig

package com.heima;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration
public class WebSocketConfig {


    @Bean
 public ServerEndpointExporter serverEndpointExporter()
 {
     return new ServerEndpointExporter();
 }



}

2.1.3 前端測試代碼

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>wsClient</title>
</head>
<body>

<script>
     // 創(chuàng)建websocket
    let ws = new WebSocket("ws://localhost:8080/myWs")
     //向服務器發(fā)送hello
    ws.onopen=function (){
        ws.send("hello")
    }
    //監(jiān)聽數(shù)據(jù)ws://localhost:8080/myWs
    ws.onmessage=function (message){
        console.log(message.data)
    }
</script>
</body>
</html>

2.1.4測試結果

2.1.4.1 當打開瀏覽器時

2.1.4.2 當關閉瀏覽器時

2.1.4.3 當刷新瀏覽器的時候

2.基于spring提供的類和接口刷新websocket服務器端

2.1：HttpSessionHandShakeInter 握手攔截器

package com.spring;

import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;

import java.util.Map;

/***
 * 握手攔截器
 */
@Component
@Slf4j
public class MyWsInterceptor extends HttpSessionHandshakeInterceptor {


    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
        log.info(request.getRemoteAddress().toString()+"開始握手");
        return super.beforeHandshake(request, response, wsHandler, attributes);
    }

    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception ex) {
         log.info(request.getRemoteAddress().toString()+"完成握手");
        super.afterHandshake(request, response, wsHandler, ex);
    }
}

2.2 MyWsHandler 主處理程序

sessionbean封裝類

import lombok.AllArgsConstructor;
import lombok.Data;
import org.springframework.web.socket.WebSocketSession;

@Data
@AllArgsConstructor
public class SessionBean {
    private WebSocketSession webSocketSession;
    private Integer clientId;
}

主處理程序

package com.spring;

import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.web.servlet.server.Session;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.AbstractWebSocketHandler;

import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

/***
 * webSocket 主處理程序
 */
@Component
@Slf4j
@EnableScheduling
public class MyWsHandler extends AbstractWebSocketHandler {
    //map有并發(fā)線程問題  所以用ConcurrentHashMap
    private static Map<String, SessionBean> map ;
    //id有并發(fā)問題 所以用Integer的安全類型
    private static AtomicInteger clientIdMaker;
    static {

        map = new ConcurrentHashMap<>();
        clientIdMaker=new AtomicInteger(0);
    }
    //連接建立
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        super.afterConnectionEstablished(session);
        //將session 進一步封裝  id采用的是自增
        SessionBean sessionBean = new SessionBean(session, clientIdMaker.getAndIncrement());
        map.put(session.getId(),sessionBean);
        log.info(map.get(session.getId()).getClientId()+"建立了連接");
    }

    //收到消息
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        super.handleTextMessage(session, message);
        log.info(map.get(session.getId()).getClientId()+":"+message.getPayload());
    }

    //傳輸異常
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        super.handleTransportError(session, exception);

        if (session.isOpen())
        {
            session.close();
        }
        map.remove(session.getId());

    }
    //連接關閉
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        super.afterConnectionClosed(session, status);
        log.info(map.get(session.getId()).getClientId()+"關閉連接");
    }

    /***
     * 向客戶端發(fā)送信息
     */
    @Scheduled(fixedRate = 2000)
    public void sendMsg() throws IOException {
        for (String key : map.keySet())
        {
            map.get(key).getWebSocketSession().sendMessage(new TextMessage("hello," +
                    "spring socket"));
        }
    }
}

2.3 WebSocketConfigurer 注冊攔截器和主處理程序以及監(jiān)聽路徑

package com.spring;

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

import javax.annotation.Resource;

@Configuration
@EnableWebSocket
public class MyWsConfig implements WebSocketConfigurer {
    @Resource
    private MyWsHandler wsHandler;
    @Resource
    private MyWsInterceptor wsInterceptor;
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(wsHandler,"/myWs1").addInterceptors(wsInterceptor).setAllowedOriginPatterns("*");
    }
}

2.4 前端測試

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>wsClient</title>
</head>
<body>

<script>
     // 創(chuàng)建websocket
    let ws = new WebSocket("ws://localhost:8080/myWs1")
     //向服務器發(fā)送hello
    ws.onopen=function (){
        ws.send("hello")
    }
    //監(jiān)聽數(shù)據(jù)ws://localhost:8080/myWs
    ws.onmessage=function (message){
        console.log(message.data)
    }
</script>
</body>
</html>

總結

到此這篇關于java注解實現(xiàn)websocket服務兩種方式的文章就介紹到這了,更多相關java websocket服務實現(xiàn)內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

$淺談Java轉義符\\|$
淺談Java轉義符\\|
java中\(zhòng)需要用\\來表示吧這個你應該知道，而split中傳入的參數(shù)是什么呢他并不是普通的字符串你可以查一下api文檔它要求傳入的是正則表達式而正則表達式也需要這個\所以在這里有這個\\
2015-06-06
Java聊天室之解決連接超時問題
這篇文章主要為大家詳細介紹了Java簡易聊天室之解決連接超時問題的方法，文中的示例代碼講解詳細，具有一定的借鑒價值，需要的可以了解一下
2022-10-10
spring cloud中微服務之間的調用以及eureka的自我保護機制詳解
這篇文章主要介紹了spring cloud中微服務之間的調用以及eureka的自我保護機制詳解,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-07-07
IDEA?2022.1.4用前注意事項
這篇文章主要介紹了IDEA?2022.1.4用前注意事項,本文通過圖文并茂的形式給大家介紹的非常詳細,感興趣的朋友跟隨小編一起看看吧
2024-08-08
面試時必問的JVM運行時數(shù)據(jù)區(qū)詳解
這篇文章主要介紹了JVM運行時數(shù)據(jù)區(qū)原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2021-08-08
Linux下java環(huán)境配置圖文方法
這篇文章主要介紹了Linux下java環(huán)境配置圖文方法,需要的朋友可以參考下
2023-06-06
java實現(xiàn)多文件上傳至本地服務器功能
這篇文章主要為大家詳細介紹了java實現(xiàn)多文件上傳至本地服務器功能，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2018-01-01
配置javaw.exe雙擊運行jar包方式
這篇文章主要介紹了配置javaw.exe雙擊運行jar包方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2024-01-01
java持久層框架mybatis防止sql注入的方法
下面小編就為大家?guī)硪黄猨ava持久層框架mybatis防止sql注入的方法。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2016-10-10
Springboot下使用Redis管道(pipeline)進行批量操作
本文主要介紹了Spring?boot?下使用Redis管道（pipeline）進行批量操作，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2023-05-05