From 49df4b6eed1d2888769e34065aa6d3c593ad6200 Mon Sep 17 00:00:00 2001 From: Xiangyuan Liu Date: Fri, 29 Mar 2024 16:48:50 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=94=AF=E6=8C=81=20/mj-{mode}=20?= =?UTF-8?q?=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- middleware/distributor.go | 2 +- router/relay-router.go | 10 +++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/middleware/distributor.go b/middleware/distributor.go index a5e40b0..10696a9 100644 --- a/middleware/distributor.go +++ b/middleware/distributor.go @@ -44,7 +44,7 @@ func Distribute() func(c *gin.Context) { // Select a channel for the user var modelRequest ModelRequest var err error - if strings.HasPrefix(c.Request.URL.Path, "/mj") { + if strings.Contains(c.Request.URL.Path, "/mj/") { relayMode := relayconstant.Path2RelayModeMidjourney(c.Request.URL.Path) if relayMode == relayconstant.RelayModeMidjourneyTaskFetch || relayMode == relayconstant.RelayModeMidjourneyTaskFetchByCondition || diff --git a/router/relay-router.go b/router/relay-router.go index 4addee0..2d8e7b3 100644 --- a/router/relay-router.go +++ b/router/relay-router.go @@ -43,7 +43,16 @@ func SetRelayRouter(router *gin.Engine) { relayV1Router.DELETE("/models/:model", controller.RelayNotImplemented) relayV1Router.POST("/moderations", controller.Relay) } + relayMjRouter := router.Group("/mj") + registerMjRouterGroup(relayMjRouter) + + relayMjModeRouter := router.Group("/:mode/mj") + registerMjRouterGroup(relayMjModeRouter) + //relayMjRouter.Use() +} + +func registerMjRouterGroup(relayMjRouter *gin.RouterGroup) { relayMjRouter.GET("/image/:id", relay.RelayMidjourneyImage) relayMjRouter.Use(middleware.TokenAuth(), middleware.Distribute()) { @@ -61,5 +70,4 @@ func SetRelayRouter(router *gin.Engine) { relayMjRouter.POST("/task/list-by-condition", controller.RelayMidjourney) relayMjRouter.POST("/insight-face/swap", controller.RelayMidjourney) } - //relayMjRouter.Use() }