diff --git a/upms/upms-biz/src/main/java/com/rax/RaxAdminApplication.java b/upms/upms-biz/src/main/java/com/rax/RaxAdminApplication.java index 4add956..bb2295c 100644 --- a/upms/upms-biz/src/main/java/com/rax/RaxAdminApplication.java +++ b/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 +@SpringBootApplication(exclude = MongoAutoConfiguration.class) +//@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 { diff --git a/upms/upms-biz/src/main/java/com/rax/vital/v1/handler/ChatHandler.java b/upms/upms-biz/src/main/java/com/rax/vital/v1/handler/ChatHandler.java index ef7bc68..b610457 100644 --- a/upms/upms-biz/src/main/java/com/rax/vital/v1/handler/ChatHandler.java +++ b/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.Executors; -import java.util.concurrent.ScheduledExecutorService; -import java.util.concurrent.TimeUnit; +import java.util.concurrent.*; 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 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 webSocketSessions = dbNameSessionList.get(dbName); webSocketSessions.add(session); + // 创建数据库表 + initTChatTable(dbName, session); // ArrayList 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 { - String patientName = jsonObject.getString("patientName"); + } else if ("msg".equals(jsonObject.getString("msgType")) || + "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 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(); + } + } + + }