feat: anonymity user

GuGoOrg · Sep 1, 2023 · ad67c5f · ad67c5f
1 parent c041c56
commit ad67c5f
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/src/constant/config/.env.example b/src/constant/config/.env.example
@@ -81,4 +81,6 @@ GORSE_APIKEY=
 # MagicUserId, please use output value
 MAGIC_USER_ID=
 # Configure your clash proxy
-CHATGPT_PROXY=
+CHATGPT_PROXY=
+# Default anonymity user
+ANONYMITY_USER=
diff --git a/src/constant/config/env.go b/src/constant/config/env.go
@@ -49,6 +49,7 @@ type envConfig struct {
 	PostgreSQLReplicaPassword string  `env:"POSTGRESQL_REPLICA_PASSWORD"`
 	OtelState                 string  `env:"TRACING_STATE" envDefault:"enable"`
 	OtelSampler               float64 `env:"TRACING_SAMPLER" envDefault:"0.01"`
+	AnonymityUser             string  `env:"ANONYMITY_USER" envDefault:"114514"`
 }
 
 func init() {

diff --git a/src/web/middleware/authenticate.go b/src/web/middleware/authenticate.go
@@ -30,7 +30,7 @@ func TokenAuthMiddleware() gin.HandlerFunc {
 			c.Request.URL.Path == "/douyin/publish/list/" ||
 			c.Request.URL.Path == "/douyin/favorite/list/" ||
 			c.Request.URL.Path == "/douyin/relation/follower/list/" {
-			c.Request.URL.RawQuery += "&actor_id=114514"
+			c.Request.URL.RawQuery += "&actor_id=" + config.EnvCfg.AnonymityUser
 			c.Next()
 			logger.WithFields(logrus.Fields{
 				"Path": c.Request.URL.Path,

diff --git a/test/k6/feed.js b/test/k6/feed.js
@@ -7,17 +7,15 @@ export const options = {
             executor: 'ramping-vus',
             gracefulStop: '30s',
             stages: [
-                { target: 1000, duration: '15s' },
-                { target: 1500, duration: '30s' },
-                { target: 1000, duration: '15s' },
+                { target: 1000, duration: '60s' },
             ],
             gracefulRampDown: '30s',
-            exec: 'login',
+            exec: 'feed',
         },
     },
 }
 
-export function login() {
+export function feed() {
     http.get('https://gugotik.endymx.qzwxsaedc.cn/douyin/feed?')
 
     sleep(3)