springbot websocket服务端、客户端实现

1、依赖

		<!--websocket 服务端-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
        </dependency>
        <!--websocket 客户端-->
        <dependency>
            <groupId>org.java-websocket</groupId>
            <artifactId>Java-WebSocket</artifactId>
            <version>1.3.5</version>
        </dependency>

2、服务端配置

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

/**
 * @Author zengqifeng
 * @CreateTime 2021/8/13 14:28
 * @Description
 */
@Component
public class WebSocketConfig {

    /**
     * springboot内置Tomcat 需要配置该bean
     * 作用是会自动扫描@ServerEndpoint注解类作为一个websocket服务端 并纳入容器管理
     * @return
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}

3、服务端业务处理逻辑

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

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

/**
 * @Author zengqifeng
 * @CreateTime 2021/8/13 14:35
 * @Description @ServerEndpoint注解标注一个类为websocket服务端
 */
@Slf4j
@Component
@ServerEndpoint(value = "/websocket")
public class WebSocketServer {

    /**
     * 线程安全容器、用于保存客户端session
     */
    private static ConcurrentHashMap<String, Session> clients = new ConcurrentHashMap<>();


    /**
     * 客户端连接
     * @param session
     */
    @OnOpen
    public void onOpen(Session session) {
        log.info("id = {}", session.getId());
        log.info("uri host = {}", session.getRequestURI().getHost());
        log.info("uri port = {}", session.getRequestURI().getPort());
        log.info("uri = {}", session.getRequestURI().toString());
        clients.put(session.getId(), session);
        log.info("客户端连接id= {},当前客户端数 = {}", session.getId(), clients.size());
    }

    /**
     * 客户端关闭
     * @param session
     */
    @OnClose
    public void onClose(Session session) {
        clients.remove(session.getId());
        log.info("客户端断连id= {}, 当前客户端数 = {}", session.getId(), clients.size());
    }

    /**
     * 出错
     * @param session
     * @param throwable
     */
    @OnError
    public void onError(Session session, Throwable throwable) {

    }

    /**
     * 收到消息
     * @param message
     * @param session
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        log.info("接收客户端id= {}发来的消息 = {}", session.getId(), message);
    }


    /**
     * 点对点消息
     * @param session
     * @param message
     */
    public void sendDirectMessage(Session session, String message) {
        log.info("向客户端id = {}发送消息 = {}", session.getId(), message);
        session.getAsyncRemote().sendText(message);
    }

    /**
     * 群发消息
     * @param message
     */
    public void sendFanoutMessage(String message) {
        log.info("群发消息 = {}", message);
        clients.values().forEach(session -> session.getAsyncRemote().sendText(message));
    }

    /**
     * 群发消息  排除指定session
     * @param message
     * @param session
     */
    public void sendFanoutMessage(String message, Session session) {
        log.info("群发消息、排除指定客户,消息内容 = {}", session);
        for (Map.Entry<String, Session> entry : clients.entrySet()) {
            if (session == entry.getValue()) {
                continue;
            }
            session.getAsyncRemote().sendText(message);
        }
    }

    /**
     * 获取所有在线客户
     * @return
     */
    public static ConcurrentHashMap getClients() {
        return clients;
    }
}

4、客户端配置

import lombok.extern.slf4j.Slf4j;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.drafts.Draft_6455;
import org.java_websocket.handshake.ServerHandshake;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import java.net.URI;

/**
 * @Author zengqifeng
 * @CreateTime 2021/8/13 16:11
 * @Description   创建客户端bean备用
 */
@Slf4j
@Component
public class WebSocketClientConfig {

        private static final String WEBSOCKET_URI = "ws://localhost:8080/websocket";

        @Bean(name = "client1")
        public WebSocketClient webSocketClient1() {
                try {
                        WebSocketClient webSocketClient = new WebSocketClient(new URI(WEBSOCKET_URI), new Draft_6455()) {
                                @Override
                                public void onOpen(ServerHandshake serverHandshake) {
                                        log.info("client1-客户端连接成功");
                                }

                                @Override
                                public void onMessage(String message) {
                                        log.info("client1-客户端收到消息:{}", message);
                                }

                                @Override
                                public void onClose(int i, String s, boolean b) {
                                        log.info("client1-关闭");
                                }

                                @Override
                                public void onError(Exception e) {
                                        log.info("client1-错误:{}", e);
                                }
                        };
                        webSocketClient.connect();
                        return webSocketClient;
                }catch (Exception e) {
                        log.info("创建websocket bean 异常,异常信息 = {}", e);
                }
                return  null;
        }


        @Bean(name = "client2")
        public WebSocketClient webSocketClient2() {
                try {
                        WebSocketClient webSocketClient = new WebSocketClient(new URI(WEBSOCKET_URI), new Draft_6455()) {
                                @Override
                                public void onOpen(ServerHandshake serverHandshake) {
                                        log.info("client2-客户端连接成功");
                                }

                                @Override
                                public void onMessage(String message) {
                                        log.info("client2-客户端收到消息:{}", message);
                                }

                                @Override
                                public void onClose(int i, String s, boolean b) {
                                        log.info("client2-关闭");
                                }

                                @Override
                                public void onError(Exception e) {
                                        log.info("client2-错误:{}", e);
                                }
                        };
                        webSocketClient.connect();
                        return webSocketClient;
                }catch (Exception e) {
                        log.info("创建websocket bean 异常,异常信息 = {}", e);
                }
                return  null;
        }
}

5、客户端业务逻辑

import com.example.demo.websocket.server.WebSocketServer;
import org.java_websocket.client.WebSocketClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import javax.websocket.Session;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Author zengqifeng
 * @CreateTime 2021/8/13 16:30
 * @Description
 */
@Service
public class WebSocketClientService {
    @Resource(name = "client1")
    private WebSocketClient webSocketClient1;

    @Resource(name = "client2")
    private WebSocketClient webSocketClient2;

    @Autowired
    private WebSocketServer webSocketServer;



    public void websocketServerSendMessage(String message) {
        webSocketServer.sendFanoutMessage("群发消息" + message);
    }

    public void websocketServerSendSignalMessage(String message) {
        ConcurrentHashMap clients = WebSocketServer.getClients();
        Session session = (Session) clients.get("0");
        webSocketServer.sendDirectMessage(session, "单发消息");
    }

    public void send1Message(String message) {
        webSocketClient1.send(message);
    }

    public void send2Message(String message) {
        webSocketClient2.send(message);
    }
}

6、controller测试

import com.example.demo.websocket.client.WebSocketClientService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

/**
 * <p>
 * websocket 前端控制器
 * </p>
 *
 * @author 曾奇峰
 * @since 2021-06-02
 */
@Api(tags = "websocket")
@RestController
@RequestMapping("/websocket")
public class WebSocketController {

    @Autowired
    private WebSocketClientService webSocketClientService;


    @ApiOperation("websocket客户端1发送消息测试")
    @GetMapping("/websocketClient1SendMessage")
    public Object websocketClient1SendMessage(@RequestParam String message) {
        webSocketClientService.send1Message(message);
        return "发送成功!";
    }

    @ApiOperation("websocket客户端2发送消息测试")
    @GetMapping("/websocketClient2SendMessage")
    public Object websocketClient2SendMessage(@RequestParam String message) {
        webSocketClientService.send2Message(message);
        return "发送成功!";
    }

    @ApiOperation("服务端单发消息")
    @GetMapping("/websocketServerSendSignalMessage")
    public Object websocketServerSendSignalMessage(@RequestParam String message) {
        webSocketClientService.websocketServerSendSignalMessage(message);
        return "单发成功!";
    }

    @ApiOperation("服务端群发消息")
    @GetMapping("/websocketServerSendMessage")
    public Object websocketServerSendMessage(@RequestParam String message) {
        webSocketClientService.websocketServerSendMessage(message);
        return "群发成功!";
    }


}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值