Test: middleware/auth

service/user/common.go

@@ -1,22 +0,0 @@
-package user
-
-import (
-	"github.com/gin-contrib/sessions"
-	"github.com/gin-gonic/gin"
-)
-
-// SetSession 设置session
-func SetSession(c *gin.Context, list map[string]interface{}) {
-	s := sessions.Default(c)
-	for key, value := range list {
-		s.Set(key, value)
-	}
-	s.Save()
-}
-
-// ClearSession 清空session
-func ClearSession(c *gin.Context) {
-	s := sessions.Default(c)
-	s.Clear()
-	s.Save()
-}

service/user/login.go

@@ -3,6 +3,7 @@ package user
 import (
 	"cloudreve/models"
 	"cloudreve/pkg/serializer"
+	"cloudreve/pkg/util"
 	"github.com/gin-gonic/gin"
 )
 
@@ -42,7 +43,7 @@ func (service *UserLoginService) Login(c *gin.Context) serializer.Response {
 	}
 
 	//登陆成功，清空并设置session
-	SetSession(c, map[string]interface{}{
+	util.SetSession(c, map[string]interface{}{
 		"user_id": expectedUser.ID,
 	})