From dfa2a5e967d70081eb3e98c6b77900e3800be273 Mon Sep 17 00:00:00 2001
From: yinbentan <yinbentan@live.com>
Date: 星期五, 28 六月 2024 14:14:33 +0800
Subject: [PATCH] 开发环境dev跳过登录验证
---
middleware/reset_pwd.go | 7 ++++++-
constvar/env.go | 21 +++++++++++++++++++++
middleware/jwt.go | 5 +++++
conf/config.go | 2 ++
4 files changed, 34 insertions(+), 1 deletions(-)
diff --git a/conf/config.go b/conf/config.go
index 8f40e75..8e703e5 100644
--- a/conf/config.go
+++ b/conf/config.go
@@ -3,6 +3,7 @@
import (
"log"
"os"
+ "wms/constvar"
"wms/pkg/logx"
"wms/pkg/mysqlx"
@@ -30,6 +31,7 @@
JWTSecret string
FileServer string //鏂囦欢鏈嶅姟鍣ㄥ湴鍧�
ServerId string //鏈嶅姟ID
+ Env constvar.Env
GrpcPort string //grpc绔彛鍙�
CompanyName string //鍏徃鍚�
}
diff --git a/constvar/env.go b/constvar/env.go
new file mode 100644
index 0000000..92d5508
--- /dev/null
+++ b/constvar/env.go
@@ -0,0 +1,21 @@
+package constvar
+
+type Env string
+
+const (
+ EnvDev = "dev"
+ EnvTest = "test"
+ EnvProd = "prod"
+)
+
+func (e Env) IsProd() bool {
+ return e == EnvProd
+}
+
+func (e Env) IsDev() bool {
+ return e == EnvDev
+}
+
+func (e Env) IsTest() bool {
+ return e == EnvTest
+}
diff --git a/middleware/jwt.go b/middleware/jwt.go
index d9d7dd4..2a8fd8f 100644
--- a/middleware/jwt.go
+++ b/middleware/jwt.go
@@ -3,12 +3,17 @@
import (
"github.com/gin-gonic/gin"
"strings"
+ "wms/conf"
"wms/pkg/contextx"
"wms/pkg/ecode"
)
func JWTAuth() gin.HandlerFunc {
return func(c *gin.Context) {
+ if conf.WebConf.Env.IsDev() {
+ c.Next()
+ return
+ }
ctx := new(contextx.Context).SetCtx(c)
// 鎴戜滑杩欓噷jwt閴存潈鍙栧ご閮ㄤ俊鎭� Authorization 鐧诲綍鏃跺洖杩斿洖token淇℃伅 杩欓噷鍓嶇闇�瑕佹妸token瀛樺偍鍒癱ookie鎴栬�呮湰鍦發ocalStorage涓� 涓嶈繃闇�瑕佽窡鍚庣鍗忓晢杩囨湡鏃堕棿 鍙互绾﹀畾鍒锋柊浠ょ墝鎴栬�呴噸鏂扮櫥褰�
token := c.Request.Header.Get("Authorization")
diff --git a/middleware/reset_pwd.go b/middleware/reset_pwd.go
index 64e2785..e9a9b2c 100644
--- a/middleware/reset_pwd.go
+++ b/middleware/reset_pwd.go
@@ -2,6 +2,7 @@
import (
"github.com/gin-gonic/gin"
+ "wms/conf"
"wms/pkg/contextx"
"wms/pkg/ecode"
"wms/request"
@@ -9,10 +10,14 @@
func VerifyResetPwd() gin.HandlerFunc {
return func(c *gin.Context) {
+ if conf.WebConf.Env.IsDev() {
+ c.Next()
+ return
+ }
ctx := new(contextx.Context).SetCtx(c)
params, ok := c.Get("claims")
if !ok {
- c.Abort()
+ c.Next()
return
}
claims := params.(*request.CustomClaims)
--
Gitblit v1.8.0