整合微信小程序和Spring Boot以实现WebSocket功能，从而发送提醒消息，涉及多个步骤。以下是详细的整合过程和关键代码，分为以下几部分：
- 准备工作
- Spring Boot WebSocket配置
- WebSocket消息处理
- 微信小程序客户端实现
- 测试和调试
1. 准备工作
在开始之前，需要确保你已经有一个基本的Spring Boot项目，并且你已经安装了微信开发者工具，用于调试和测试微信小程序。
2. Spring Boot WebSocket配置
首先，确保你的Spring Boot项目中已经引入了WebSocket依赖。在pom.xml中添加以下依赖：
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency>
接下来，创建一个配置类来启用WebSocket：
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; @Configuration @EnableWebSocket public class WebSocketConfig implements WebSocketConfigurer { @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(new MyWebSocketHandler(), "/websocket") .setAllowedOrigins("*"); // 允许所有来源 } }
3. WebSocket消息处理
创建一个WebSocket处理类：
import org.springframework.web.socket.CloseStatus; import org.springframework.web.socket.TextMessage; import org.springframework.web.socket.WebSocketSession; import org.springframework.web.socket.handler.TextWebSocketHandler; import java.util.HashSet; import java.util.Set; public class MyWebSocketHandler extends TextWebSocketHandler { private static Set<WebSocketSession> sessions = new HashSet<>(); @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { sessions.add(session); } @Override public void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { // 处理来自客户端的消息 String payload = message.getPayload(); // 可以在这里根据payload的内容做出不同的处理 } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { sessions.remove(session); } public static void sendMessageToAll(String message) throws Exception { for (WebSocketSession session : sessions) { if (session.isOpen()) { session.sendMessage(new TextMessage(message)); } } } }
创建一个Controller来触发消息发送：
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; @RestController @RequestMapping("/api") public class WebSocketController { @GetMapping("/send") public String sendMessage(@RequestParam String message) { try { MyWebSocketHandler.sendMessageToAll(message); return "Message sent!"; } catch (Exception e) { return "Failed to send message: " + e.getMessage(); } } }
4. 微信小程序客户端实现
在微信小程序中，创建一个WebSocket连接，并处理消息。
在app.js中：
App({ globalData: { websocket: null, }, onLaunch: function () { this.globalData.websocket = wx.connectSocket({ url: 'ws://你的服务器地址/websocket', success: () => { console.log('WebSocket连接成功'); }, fail: (error) => { console.error('WebSocket连接失败', error); } }); this.globalData.websocket.onMessage((message) => { console.log('收到消息:', message.data); // 在这里处理接收到的消息，例如显示通知 }); this.globalData.websocket.onClose(() => { console.log('WebSocket已关闭'); }); } });
在页面中，你可以通过app.globalData.websocket.send发送消息：
const app = getApp(); Page({ data: { message: '' }, sendMessage: function () { if (app.globalData.websocket && this.data.message) { app.globalData.websocket.send({ data: this.data.message, success: () => { console.log('消息发送成功'); }, fail: (error) => { console.error('消息发送失败', error); } }); } }, bindMessageInput: function (e) { this.setData({ message: e.detail.value }); } });
5. 测试和调试
使用微信开发者工具运行微信小程序。
在微信小程序中打开相应页面，输入消息并点击发送，观察是否成功发送消息。
使用浏览器访问Spring Boot提供的API端点，触发消息发送，观察微信小程序是否接收到消息。
通过上述步骤，你已经完成了微信小程序和Spring Boot的WebSocket整合，并实现了消息发送和接收的基本功能。如果有更复杂的业务需求，可以根据实际情况进一步扩展和优化代码。
