|
|
@@ -14,102 +14,102 @@ import (
|
|
|
//func CacheGetUserGroup(id int) (group string, err error) {
|
|
|
// if !common.RedisEnabled {
|
|
|
// return GetUserGroup(id)
|
|
|
- // }
|
|
|
- // group, err = common.RedisGet(fmt.Sprintf("user_group:%d", id))
|
|
|
- // if err != nil {
|
|
|
- // group, err = GetUserGroup(id)
|
|
|
- // if err != nil {
|
|
|
- // return "", err
|
|
|
- // }
|
|
|
- // err = common.RedisSet(fmt.Sprintf("user_group:%d", id), group, time.Duration(constant.UserId2GroupCacheSeconds)*time.Second)
|
|
|
- // if err != nil {
|
|
|
- // common.SysError("Redis set user group error: " + err.Error())
|
|
|
- // }
|
|
|
- // }
|
|
|
- // return group, err
|
|
|
- //}
|
|
|
+// }
|
|
|
+// group, err = common.RedisGet(fmt.Sprintf("user_group:%d", id))
|
|
|
+// if err != nil {
|
|
|
+// group, err = GetUserGroup(id)
|
|
|
+// if err != nil {
|
|
|
+// return "", err
|
|
|
+// }
|
|
|
+// err = common.RedisSet(fmt.Sprintf("user_group:%d", id), group, time.Duration(constant.UserId2GroupCacheSeconds)*time.Second)
|
|
|
+// if err != nil {
|
|
|
+// common.SysError("Redis set user group error: " + err.Error())
|
|
|
+// }
|
|
|
+// }
|
|
|
+// return group, err
|
|
|
+//}
|
|
|
//
|
|
|
- //func CacheGetUsername(id int) (username string, err error) {
|
|
|
- // if !common.RedisEnabled {
|
|
|
- // return GetUsernameById(id)
|
|
|
- // }
|
|
|
- // username, err = common.RedisGet(fmt.Sprintf("user_name:%d", id))
|
|
|
- // if err != nil {
|
|
|
- // username, err = GetUsernameById(id)
|
|
|
- // if err != nil {
|
|
|
- // return "", err
|
|
|
- // }
|
|
|
- // err = common.RedisSet(fmt.Sprintf("user_name:%d", id), username, time.Duration(constant.UserId2GroupCacheSeconds)*time.Second)
|
|
|
- // if err != nil {
|
|
|
- // common.SysError("Redis set user group error: " + err.Error())
|
|
|
- // }
|
|
|
- // }
|
|
|
- // return username, err
|
|
|
- //}
|
|
|
+//func CacheGetUsername(id int) (username string, err error) {
|
|
|
+// if !common.RedisEnabled {
|
|
|
+// return GetUsernameById(id)
|
|
|
+// }
|
|
|
+// username, err = common.RedisGet(fmt.Sprintf("user_name:%d", id))
|
|
|
+// if err != nil {
|
|
|
+// username, err = GetUsernameById(id)
|
|
|
+// if err != nil {
|
|
|
+// return "", err
|
|
|
+// }
|
|
|
+// err = common.RedisSet(fmt.Sprintf("user_name:%d", id), username, time.Duration(constant.UserId2GroupCacheSeconds)*time.Second)
|
|
|
+// if err != nil {
|
|
|
+// common.SysError("Redis set user group error: " + err.Error())
|
|
|
+// }
|
|
|
+// }
|
|
|
+// return username, err
|
|
|
+//}
|
|
|
//
|
|
|
- //func CacheGetUserQuota(id int) (quota int, err error) {
|
|
|
- // if !common.RedisEnabled {
|
|
|
- // return GetUserQuota(id)
|
|
|
- // }
|
|
|
- // quotaString, err := common.RedisGet(fmt.Sprintf("user_quota:%d", id))
|
|
|
- // if err != nil {
|
|
|
- // quota, err = GetUserQuota(id)
|
|
|
- // if err != nil {
|
|
|
- // return 0, err
|
|
|
- // }
|
|
|
- // return quota, nil
|
|
|
- // }
|
|
|
- // quota, err = strconv.Atoi(quotaString)
|
|
|
- // return quota, nil
|
|
|
- //}
|
|
|
+//func CacheGetUserQuota(id int) (quota int, err error) {
|
|
|
+// if !common.RedisEnabled {
|
|
|
+// return GetUserQuota(id)
|
|
|
+// }
|
|
|
+// quotaString, err := common.RedisGet(fmt.Sprintf("user_quota:%d", id))
|
|
|
+// if err != nil {
|
|
|
+// quota, err = GetUserQuota(id)
|
|
|
+// if err != nil {
|
|
|
+// return 0, err
|
|
|
+// }
|
|
|
+// return quota, nil
|
|
|
+// }
|
|
|
+// quota, err = strconv.Atoi(quotaString)
|
|
|
+// return quota, nil
|
|
|
+//}
|
|
|
//
|
|
|
- //func CacheUpdateUserQuota(id int) error {
|
|
|
- // if !common.RedisEnabled {
|
|
|
- // return nil
|
|
|
- // }
|
|
|
- // quota, err := GetUserQuota(id)
|
|
|
- // if err != nil {
|
|
|
- // return err
|
|
|
- // }
|
|
|
- // return cacheSetUserQuota(id, quota)
|
|
|
- //}
|
|
|
+//func CacheUpdateUserQuota(id int) error {
|
|
|
+// if !common.RedisEnabled {
|
|
|
+// return nil
|
|
|
+// }
|
|
|
+// quota, err := GetUserQuota(id)
|
|
|
+// if err != nil {
|
|
|
+// return err
|
|
|
+// }
|
|
|
+// return cacheSetUserQuota(id, quota)
|
|
|
+//}
|
|
|
//
|
|
|
- //func cacheSetUserQuota(id int, quota int) error {
|
|
|
- // err := common.RedisSet(fmt.Sprintf("user_quota:%d", id), fmt.Sprintf("%d", quota), time.Duration(constant.UserId2QuotaCacheSeconds)*time.Second)
|
|
|
- // return err
|
|
|
- //}
|
|
|
+//func cacheSetUserQuota(id int, quota int) error {
|
|
|
+// err := common.RedisSet(fmt.Sprintf("user_quota:%d", id), fmt.Sprintf("%d", quota), time.Duration(constant.UserId2QuotaCacheSeconds)*time.Second)
|
|
|
+// return err
|
|
|
+//}
|
|
|
//
|
|
|
- //func CacheDecreaseUserQuota(id int, quota int) error {
|
|
|
- // if !common.RedisEnabled {
|
|
|
- // return nil
|
|
|
- // }
|
|
|
- // err := common.RedisDecrease(fmt.Sprintf("user_quota:%d", id), int64(quota))
|
|
|
- // return err
|
|
|
- //}
|
|
|
+//func CacheDecreaseUserQuota(id int, quota int) error {
|
|
|
+// if !common.RedisEnabled {
|
|
|
+// return nil
|
|
|
+// }
|
|
|
+// err := common.RedisDecrease(fmt.Sprintf("user_quota:%d", id), int64(quota))
|
|
|
+// return err
|
|
|
+//}
|
|
|
//
|
|
|
- //func CacheIsUserEnabled(userId int) (bool, error) {
|
|
|
- // if !common.RedisEnabled {
|
|
|
- // return IsUserEnabled(userId)
|
|
|
- // }
|
|
|
- // enabled, err := common.RedisGet(fmt.Sprintf("user_enabled:%d", userId))
|
|
|
- // if err == nil {
|
|
|
- // return enabled == "1", nil
|
|
|
- // }
|
|
|
+//func CacheIsUserEnabled(userId int) (bool, error) {
|
|
|
+// if !common.RedisEnabled {
|
|
|
+// return IsUserEnabled(userId)
|
|
|
+// }
|
|
|
+// enabled, err := common.RedisGet(fmt.Sprintf("user_enabled:%d", userId))
|
|
|
+// if err == nil {
|
|
|
+// return enabled == "1", nil
|
|
|
+// }
|
|
|
//
|
|
|
- // userEnabled, err := IsUserEnabled(userId)
|
|
|
- // if err != nil {
|
|
|
- // return false, err
|
|
|
- // }
|
|
|
- // enabled = "0"
|
|
|
- // if userEnabled {
|
|
|
- // enabled = "1"
|
|
|
- // }
|
|
|
- // err = common.RedisSet(fmt.Sprintf("user_enabled:%d", userId), enabled, time.Duration(constant.UserId2StatusCacheSeconds)*time.Second)
|
|
|
- // if err != nil {
|
|
|
- // common.SysError("Redis set user enabled error: " + err.Error())
|
|
|
- // }
|
|
|
- // return userEnabled, err
|
|
|
- //}
|
|
|
+// userEnabled, err := IsUserEnabled(userId)
|
|
|
+// if err != nil {
|
|
|
+// return false, err
|
|
|
+// }
|
|
|
+// enabled = "0"
|
|
|
+// if userEnabled {
|
|
|
+// enabled = "1"
|
|
|
+// }
|
|
|
+// err = common.RedisSet(fmt.Sprintf("user_enabled:%d", userId), enabled, time.Duration(constant.UserId2StatusCacheSeconds)*time.Second)
|
|
|
+// if err != nil {
|
|
|
+// common.SysError("Redis set user enabled error: " + err.Error())
|
|
|
+// }
|
|
|
+// return userEnabled, err
|
|
|
+//}
|
|
|
|
|
|
var group2model2channels map[string]map[string][]*Channel
|
|
|
var channelsIDM map[int]*Channel
|
|
|
@@ -181,7 +181,7 @@ func CacheGetRandomSatisfiedChannel(group string, model string, retry int) (*Cha
|
|
|
}
|
|
|
|
|
|
// if memory cache is disabled, get channel directly from database
|
|
|
- if (!common.MemoryCacheEnabled) {
|
|
|
+ if !common.MemoryCacheEnabled {
|
|
|
return GetRandomSatisfiedChannel(group, model, retry)
|
|
|
}
|
|
|
channelSyncLock.RLock()
|