فهرست منبع

fix: enable cors for token routers and dashboard routers

CaIon 2 سال پیش
والد
کامیت
946eed4d60
3فایلهای تغییر یافته به همراه8 افزوده شده و 2 حذف شده
  1. 1 1
      controller/relay-text.go
  2. 6 1
      router/api-router.go
  3. 1 0
      router/dashboard.go

+ 1 - 1
controller/relay-text.go

@@ -212,7 +212,7 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 		// in this case, we do not pre-consume quota
 		// because the user has enough quota
 		preConsumedQuota = 0
-		common.LogInfo(c.Request.Context(), fmt.Sprintf("user %d has enough quota %d, trusted and no need to pre-consume", userId, userQuota))
+		//common.LogInfo(c.Request.Context(), fmt.Sprintf("user %d has enough quota %d, trusted and no need to pre-consume", userId, userQuota))
 	}
 	if consumeQuota && preConsumedQuota > 0 {
 		err := model.PreConsumeTokenQuota(tokenId, preConsumedQuota)

+ 6 - 1
router/api-router.go

@@ -109,7 +109,12 @@ func SetApiRouter(router *gin.Engine) {
 		logRoute.GET("/search", middleware.AdminAuth(), controller.SearchAllLogs)
 		logRoute.GET("/self", middleware.UserAuth(), controller.GetUserLogs)
 		logRoute.GET("/self/search", middleware.UserAuth(), controller.SearchUserLogs)
-		logRoute.GET("/token", controller.GetLogByKey)
+
+		logRoute.Use(middleware.CORS())
+		{
+			logRoute.GET("/token", controller.GetLogByKey)
+
+		}
 		groupRoute := apiRouter.Group("/group")
 		groupRoute.Use(middleware.AdminAuth())
 		{

+ 1 - 0
router/dashboard.go

@@ -11,6 +11,7 @@ func SetDashboardRouter(router *gin.Engine) {
 	apiRouter := router.Group("/")
 	apiRouter.Use(gzip.Gzip(gzip.DefaultCompression))
 	apiRouter.Use(middleware.GlobalAPIRateLimit())
+	apiRouter.Use(middleware.CORS())
 	apiRouter.Use(middleware.TokenAuth())
 	{
 		apiRouter.GET("/dashboard/billing/subscription", controller.GetSubscription)