Преглед изворни кода

feat: 渠道新可选是否自动禁用功能

CaIon пре 2 година
родитељ
комит
b06c370fd9
1 измењених фајлова са 6 додато и 1 уклоњено
  1. 6 1
      middleware/distributor.go

+ 6 - 1
middleware/distributor.go

@@ -87,7 +87,12 @@ func Distribute() func(c *gin.Context) {
 		c.Set("channel", channel.Type)
 		c.Set("channel_id", channel.Id)
 		c.Set("channel_name", channel.Name)
-		c.Set("auto_ban", channel.AutoBan)
+		ban := true
+		// parse *int to bool
+		if channel.AutoBan != nil && *channel.AutoBan == 0 {
+			ban = false
+		}
+		c.Set("auto_ban", ban)
 		c.Set("model_mapping", channel.GetModelMapping())
 		c.Request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", channel.Key))
 		c.Set("base_url", channel.GetBaseURL())