From 1add5ac60ec38dc1a663fb6015c052b6f38161a4 Mon Sep 17 00:00:00 2001
From: republicline <1464474399@qq.com>
Date: Tue, 8 Oct 2024 11:50:12 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20v2=E6=9C=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 upms/upms-biz/pom.xml                         |  5 +++
 .../com/rax/admin/timmer/LogTimerTask.java    |  1 -
 .../java/com/rax/admin/timmer/MongoTimer.java | 31 +++++++++++++++++++
 .../java/com/rax/admin/timmer/MySqlTimer.java | 10 ++++++
 .../v2/interceptor/WebSocketInterceptors.java |  7 +++++
 5 files changed, 53 insertions(+), 1 deletion(-)
 create mode 100644 upms/upms-biz/src/main/java/com/rax/admin/timmer/MongoTimer.java
 create mode 100644 upms/upms-biz/src/main/java/com/rax/admin/timmer/MySqlTimer.java

diff --git a/upms/upms-biz/pom.xml b/upms/upms-biz/pom.xml
index 3316ef9..eb63a18 100644
--- a/upms/upms-biz/pom.xml
+++ b/upms/upms-biz/pom.xml
@@ -132,6 +132,11 @@
 			<groupId>cn.hutool</groupId>
 			<artifactId>hutool-core</artifactId>
 		</dependency>
+		<dependency>
+			<groupId>org.mongodb</groupId>
+			<artifactId>mongo-java-driver</artifactId>
+			<version>3.12.14</version>
+		</dependency>
 	</dependencies>
 
 	<build>
diff --git a/upms/upms-biz/src/main/java/com/rax/admin/timmer/LogTimerTask.java b/upms/upms-biz/src/main/java/com/rax/admin/timmer/LogTimerTask.java
index bcbf627..098eaef 100644
--- a/upms/upms-biz/src/main/java/com/rax/admin/timmer/LogTimerTask.java
+++ b/upms/upms-biz/src/main/java/com/rax/admin/timmer/LogTimerTask.java
@@ -23,7 +23,6 @@ import java.util.List;
  */
 @Slf4j
 @Component
-@Service
 public class LogTimerTask {
 
 	@Autowired
diff --git a/upms/upms-biz/src/main/java/com/rax/admin/timmer/MongoTimer.java b/upms/upms-biz/src/main/java/com/rax/admin/timmer/MongoTimer.java
new file mode 100644
index 0000000..811881f
--- /dev/null
+++ b/upms/upms-biz/src/main/java/com/rax/admin/timmer/MongoTimer.java
@@ -0,0 +1,31 @@
+package com.rax.admin.timmer;
+
+import com.mongodb.client.MongoClient;
+import com.mongodb.client.MongoClients;
+import com.mongodb.client.MongoDatabase;
+import com.mongodb.client.MongoIterable;
+import lombok.extern.slf4j.Slf4j;
+import org.aspectj.weaver.ast.Var;
+import org.springframework.stereotype.Component;
+
+/**
+ * @project_name: rax-remote-v2
+ * @time: 2024/10/8 10:57
+ * @author: republicline
+ * @description: mongoDB定时备份
+ */
+@Slf4j
+@Component
+public class MongoTimer {
+	String connectionString = "mongodb://localhost:27017";
+	MongoClient mongoClient = MongoClients.create(connectionString);
+
+
+
+	public void backup() {
+		MongoIterable<String> DbNames = mongoClient.listDatabaseNames();
+		for (String dbName : DbNames) {
+
+		}
+	}
+}
diff --git a/upms/upms-biz/src/main/java/com/rax/admin/timmer/MySqlTimer.java b/upms/upms-biz/src/main/java/com/rax/admin/timmer/MySqlTimer.java
new file mode 100644
index 0000000..78891d9
--- /dev/null
+++ b/upms/upms-biz/src/main/java/com/rax/admin/timmer/MySqlTimer.java
@@ -0,0 +1,10 @@
+package com.rax.admin.timmer;
+
+/**
+ * @project_name: rax-remote-v2
+ * @time: 2024/10/8 10:57
+ * @author: republicline
+ * @description: mysql定时备份
+ */
+public class MySqlTimer {
+}
diff --git a/upms/upms-biz/src/main/java/com/rax/vital/v2/interceptor/WebSocketInterceptors.java b/upms/upms-biz/src/main/java/com/rax/vital/v2/interceptor/WebSocketInterceptors.java
index 96f18a9..396d0bb 100644
--- a/upms/upms-biz/src/main/java/com/rax/vital/v2/interceptor/WebSocketInterceptors.java
+++ b/upms/upms-biz/src/main/java/com/rax/vital/v2/interceptor/WebSocketInterceptors.java
@@ -44,6 +44,13 @@ public class WebSocketInterceptors implements HandshakeInterceptor {
 	@Override
 	public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
 		String query = request.getURI().getQuery();
+		if (query == null) {
+			return false;
+		}
+		Map<String, String> params = GetHttpParamUtil.getParams(query);
+		if (!params.containsKey("token")) {
+			return false;
+		}
 		String token = GetHttpParamUtil.getParam(query, "token");
 		OAuth2Authorization authorization = authorizationService.findByToken(token, OAuth2TokenType.ACCESS_TOKEN);
 		if (authorization == null) {