删除无用代码 WebSocket
This commit is contained in:
parent
e14962bc5f
commit
e44f1a301f
@ -1,25 +0,0 @@
|
||||
package com.xkrs.websocket.config;
|
||||
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
|
||||
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
|
||||
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
|
||||
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
|
||||
|
||||
/**
|
||||
* websocket广播配置
|
||||
* @author xkrs
|
||||
*/
|
||||
@Configuration
|
||||
@EnableWebSocketMessageBroker
|
||||
public class WebSocketBrokerConfig implements WebSocketMessageBrokerConfigurer {
|
||||
@Override
|
||||
public void registerStompEndpoints(StompEndpointRegistry registry) {
|
||||
registry.addEndpoint("/endPointClientServer").withSockJS();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void configureMessageBroker(MessageBrokerRegistry registry) {
|
||||
registry.enableSimpleBroker("/ws");
|
||||
}
|
||||
}
|
||||
@ -1,16 +0,0 @@
|
||||
package com.xkrs.websocket.config;
|
||||
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
|
||||
|
||||
/**
|
||||
* @author xkrs
|
||||
*/
|
||||
@Configuration
|
||||
public class WebSocketConfig {
|
||||
@Bean
|
||||
public ServerEndpointExporter serverEndpointExporter(){
|
||||
return new ServerEndpointExporter();
|
||||
}
|
||||
}
|
||||
@ -1,18 +0,0 @@
|
||||
package com.xkrs.websocket.controller;
|
||||
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.RequestMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
/**
|
||||
* @author xkrs
|
||||
*/
|
||||
@RestController
|
||||
@RequestMapping("/test")
|
||||
public class TestController {
|
||||
|
||||
@GetMapping("/hello")
|
||||
public String sayHello(){
|
||||
return "asdasd";
|
||||
}
|
||||
}
|
||||
@ -1,24 +0,0 @@
|
||||
package com.xkrs.websocket.controller.websocket;
|
||||
|
||||
import com.xkrs.websocket.service.WebSocketServer;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
/**
|
||||
* @author xkrs
|
||||
*/
|
||||
@RestController
|
||||
@RequestMapping("/api/socket")
|
||||
public class WebSocketController {
|
||||
|
||||
@GetMapping("/sendAll")
|
||||
public String sendAll(@RequestParam String msg) {
|
||||
WebSocketServer.broadInfo(msg);
|
||||
return "success";
|
||||
}
|
||||
|
||||
@RequestMapping(value = "/sendOne", method = RequestMethod.GET)
|
||||
public String sendOne(@RequestParam String msg, @RequestParam String session) {
|
||||
WebSocketServer.sendMsg(msg, session);
|
||||
return "success";
|
||||
}
|
||||
}
|
||||
@ -1,21 +0,0 @@
|
||||
package com.xkrs.websocket.controller.websocketbroker;
|
||||
|
||||
import com.xkrs.websocket.websocket.ClientServerMessage;
|
||||
import com.xkrs.websocket.websocket.ServerClientMessage;
|
||||
import org.springframework.messaging.handler.annotation.MessageMapping;
|
||||
import org.springframework.messaging.handler.annotation.SendTo;
|
||||
import org.springframework.stereotype.Controller;
|
||||
|
||||
/**
|
||||
* @author xkrs
|
||||
*/
|
||||
@Controller
|
||||
public class WebSocketBrokerController {
|
||||
|
||||
@MessageMapping("/hello")
|
||||
@SendTo("/ws/getResponse")
|
||||
public ServerClientMessage say(ClientServerMessage message) throws InterruptedException {
|
||||
Thread.sleep(3000);
|
||||
return new ServerClientMessage("hello," + message.getName() + "!");
|
||||
}
|
||||
}
|
||||
@ -1,128 +0,0 @@
|
||||
package com.xkrs.websocket.service;
|
||||
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import javax.annotation.PostConstruct;
|
||||
import javax.websocket.*;
|
||||
import javax.websocket.server.ServerEndpoint;
|
||||
import java.io.IOException;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.CopyOnWriteArraySet;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
|
||||
/**
|
||||
* @author xkrs
|
||||
*/
|
||||
@Component
|
||||
@ServerEndpoint(value = "/ws/asset")
|
||||
public class WebSocketServer {
|
||||
|
||||
@PostConstruct
|
||||
public void init() {
|
||||
System.out.println("websocketBroker loading......");
|
||||
}
|
||||
|
||||
private static Logger log = LoggerFactory.getLogger(WebSocketServer.class);
|
||||
private static final AtomicInteger ONLINE_COUNT = new AtomicInteger(0);
|
||||
private static CopyOnWriteArraySet<Session> sessionSet = new CopyOnWriteArraySet<>();
|
||||
|
||||
/**
|
||||
* 连接建立
|
||||
*/
|
||||
@OnOpen
|
||||
public void onOpen(Session session) {
|
||||
sessionSet.add(session);
|
||||
//在线加1
|
||||
int cnt = ONLINE_COUNT.incrementAndGet();
|
||||
//log.info("have a new connect:{}", cnt);
|
||||
//sendMsg(session, "connect success");
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接关闭
|
||||
*
|
||||
* @param session
|
||||
*/
|
||||
@OnClose
|
||||
public void onClose(Session session) {
|
||||
sessionSet.remove(session);
|
||||
//在线减1
|
||||
int cnt = ONLINE_COUNT.decrementAndGet();
|
||||
log.info("have a connect close:{}", cnt);
|
||||
}
|
||||
|
||||
/**
|
||||
* 收到客户端消息后调用的方法
|
||||
*
|
||||
* @param session
|
||||
* @param msg
|
||||
*/
|
||||
@OnMessage
|
||||
public void onMessage(String msg, Session session) {
|
||||
log.info("receive the msg:{}", msg);
|
||||
}
|
||||
|
||||
private static void sendMsg(Session session, String msg) {
|
||||
try {
|
||||
session.getBasicRemote().sendText(String.format(msg));
|
||||
} catch (IOException e) {
|
||||
log.error("发送信息出错：{}", e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
private static void sendDate(Session session, Map map) {
|
||||
try {
|
||||
session.getBasicRemote().sendObject(map);
|
||||
} catch (IOException | EncodeException e) {
|
||||
log.error("发送信息出错：{}", e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 指定session发送
|
||||
*/
|
||||
public static void sendMsg(String msg, String sessionId) {
|
||||
Session session = null;
|
||||
for (Session s : sessionSet) {
|
||||
if (s.getId().equals(sessionId)) {
|
||||
session = s;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (session != null) {
|
||||
sendMsg(session, msg);
|
||||
} else {
|
||||
log.warn("not find your session");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 群发消息
|
||||
*/
|
||||
public static void broadInfo(String msg) {
|
||||
for (Session s : sessionSet) {
|
||||
synchronized (sessionSet){
|
||||
if (s.isOpen()) {
|
||||
sendMsg(s, msg);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 群发map消息
|
||||
* @param map
|
||||
*/
|
||||
public static void broadInfoMap(Map map) {
|
||||
for (Session s : sessionSet) {
|
||||
if (s.isOpen()) {
|
||||
sendDate(s, map);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@ -1,18 +0,0 @@
|
||||
package com.xkrs.websocket.websocket;
|
||||
|
||||
/**
|
||||
* @author xkrs
|
||||
*/
|
||||
public class ClientServerMessage {
|
||||
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
public void setName(String name) {
|
||||
this.name = name;
|
||||
}
|
||||
|
||||
private String name;
|
||||
|
||||
}
|
||||
@ -1,21 +0,0 @@
|
||||
package com.xkrs.websocket.websocket;
|
||||
|
||||
/**
|
||||
* @author xkrs
|
||||
*/
|
||||
public class ServerClientMessage {
|
||||
|
||||
public String getResponseMsg() {
|
||||
return responseMsg;
|
||||
}
|
||||
|
||||
public void setResponseMsg(String responseMsg) {
|
||||
this.responseMsg = responseMsg;
|
||||
}
|
||||
|
||||
private String responseMsg;
|
||||
|
||||
public ServerClientMessage(String responseMsg) {
|
||||
this.responseMsg = responseMsg;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user