From dda1971f80a8eb243e8efe44a7e2a0c34efc00dc Mon Sep 17 00:00:00 2001 From: xiaoCJ <406612557@qq.com> Date: Thu, 27 Jul 2023 16:40:10 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9Ewebsocket=E9=85=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../config/WebSocketConfig.java | 24 --------- .../config/security/WebSecurityConfig.java | 1 + .../config/websocket/WebSocket.java | 53 +++++++++++++++++++ .../config/websocket/WebSocketConfig.java | 18 +++++++ .../service/WainingService.java | 17 +++--- 5 files changed, 81 insertions(+), 32 deletions(-) delete mode 100644 src/main/java/com/sztzjy/forex/trading_trading/config/WebSocketConfig.java create mode 100644 src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocket.java create mode 100644 src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocketConfig.java diff --git a/src/main/java/com/sztzjy/forex/trading_trading/config/WebSocketConfig.java b/src/main/java/com/sztzjy/forex/trading_trading/config/WebSocketConfig.java deleted file mode 100644 index 25b4454..0000000 --- a/src/main/java/com/sztzjy/forex/trading_trading/config/WebSocketConfig.java +++ /dev/null @@ -1,24 +0,0 @@ -package com.sztzjy.forex.trading_trading.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; - -@Configuration -@EnableWebSocketMessageBroker -public class WebSocketConfig implements WebSocketMessageBrokerConfigurer { - - // 配置消息代理 - @Override - public void configureMessageBroker(MessageBrokerRegistry registry) { - registry.enableSimpleBroker("/topic"); - registry.setApplicationDestinationPrefixes("/app"); - } - - @Override - public void registerStompEndpoints(StompEndpointRegistry registry) { - registry.addEndpoint("/ws").withSockJS(); // 配置 WebSocket 端点 - } -} \ No newline at end of file diff --git a/src/main/java/com/sztzjy/forex/trading_trading/config/security/WebSecurityConfig.java b/src/main/java/com/sztzjy/forex/trading_trading/config/security/WebSecurityConfig.java index 0b39cd8..406c890 100644 --- a/src/main/java/com/sztzjy/forex/trading_trading/config/security/WebSecurityConfig.java +++ b/src/main/java/com/sztzjy/forex/trading_trading/config/security/WebSecurityConfig.java @@ -80,6 +80,7 @@ public class WebSecurityConfig extends WebSecurityConfigurerAdapter { .antMatchers("/v2/**").permitAll() .antMatchers("/test/**").permitAll() .antMatchers("/druid/**").permitAll() + .antMatchers("/webSocket/**").permitAll() .antMatchers(anonymousUrls.toArray(new String[]{})).permitAll() .anyRequest().authenticated(); } diff --git a/src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocket.java b/src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocket.java new file mode 100644 index 0000000..98b2c43 --- /dev/null +++ b/src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocket.java @@ -0,0 +1,53 @@ +package com.sztzjy.forex.trading_trading.config.websocket; + +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Component; + +import javax.websocket.*; +import javax.websocket.server.PathParam; +import javax.websocket.server.ServerEndpoint; +import java.util.Map; +import java.util.concurrent.ConcurrentHashMap; + +@ServerEndpoint("/webSocket/{username}") +@Slf4j +@Component +public class WebSocket { + + private static int onlineCount = 0; + private static Map clients = new ConcurrentHashMap(); + private Session session; + private String username; + + @OnOpen + public void onOpen(@PathParam("username") String username, Session session) { + this.username = username; + this.session = session; + WebSocket.onlineCount++; + clients.put(username, this); + } + + @OnClose + public void onClose() { + clients.remove(username); + WebSocket.onlineCount--; + } + + @OnMessage + public void onMessage(String message) {} + + @OnError + public void onError(Session session, Throwable throwable) { + log.error("WebSocket发生错误：" + throwable.getMessage()); + } + + public static void sendMessage(String message) { + // 向所有连接websocket的客户端发送消息 + // 可以修改为对某个客户端发消息 + for (WebSocket item : clients.values()) { + item.session.getAsyncRemote().sendText(message); + } + } + +} + diff --git a/src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocketConfig.java b/src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocketConfig.java new file mode 100644 index 0000000..c965b28 --- /dev/null +++ b/src/main/java/com/sztzjy/forex/trading_trading/config/websocket/WebSocketConfig.java @@ -0,0 +1,18 @@ +package com.sztzjy.forex.trading_trading.config.websocket; + +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.web.socket.config.annotation.EnableWebSocket; +import org.springframework.web.socket.server.standard.ServerEndpointExporter; + +@Configuration +@EnableWebSocket +public class WebSocketConfig { + + @Bean + public ServerEndpointExporter serverEndpointExporter() { + return new ServerEndpointExporter(); + } + +} + diff --git a/src/main/java/com/sztzjy/forex/trading_trading/service/WainingService.java b/src/main/java/com/sztzjy/forex/trading_trading/service/WainingService.java index 717a874..1da854c 100644 --- a/src/main/java/com/sztzjy/forex/trading_trading/service/WainingService.java +++ b/src/main/java/com/sztzjy/forex/trading_trading/service/WainingService.java @@ -3,15 +3,13 @@ package com.sztzjy.forex.trading_trading.service; import cn.hutool.core.util.IdUtil; import com.github.pagehelper.PageHelper; import com.github.pagehelper.PageInfo; +import com.sztzjy.forex.trading_trading.config.websocket.WebSocket; import com.sztzjy.forex.trading_trading.entity.*; import com.sztzjy.forex.trading_trading.mappers.*; import com.sztzjy.forex.trading_trading.util.ForexMarketDateUtil; import com.sztzjy.forex.trading_trading.util.RedisUtil; -import com.sztzjy.forex.trading_trading.util.file.IFileUtil; -import org.apache.poi.ss.formula.functions.T; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; -import org.springframework.messaging.simp.SimpMessagingTemplate; import org.springframework.scheduling.annotation.Scheduled; import org.springframework.stereotype.Service; @@ -29,8 +27,6 @@ public class WainingService { @Autowired private MarketWarningMapper marketWarningMapper; @Autowired - private SimpMessagingTemplate simpMessagingTemplate; - @Autowired private MarketWarningRecordMapper marketWarningRecordMapper; @Autowired private ForexMarketDateUtil forexMarketDateUtil; @@ -46,6 +42,8 @@ public class WainingService { private TransactionNumberWarningMapper transactionNumberWarningMapper; @Autowired private TransactionNumberWarningRecordMapper transactionNumberWarningRecordMapper; + @Autowired + private WebSocket webSocket; //保证金比较接口 public String compareMarginLevels(String memberId, String trainingId) { @@ -70,7 +68,7 @@ public class WainingService { marginWarningRecord.setWarningTime(current); marginWarningRecord.setEarlyWarningLevel(memberLevel); marginWarningRecordMapper.updateByPrimaryKeySelective(marginWarningRecord); - simpMessagingTemplate.convertAndSend("/topic/margin-change", message); + WebSocket.sendMessage(message); } } } @@ -101,6 +99,8 @@ public class WainingService { BeanUtils.copyProperties(member, marginWarning); marginWarning.setStatus(0); marginWarning.setId(IdUtil.simpleUUID()); + marginWarning.setMarginLevel(50.12); + marginWarning.setNetValue(50.2131); marginWarningMapper.insert(marginWarning); } return marginWarning; @@ -195,7 +195,7 @@ public class WainingService { if (direction.equals("买")) { // 3.拿redis的价位和新增的高价低价做比较，==的时候就弹出提示，并且存入行情预警记录表 if (buyPic.equals(upperBreakPrice) || buyPic.equals(lowerBreakPrice)) { - simpMessagingTemplate.convertAndSend("/topic/alerts", message); + WebSocket.sendMessage(message); marketWarningRecord.setWarningPrice(buyPic); marketWarningRecord.setWarningTime((new Timestamp(new Date().getTime()))); marketWarningRecordMapper.updateByPrimaryKeySelective(marketWarningRecord); @@ -203,7 +203,7 @@ public class WainingService { } } else { if (sellPicValue.equals(upperBreakPrice) || sellPicValue.equals(lowerBreakPrice)) { - simpMessagingTemplate.convertAndSend("/topic/alerts", message); + WebSocket.sendMessage(message); marketWarningRecord.setWarningPrice(sellPicValue); marketWarningRecord.setWarningTime((new Timestamp(new Date().getTime()))); marketWarningRecordMapper.updateByPrimaryKeySelective(marketWarningRecord); @@ -319,6 +319,7 @@ public class WainingService { // 如果当前交易次数在到达用户设置的预警时间时还是小于，那么就返回消息给前端提示 tnwr.setWarningingCount(openingTrades); tnwr.setWarningTime(new Timestamp(new Date().getTime())); + WebSocket.sendMessage(message); transactionNumberWarningRecordMapper.updateByPrimaryKeySelective(tnwr); } }