feat：完善项目信息

cmd/serve/init.go

@@ -15,6 +15,7 @@ import (
 	"SciencesServer/utils"
 	"fmt"
 	"github.com/spf13/cobra"
+	"net/http"
 	"strings"
 )
 
@@ -69,13 +70,25 @@ func run() {
 	// 开启Elasticsearch
 	es.NewInstance(es.WithEsAddress([]string{config.SettingInfo.ESServer.Host})).Init().Local()
 	// 开启web
-	web.NewWeb()(&web.WebConfig{
-		Port: config.SettingInfo.Server.Port, ReadTimeout: config.SettingInfo.Server.ReadTimeout,
-		WriteTimeout: config.SettingInfo.Server.WriteTimeout, IdleTimeout: config.SettingInfo.Server.IdleTimeout,
-	}).Run(router.NewRouter(&router.Option{
-		Mode: config.Mode, IsCors: true,
-		RateLimitConfig: &router.RateLimitConfig{
-			IsRate: true, Limit: config.SettingInfo.Rate.Limit, Capacity: config.SettingInfo.Rate.Capacity,
-		},
-	}).Init())
+	web.NewWeb(web.WithPort(config.SettingInfo.Server.Port),
+		web.WithReadTimeout(config.SettingInfo.Server.ReadTimeout),
+		web.WithWriteTimeout(config.SettingInfo.Server.WriteTimeout),
+		web.WithIdleTimeout(config.SettingInfo.Server.IdleTimeout),
+		web.WithHandler(router.NewRouter(&router.Option{
+			Mode: config.Mode, IsCors: true,
+			RateLimitConfig: &router.RateLimitConfig{
+				IsRate: true, Limit: config.SettingInfo.Rate.Limit, Capacity: config.SettingInfo.Rate.Capacity,
+			},
+		}).Init()),
+		web.WithFunction(func(src string) (bool, func(r *http.Request)) {
+			value, _ := cache.Cache.HGet(config.RedisKeyForTenant, src)
+
+			if value == "" {
+				return true, nil
+			}
+			return true, func(r *http.Request) {
+				r.Header.Set("", value)
+			}
+		}),
+	).Run()
 }