Skip to content

Commit d365f42

Browse files
committed
style: Add advance middleware
1 parent 60ea68e commit d365f42

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

api/router/api.go

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,7 @@ func NewRouter(config *config.Config, enableSwagger bool) (*gin.Engine, error) {
8181
middlewareCollection.Auth.NeedPhoneVerified = middleware.NeedPhoneVerified(config)
8282
middlewareCollection.Repo.RepoExists = middleware.RepoExists(config)
8383
middlewareCollection.License.Check = middleware.CheckLicense(config)
84-
middlewareCollection.API.Captcha = middleware.NewCaptcha(config).Handle
85-
rateLimitMiddleware := middleware.RateLimiter(config, middleware.WithTimeBucketRateLimter(config), middleware.WithIPCheck())
86-
middlewareCollection.API.IPLimiter = middleware.IPLocationCheck(config)
87-
middlewareCollection.API.RateLimter = rateLimitMiddleware
84+
8885
//add router for golang pprof
8986
debugGroup := r.Group("/debug", middlewareCollection.Auth.NeedAPIKey)
9087
pprof.RouteRegister(debugGroup, "pprof")
@@ -186,7 +183,7 @@ func NewRouter(config *config.Config, enableSwagger bool) (*gin.Engine, error) {
186183

187184
r.Use(middleware.LocalizedErrorMiddleware())
188185
r.Use(middleware.Authenticator(config))
189-
r.Use(middlewareCollection.API.IPLimiter, middlewareCollection.API.RateLimter, middlewareCollection.API.Captcha)
186+
useAdvancedMiddleware(r, config)
190187
apiGroup := r.Group("/api/v1")
191188

192189
versionHandler := handler.NewVersionHandler()

api/router/api_ce.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ import (
99
"opencsg.com/csghub-server/common/config"
1010
)
1111

12+
func useAdvancedMiddleware(r *gin.Engine, config *config.Config) {}
13+
1214
func createAdvancedRoutes(apiGroup *gin.RouterGroup, middlewareCollection middleware.MiddlewareCollection, config *config.Config) error {
1315
return nil
1416
}

0 commit comments

Comments
 (0)