淘特

upstream/itao/itao.go

@@ -20,6 +20,7 @@ type Config struct {
 	ApiUrl    string
 	AppKey    string
 	AppSecret string
+	DB        int
 }
 
 var server *Config
@@ -30,6 +31,9 @@ const Host = "https://open.huoju6.com/openapi/param2/1/com.huoju6.open/"
 
 func New(req *Config) {
 	server = req
+	if server.DB == 0 {
+		server.DB = 10
+	}
 	return
 }
 
@@ -156,7 +160,7 @@ func AccessToken(ctx context.Context) (res string, err error) {
 	defer func() {
 		_ = conn.Close()
 	}()
-	_, _ = conn.DoVar("SELECT", 10)
+	_, _ = conn.DoVar("SELECT", server.DB)
 	cache, _ := conn.DoVar("HGETALL", CacheKey)
 	if cache.IsEmpty() {
 		g.Log().Ctx(ctx).Cat(PkgName).Line(true).Info("获取token失败")
@@ -186,7 +190,7 @@ func RefreshToken(ctx context.Context) (res *AuthTokenRes, err error) {
 	defer func() {
 		_ = conn.Close()
 	}()
-	_, _ = conn.DoVar("SELECT", 10)
+	_, _ = conn.DoVar("SELECT", server.DB)
 	cache, _ := conn.DoVar("HGETALL", CacheKey)
 	if cache.IsEmpty() {
 		g.Log().Ctx(ctx).Cat(PkgName).Line(true).Info("获取token失败")