Browse Source

fix: fix channel panic

CaIon 2 years ago
parent
commit
3e13810ca2
1 changed files with 4 additions and 0 deletions
  1. 4 0
      middleware/distributor.go

+ 4 - 0
middleware/distributor.go

@@ -108,6 +108,10 @@ func Distribute() func(c *gin.Context) {
 				abortWithMessage(c, http.StatusServiceUnavailable, message)
 				return
 			}
+			if channel == nil {
+				abortWithMessage(c, http.StatusServiceUnavailable, fmt.Sprintf("当前分组 %s 下对于模型 %s 无可用渠道(数据库一致性已被破坏)", userGroup, modelRequest.Model))
+				return
+			}
 		}
 		c.Set("channel", channel.Type)
 		c.Set("channel_id", channel.Id)