springboot 微信小程序整合websocket，实现发送提醒消息

整合微信小程序和Spring Boot以实现WebSocket功能，从而发送提醒消息，涉及多个步骤。以下是详细的整合过程和关键代码，分为以下几部分：

1. 准备工作
2. Spring Boot WebSocket配置
3. WebSocket消息处理
4. 微信小程序客户端实现
5. 测试和调试

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. 测试和调试

1. 启动Spring Boot应用程序。
2. 使用微信开发者工具运行微信小程序。
3. 在微信小程序中打开相应页面，输入消息并点击发送，观察是否成功发送消息。
4. 使用浏览器访问Spring Boot提供的API端点，触发消息发送，观察微信小程序是否接收到消息。

通过上述步骤，你已经完成了微信小程序和Spring Boot的WebSocket整合，并实现了消息发送和接收的基本功能。如果有更复杂的业务需求，可以根据实际情况进一步扩展和优化代码。