relay-router.go 558 B

1234567891011121314151617181920
  1. package router
  2. import (
  3. "github.com/gin-gonic/gin"
  4. "one-api/controller"
  5. "one-api/middleware"
  6. )
  7. func SetRelayRouter(router *gin.Engine) {
  8. relayV1Router := router.Group("/v1")
  9. relayV1Router.Use(middleware.GlobalAPIRateLimit(), middleware.TokenAuth(), middleware.Distribute())
  10. {
  11. relayV1Router.Any("/*path", controller.Relay)
  12. }
  13. relayDashboardRouter := router.Group("/dashboard")
  14. relayDashboardRouter.Use(middleware.GlobalAPIRateLimit(), middleware.TokenAuth(), middleware.Distribute())
  15. {
  16. relayDashboardRouter.Any("/*path", controller.Relay)
  17. }
  18. }