usermngmt/cache/role.go

package cache

import (
	"context"
	"encoding/json"
	"log"
	"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 := mc.Set(role.ID.Hex(), entry); err != nil {
				logger.Error("usermngmt - CacheRole", logger.LogData{
					"err": err.Error(),
				})
				return
			}
			log.Println("set role", CachedRole{
				Role:        role.Code,
				IsAdmin:     role.IsAdmin,
				Permissions: rolePermissions,
			}) // TODO remove
		}(value)
	}

	wg.Wait()
	return
}

// GetCachedRole ...
func GetCachedRole(key string) CachedRole {
	entry, err := mc.Get(key)
	if err != nil {
		Roles()
		entry, _ = mc.Get(key)
	}
	var cachedRole CachedRole
	if err = json.Unmarshal(entry, &cachedRole); err != nil {
		logger.Error("usermngmt - GetCachedRole - Unmarshal", logger.LogData{
			"err": err.Error(),
		})
	}
	log.Println("get role", cachedRole) // TODO remove
	return cachedRole
}