采用中间件授权的方式

api/handler/sd_handler.go

@@ -10,6 +10,7 @@ package handler
 import (
 	"fmt"
 	"geekai/core"
+	"geekai/core/middleware"
 	"geekai/core/types"
 	"geekai/service"
 	"geekai/service/oss"
@@ -58,12 +59,19 @@ func NewSdJobHandler(app *core.AppServer,
 
 // RegisterRoutes 注册路由
 func (h *SdJobHandler) RegisterRoutes() {
-	group := h.App.Engine.Group("/api/sd")
-	group.POST("image", h.Image)
-	group.GET("jobs", h.JobList)
+	group := h.App.Engine.Group("/api/sd/")
+
+	// 公开接口，不需要授权
 	group.GET("imgWall", h.ImgWall)
-	group.GET("remove", h.Remove)
-	group.GET("publish", h.Publish)
+
+	// 需要用户授权的接口
+	group.Use(middleware.UserAuthMiddleware(h.App.Config.Session.SecretKey, h.App.Redis))
+	{
+		group.POST("image", h.Image)
+		group.GET("jobs", h.JobList)
+		group.GET("remove", h.Remove)
+		group.GET("publish", h.Publish)
+	}
 }
 
 func (h *SdJobHandler) preCheck(c *gin.Context) bool {