6748b006b7
This commit refactors the logging mechanism across the application by replacing direct logger calls with a centralized logging approach using the `common` package. Key changes include: - Replaced instances of `logger.SysLog` and `logger.FatalLog` with `common.SysLog` and `common.FatalLog` for consistent logging practices. - Updated resource initialization error handling to utilize the new logging structure, enhancing maintainability and readability. - Minor adjustments to improve code clarity and organization throughout various modules. This change aims to streamline logging and improve the overall architecture of the codebase.
77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
package setting
|
||
|
||
import (
|
||
"encoding/json"
|
||
"one-api/common"
|
||
"sync"
|
||
)
|
||
|
||
var userUsableGroups = map[string]string{
|
||
"default": "默认分组",
|
||
"vip": "vip分组",
|
||
}
|
||
var userUsableGroupsMutex sync.RWMutex
|
||
|
||
func GetUserUsableGroupsCopy() map[string]string {
|
||
userUsableGroupsMutex.RLock()
|
||
defer userUsableGroupsMutex.RUnlock()
|
||
|
||
copyUserUsableGroups := make(map[string]string)
|
||
for k, v := range userUsableGroups {
|
||
copyUserUsableGroups[k] = v
|
||
}
|
||
return copyUserUsableGroups
|
||
}
|
||
|
||
func UserUsableGroups2JSONString() string {
|
||
userUsableGroupsMutex.RLock()
|
||
defer userUsableGroupsMutex.RUnlock()
|
||
|
||
jsonBytes, err := json.Marshal(userUsableGroups)
|
||
if err != nil {
|
||
common.SysLog("error marshalling user groups: " + err.Error())
|
||
}
|
||
return string(jsonBytes)
|
||
}
|
||
|
||
func UpdateUserUsableGroupsByJSONString(jsonStr string) error {
|
||
userUsableGroupsMutex.Lock()
|
||
defer userUsableGroupsMutex.Unlock()
|
||
|
||
userUsableGroups = make(map[string]string)
|
||
return json.Unmarshal([]byte(jsonStr), &userUsableGroups)
|
||
}
|
||
|
||
func GetUserUsableGroups(userGroup string) map[string]string {
|
||
groupsCopy := GetUserUsableGroupsCopy()
|
||
if userGroup == "" {
|
||
if _, ok := groupsCopy["default"]; !ok {
|
||
groupsCopy["default"] = "default"
|
||
}
|
||
}
|
||
// 如果userGroup不在UserUsableGroups中，返回UserUsableGroups + userGroup
|
||
if _, ok := groupsCopy[userGroup]; !ok {
|
||
groupsCopy[userGroup] = "用户分组"
|
||
}
|
||
// 如果userGroup在UserUsableGroups中，返回UserUsableGroups
|
||
return groupsCopy
|
||
}
|
||
|
||
func GroupInUserUsableGroups(groupName string) bool {
|
||
userUsableGroupsMutex.RLock()
|
||
defer userUsableGroupsMutex.RUnlock()
|
||
|
||
_, ok := userUsableGroups[groupName]
|
||
return ok
|
||
}
|
||
|
||
func GetUsableGroupDescription(groupName string) string {
|
||
userUsableGroupsMutex.RLock()
|
||
defer userUsableGroupsMutex.RUnlock()
|
||
|
||
if desc, ok := userUsableGroups[groupName]; ok {
|
||
return desc
|
||
}
|
||
return groupName
|
||
}
|