From 52615d1646184deeaaf62f21d22bdda8ce08cc65 Mon Sep 17 00:00:00 2001
From: TinyAnts <tinyants@163.com>
Date: Thu, 17 Nov 2022 14:53:29 +0800
Subject: [PATCH] =?UTF-8?q?sa-token=E5=BC=82=E5=B8=B8=E6=8B=A6=E6=88=AA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/mdd/admin/config/StpInException.java  | 41 +++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 server/like-admin/src/main/java/com/mdd/admin/config/StpInException.java

diff --git a/server/like-admin/src/main/java/com/mdd/admin/config/StpInException.java b/server/like-admin/src/main/java/com/mdd/admin/config/StpInException.java
new file mode 100644
index 00000000..7aa045a3
--- /dev/null
+++ b/server/like-admin/src/main/java/com/mdd/admin/config/StpInException.java
@@ -0,0 +1,41 @@
+package com.mdd.admin.config;
+
+import cn.dev33.satoken.exception.NotLoginException;
+import cn.dev33.satoken.exception.NotPermissionException;
+import com.mdd.common.core.AjaxResult;
+import com.mdd.common.enums.HttpEnum;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.http.HttpStatus;
+import org.springframework.web.bind.annotation.ControllerAdvice;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.ResponseBody;
+import org.springframework.web.bind.annotation.ResponseStatus;
+
+/**
+ * Sa-Token的异常拦截
+ */
+@Slf4j
+@ControllerAdvice
+public class StpInException {
+
+    /**
+     * 拦截登录异常
+     */
+    @ResponseStatus(HttpStatus.OK)
+    @ExceptionHandler(NotLoginException.class)
+    @ResponseBody
+    public AjaxResult<Object> handleNotLoginException(){
+        return AjaxResult.failed(HttpEnum.TOKEN_INVALID.getCode(), HttpEnum.TOKEN_INVALID.getMsg());
+    }
+
+    /**
+     * 拦截权限异常
+     */
+    @ResponseStatus(HttpStatus.OK)
+    @ExceptionHandler(NotPermissionException.class)
+    @ResponseBody
+    public AjaxResult<Object> handleNotPermissionException(){
+        return AjaxResult.failed(HttpEnum.NO_PERMISSION.getCode(), HttpEnum.NO_PERMISSION.getMsg());
+    }
+
+}