|
|
@@ -10,16 +10,18 @@ type Ability struct {
|
|
|
Model string `json:"model" gorm:"primaryKey;autoIncrement:false"`
|
|
|
ChannelId int `json:"channel_id" gorm:"primaryKey;autoIncrement:false;index"`
|
|
|
Enabled bool `json:"enabled"`
|
|
|
- Priority *int64 `json:"priority" gorm:"bigint;default:0"`
|
|
|
+ Priority *int64 `json:"priority" gorm:"bigint;default:0;index"`
|
|
|
}
|
|
|
|
|
|
func GetRandomSatisfiedChannel(group string, model string) (*Channel, error) {
|
|
|
ability := Ability{}
|
|
|
var err error = nil
|
|
|
if common.UsingSQLite {
|
|
|
- err = DB.Where("`group` = ? and model = ? and enabled = 1", group, model).Order("CASE WHEN priority <> 0 THEN priority ELSE RANDOM() END DESC ").Limit(1).First(&ability).Error
|
|
|
+ maxPrioritySubQuery := DB.Model(&Ability{}).Select("MAX(priority)").Where("`group` = ? and model = ? and enabled = 1", group, model)
|
|
|
+ err = DB.Where("`group` = ? and model = ? and enabled = 1 and priority = (?)", group, model, maxPrioritySubQuery).Order("RANDOM()").Limit(1).First(&ability).Error
|
|
|
} else {
|
|
|
- err = DB.Where("`group` = ? and model = ? and enabled = 1", group, model).Order("CASE WHEN priority <> 0 THEN priority ELSE RAND() END DESC").Limit(1).First(&ability).Error
|
|
|
+ maxPrioritySubQuery := DB.Model(&Ability{}).Select("MAX(priority)").Where("group = ? and model = ? and enabled = 1", group, model)
|
|
|
+ err = DB.Where("`group` = ? and model = ? and enabled = 1 and priority = (?)", group, model, maxPrioritySubQuery).Order("RAND()").Limit(1).First(&ability).Error
|
|
|
}
|
|
|
if err != nil {
|
|
|
return nil, err
|