71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
|
package cache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/Selly-Modules/logger"
|
||
|
|
"github.com/Selly-Modules/usermngmt/model"
|
||
|
|
"go.mongodb.org/mongo-driver/bson"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Roles ...
|
||
|
|
func Roles() {
|
||
|
|
var (
|
||
|
|
ctx = context.Background()
|
||
|
|
wg sync.WaitGroup
|
||
|
|
cond = bson.M{}
|
||
|
|
)
|
||
|
|
|
||
|
|
// Find
|
||
|
|
roles := roleFindByCondition(ctx, cond)
|
||
|
|
permissions := permissionFindByCondition(ctx, cond)
|
||
|
|
|
||
|
|
wg.Add(len(roles))
|
||
|
|
for _, value := range roles {
|
||
|
|
go func(role model.DBRole) {
|
||
|
|
defer wg.Done()
|
||
|
|
rolePermissions := make([]string, 0)
|
||
|
|
// Get role permissions
|
||
|
|
for _, permission := range permissions {
|
||
|
|
if permission.RoleID == role.ID {
|
||
|
|
rolePermissions = append(rolePermissions, permission.Code)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Cache Role
|
||
|
|
entry, _ := json.Marshal(CachedRole{
|
||
|
|
Role: role.Code,
|
||
|
|
IsAdmin: role.IsAdmin,
|
||
|
|
Permissions: rolePermissions,
|
||
|
|
})
|
||
|
|
if err := cache.Set(role.ID.Hex(), entry); err != nil {
|
||
|
|
logger.Error("usermngmt - CacheRole", logger.LogData{
|
||
|
|
"err": err.Error(),
|
||
|
|
})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}(value)
|
||
|
|
}
|
||
|
|
|
||
|
|
wg.Wait()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetCachedRole ...
|
||
|
|
func GetCachedRole(key string) CachedRole {
|
||
|
|
entry, err := cache.Get(key)
|
||
|
|
if err != nil {
|
||
|
|
Roles()
|
||
|
|
entry, _ = cache.Get(key)
|
||
|
|
}
|
||
|
|
var cachedRole CachedRole
|
||
|
|
if err = json.Unmarshal(entry, &cachedRole); err != nil {
|
||
|
|
logger.Error("usermngmt - GetCachedRole - Unmarshal", logger.LogData{
|
||
|
|
"err": err.Error(),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return cachedRole
|
||
|
|
}
|