add: 一期代码

upms/upms-biz/src/main/java/com/rax/RaxAdminApplication.java

@@ -4,6 +4,7 @@ import com.rax.common.security.annotation.EnableRaxResourceServer;
 import com.rax.common.swagger.annotation.EnableRaxDoc;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.autoconfigure.mongo.MongoAutoConfiguration;
 import org.springframework.boot.builder.SpringApplicationBuilder;
 import org.springframework.boot.web.servlet.support.SpringBootServletInitializer;
 import org.springframework.context.annotation.ComponentScan;
@@ -15,12 +16,12 @@ import org.springframework.scheduling.annotation.EnableScheduling;
 @EnableRaxResourceServer
 @EnableScheduling // 开启定时任务 -> 清除日志相关内容
 // 一期下面解注, 二期注释掉
-//@SpringBootApplication(exclude = MongoAutoConfiguration.class)
+@SpringBootApplication(exclude = MongoAutoConfiguration.class)
-@SpringBootApplication
+//@SpringBootApplication
 @ComponentScan(
 		basePackages = {"com.rax"},
 		excludeFilters = {
-				@ComponentScan.Filter(type = FilterType.REGEX, pattern = "com.rax.vital.v1.*")
+				@ComponentScan.Filter(type = FilterType.REGEX, pattern = "com.rax.vital.v2.*")
 		})
 public class RaxAdminApplication extends SpringBootServletInitializer {
 

upms/upms-biz/src/main/java/com/rax/vital/v1/handler/ChatHandler.java

@@ -1,22 +1,29 @@
 package com.rax.vital.v1.handler;
 
+import cn.hutool.core.date.DateUtil;
+import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
+import com.rax.vital.common.datasource.CustomDataSource;
+import com.rax.vital.common.datasource.MySQLSource;
 import com.rax.vital.common.util.GetHttpParamUtil;
 import com.rax.vital.v1.medicine.service.ChatService;
 import jakarta.annotation.Resource;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.security.oauth2.server.authorization.OAuth2Authorization;
 import org.springframework.security.oauth2.server.authorization.OAuth2AuthorizationService;
 import org.springframework.security.oauth2.server.authorization.OAuth2TokenType;
+import org.springframework.util.StringUtils;
 import org.springframework.web.socket.*;
 
+import java.io.IOException;
 import java.net.URLDecoder;
+import java.sql.*;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
-import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.*;
-import java.util.concurrent.Executors;
-import java.util.concurrent.ScheduledExecutorService;
-import java.util.concurrent.TimeUnit;
 
 public class ChatHandler implements WebSocketHandler {
 	@Resource
@@ -25,6 +32,23 @@ public class ChatHandler implements WebSocketHandler {
 	@Resource
 	private ChatService chatService;
 
+	@Autowired
+	private RedisTemplate redisTemplate;
+
+	// mysql地址
+	@Value("${vital-sign.mysql.host}")
+	private String mysqlHost;
+
+	// mysql用户名
+	@Value("${vital-sign.mysql.username}")
+	private String mysqlUsername;
+
+	// mysql用户密码
+	@Value("${vital-sign.mysql.password}")
+	private String mysqlPassword;
+
+	private static final ExecutorService EXECUTOR_SERVICE = Executors.newFixedThreadPool(5);
+
 	private Map<String, ScheduledExecutorService> timerTaskMap = new ConcurrentHashMap();
 
 	// dbName -> sessionList
@@ -47,9 +71,9 @@ public class ChatHandler implements WebSocketHandler {
 		OAuth2Authorization authorization = authorizationService.findByToken(token, OAuth2TokenType.ACCESS_TOKEN);
 		if (authorization != null) {
 			String username = authorization.getPrincipalName();
+
 			String payload = (String) message.getPayload();
 			JSONObject jsonObject = JSONObject.parseObject(payload);
-
 			if ("heartbeat".equals(jsonObject.getString("msgType"))) {
 //				session.sendMessage(new TextMessage(""));
 			} else if ("init".equals(jsonObject.getString("msgType"))) {
@@ -71,6 +95,8 @@ public class ChatHandler implements WebSocketHandler {
 				}
 				List<WebSocketSession> webSocketSessions = dbNameSessionList.get(dbName);
 				webSocketSessions.add(session);
+				// 创建数据库表
+				initTChatTable(dbName, session);
 //				ArrayList<String> sessionIDList = new ArrayList<>();
 //				for (WebSocketSession webSocketSession : webSocketSessions) {
 //					if (webSocketSession.isOpen()) {
@@ -84,14 +110,31 @@ public class ChatHandler implements WebSocketHandler {
 //				redisTemplate.expire("chat_" + dbName, 60 * 60, TimeUnit.SECONDS);
 //				redisTemplate.opsForHash().putAll("chat_sessionId_" + dbName, sessionDbMap);
 //				redisTemplate.expire("chat_" + dbName, 60 * 60, TimeUnit.SECONDS);
-			} else {
+			} else if ("msg".equals(jsonObject.getString("msgType")) ||
-				String patientName = jsonObject.getString("patientName");
+					"audio".equals(jsonObject.getString("msgType"))) {
+				//String patientName = jsonObject.getString("patientName");
 				String dbName = 'a' + jsonObject.getString("idNum");
-				String date = jsonObject.getString("date");
+				//String date = jsonObject.getString("date");
+				String msgType = jsonObject.getString("msgType");
 				// 消息内容
-				String msg = jsonObject.getString("msg");
+				String content = jsonObject.getString("msg");
 				List<WebSocketSession> webSocketSessions = dbNameSessionList.get(dbName);
-				chatService.sendMessageMysql(username, patientName, dbName, date, session, msg, webSocketSessions);
+				// 转发消息
+				JSONObject param = new JSONObject();
+				param.put("msgType", msgType);
+				param.put("createUser", username);
+				param.put("createTime", DateUtil.now());
+				param.put("msg", content);
+				param.put("dbName", dbName);
+				for (WebSocketSession webSocketSession : webSocketSessions) {
+					webSocketSession.sendMessage(new TextMessage(param.toJSONString().getBytes()));
+				}
+
+				// 异步入库
+				EXECUTOR_SERVICE.execute(() -> saveChatMsg(param.toJSONString()));
+
+				// 之前的
+				// chatService.sendMessageMysql(username, patientName, dbName, msgType, session, msg, webSocketSessions);
 			}
 		} else {
 			String msgContent = "token无效,认证失败";
@@ -112,7 +155,7 @@ public class ChatHandler implements WebSocketHandler {
 	public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) {
 		System.out.println("CloseStatus: " + closeStatus.getReason() + closeStatus.getCode());
 		stopHeartbeat(session);
-		chatService.stopTask(session.getId());
+		//chatService.stopTask(session.getId());
 		stopMap(session);
 	}
 
@@ -132,7 +175,7 @@ public class ChatHandler implements WebSocketHandler {
 						session.sendMessage(new TextMessage(jsonObject.toJSONString().getBytes()));
 					} else {
 						stopHeartbeat(session);
-						chatService.stopTask(session.getId());
+						//chatService.stopTask(session.getId());
 						session.close();
 					}
 				} catch (Exception e) {
@@ -157,4 +200,110 @@ public class ChatHandler implements WebSocketHandler {
 		sessionDbMap.remove(session.getId());
 	}
 
+
+	// 初始化聊天表格
+	private void initTChatTable(String dbName, WebSocketSession session) {
+		try {
+
+			CustomDataSource source = new MySQLSource(mysqlHost, mysqlPassword, mysqlUsername, dbName);
+			source.open();
+			Connection connection = source.getConnection();
+			if (connection != null) {
+				String tableName = "t_chat";
+				Statement statement = connection.createStatement();
+				// 检查表是否存在的 SQL 语句
+				String checkTableSQL = "SELECT COUNT(*) FROM information_schema.tables " +
+						"WHERE table_schema = DATABASE() AND table_name = '" + tableName + "';";
+				ResultSet resultSet = statement.executeQuery(checkTableSQL);
+				if (resultSet.next() && resultSet.getInt(1) == 0) { // 表不存在
+					// 表不存在，执行创建操作
+					String createTableSQL = """
+							CREATE TABLE %s (
+							  `id` int NOT NULL AUTO_INCREMENT,
+							  `content` longblob,
+							  `create_time` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL,
+							  `create_user` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL,
+							  `msg_type` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL,
+							  `deleted` bit(1) NULL DEFAULT NULL,
+							  `revoked` bit(1) NULL DEFAULT NULL,
+							  PRIMARY KEY (`id`) USING BTREE
+							) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_0900_ai_ci ROW_FORMAT = Dynamic;
+							""".formatted(tableName);
+					statement.execute(createTableSQL);
+				}
+				source.close();
+			} else {
+				source.close();
+				JSONObject jsonObject = new JSONObject();
+				jsonObject.put("status", 1);
+				jsonObject.put("msg", "数据库未创建, 请重新连接");
+				jsonObject.put("msgType", "msg");
+				session.sendMessage(new TextMessage(jsonObject.toJSONString().getBytes()));
+			}
+
+		} catch (SQLException | IOException e) {
+			throw new RuntimeException(e);
+		}
+//	    创建数据库,查询历史
+//		DatabaseMetaData metaData = connection.getMetaData();
+//		ResultSet tablesx = metaData.getTables(null, null, tableName, new String[]{"TABLE"});
+//		String sql = "select content, create_time \"creatTime\", create_user \"createUser\", create_name \"createName\" from t_chat where deleted = 0 and revoked = 0 order by create_time asc ";
+//		if (tablesx.next()) {
+//
+//						Statement statement = connection.createStatement();
+//						ResultSet resultSet = statement.executeQuery(sql);
+//						while (resultSet.next()) {
+//							Map map = new HashMap();
+//							map.put("content", resultSet.getString("content"));
+//							map.put("creatTime", resultSet.getString("creatTime"));
+//							map.put("createUser", resultSet.getString("createUser"));
+//							map.put("createName", resultSet.getString("createName"));
+//							history.add(map);
+//						}
+//						JSONObject param = new JSONObject();
+//						param.put("history", history);
+//						param.put("msgType", "msg");
+////						session.sendMessage(new TextMessage(param.toJSONString().getBytes()));
+//						for (WebSocketSession webSocketSession : webSocketSessionList) {
+//							webSocketSession.sendMessage(new TextMessage(param.toJSONString().getBytes()));
+//						}
+//		}
+	}
+
+
+	// 保存聊天消息到数据库
+	private void saveChatMsg(String jsonStr) {
+		// 发消息, 入库
+		if (StringUtils.hasText(jsonStr)) {
+			JSONObject jsonObj = JSON.parseObject(jsonStr);
+
+			String msgType = jsonObj.getString("msgType");
+			Object content = jsonObj.get("content");
+			String username = jsonObj.getString("createUser");
+			String dbName = jsonObj.getString("dbName");
+
+			String now = DateUtil.now();
+
+			MySQLSource source = new MySQLSource(mysqlHost, mysqlPassword, mysqlUsername, dbName);
+			source.open();
+			Connection connection = source.getConnection();
+			PreparedStatement preparedStatement = null;
+			try {
+				preparedStatement = connection.prepareStatement("INSERT INTO t_chat (content, create_time, create_user, msg_type, deleted, revoked) VALUES (?, ?, ?, ?, ?, ?)");
+				preparedStatement.setObject(1, content);
+				preparedStatement.setString(2, now);
+				preparedStatement.setString(3, username);
+				preparedStatement.setString(4, msgType);
+				preparedStatement.setInt(5, 0);
+				preparedStatement.setInt(6, 0);
+				preparedStatement.execute();
+			} catch (SQLException e) {
+				source.close();
+				e.printStackTrace();
+			}
+			source.close();
+		}
+	}
+
+
 }