From bb04045712d55b0690bcd941711a764c8febbd10 Mon Sep 17 00:00:00 2001
From: ChangFeng <752492631@qq.com>
Date: Fri, 12 Apr 2024 11:43:03 +0800
Subject: [PATCH] =?UTF-8?q?2024.4.12=20=E5=8A=A0=E5=85=A5=E4=BA=86SaToken?=
 =?UTF-8?q?=E7=9A=84=E6=B3=A8=E8=A7=A3=E9=89=B4=E6=9D=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../videotask/config/SaTokenConfigure.java       | 16 ++++++++++++++++
 .../videotask/controller/NodeControl.java        |  2 ++
 .../videotask/controller/UserControl.java        |  6 ++++--
 .../hongliang/videotask/mappers/NodeMapper.java  | 12 ++++++++++++
 4 files changed, 34 insertions(+), 2 deletions(-)
 create mode 100644 src/main/java/com/hongliang/videotask/config/SaTokenConfigure.java

diff --git a/src/main/java/com/hongliang/videotask/config/SaTokenConfigure.java b/src/main/java/com/hongliang/videotask/config/SaTokenConfigure.java
new file mode 100644
index 0000000..fc43fc6
--- /dev/null
+++ b/src/main/java/com/hongliang/videotask/config/SaTokenConfigure.java
@@ -0,0 +1,16 @@
+package com.hongliang.videotask.config;
+
+import cn.dev33.satoken.interceptor.SaInterceptor;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+
+@Configuration
+public class SaTokenConfigure implements WebMvcConfigurer {
+    // 注册 Sa-Token 拦截器，打开注解式鉴权功能
+    @Override
+    public void addInterceptors(InterceptorRegistry registry) {
+        // 注册 Sa-Token 拦截器，打开注解式鉴权功能
+        registry.addInterceptor(new SaInterceptor()).addPathPatterns("/**");
+    }
+}
diff --git a/src/main/java/com/hongliang/videotask/controller/NodeControl.java b/src/main/java/com/hongliang/videotask/controller/NodeControl.java
index dd22ab0..102dae8 100644
--- a/src/main/java/com/hongliang/videotask/controller/NodeControl.java
+++ b/src/main/java/com/hongliang/videotask/controller/NodeControl.java
@@ -1,5 +1,6 @@
 package com.hongliang.videotask.controller;
 
+import cn.dev33.satoken.annotation.SaCheckLogin;
 import com.hongliang.videotask.common.Response;
 import com.hongliang.videotask.common.ResponseCode;
 import com.hongliang.videotask.service.impl.NodeServiceImpl;
@@ -17,6 +18,7 @@ public class NodeControl {
     @Resource
     private NodeServiceImpl nodeService;
 
+    @SaCheckLogin
     @PostMapping("list")
     public ResponseEntity<?> GetNodeListByCondition(@RequestBody String condition){
         return ResponseEntity.ok(new Response(ResponseCode.OK,"success",
diff --git a/src/main/java/com/hongliang/videotask/controller/UserControl.java b/src/main/java/com/hongliang/videotask/controller/UserControl.java
index 9b4340d..231b638 100644
--- a/src/main/java/com/hongliang/videotask/controller/UserControl.java
+++ b/src/main/java/com/hongliang/videotask/controller/UserControl.java
@@ -1,5 +1,6 @@
 package com.hongliang.videotask.controller;
 
+import cn.dev33.satoken.annotation.SaCheckLogin;
 import cn.dev33.satoken.stp.StpUtil;
 import com.hongliang.videotask.bean.LoginBean;
 import com.hongliang.videotask.bean.LoginResultBean;
@@ -56,8 +57,9 @@ public class UserControl {
         return ResponseEntity.ok(new Response(ResponseCode.OK,"成功退出",null));
     }
 
-    @GetMapping("getuserbyid")
-    public ResponseEntity<?> GetUserByID(@RequestParam("id") int id){
+    @SaCheckLogin
+    @GetMapping("/user/{id}")
+    public ResponseEntity<?> GetUserByID(@PathVariable("id") int id){
         return ResponseEntity.ok(new Response(ResponseCode.OK,"成功",userService.GetUserByID(id)));
     }
 }
diff --git a/src/main/java/com/hongliang/videotask/mappers/NodeMapper.java b/src/main/java/com/hongliang/videotask/mappers/NodeMapper.java
index d20844f..5381f00 100644
--- a/src/main/java/com/hongliang/videotask/mappers/NodeMapper.java
+++ b/src/main/java/com/hongliang/videotask/mappers/NodeMapper.java
@@ -2,7 +2,10 @@ package com.hongliang.videotask.mappers;
 
 import com.hongliang.videotask.MapperProvider.NodeProvider;
 import com.hongliang.videotask.bean.NodeBean;
+import org.apache.ibatis.annotations.Insert;
+import org.apache.ibatis.annotations.Options;
 import org.apache.ibatis.annotations.SelectProvider;
+import org.apache.ibatis.annotations.Update;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
@@ -15,4 +18,13 @@ public interface NodeMapper {
 
     @SelectProvider(type=NodeProvider.class,method="GetTotal")
     int GetTotal(Map<String,Object> condition);
+
+    @Insert("insert into node (nodename) values (#nodename)")
+    @Options(useGeneratedKeys = true, keyProperty = "id")
+    int InsertNode(NodeBean nodeBean);
+
+    @Update("update node set nodename=#{#nodename} where id=#{id}")
+    int UpdateNode(NodeBean nodeBean);
+
+
 }