mattermost/store/local_cache_supplier_roles.go

// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package store

import (
	"context"

	"github.com/mattermost/mattermost-server/model"
)

func (s *LocalCacheSupplier) handleClusterInvalidateRole(msg *model.ClusterMessage) {
	if msg.Data == CLEAR_CACHE_MESSAGE_DATA {
		s.roleCache.Purge()
	} else {
		s.roleCache.Remove(msg.Data)
	}
}

func (s *LocalCacheSupplier) RoleSave(ctx context.Context, role *model.Role, hints ...LayeredStoreHint) (*model.Role, *model.AppError) {
	if len(role.Name) != 0 {
		defer s.doInvalidateCacheCluster(s.roleCache, role.Name)
	}
	return s.Next().RoleSave(ctx, role, hints...)
}

func (s *LocalCacheSupplier) RoleGet(ctx context.Context, roleId string, hints ...LayeredStoreHint) (*model.Role, *model.AppError) {
	// Roles are cached by name, as that is most commonly how they are looked up.
	// This means that no caching is supported on roles being looked up by ID.
	return s.Next().RoleGet(ctx, roleId, hints...)
}

func (s *LocalCacheSupplier) RoleGetAll(ctx context.Context, hints ...LayeredStoreHint) ([]*model.Role, *model.AppError) {
	// Roles are cached by name, as that is most commonly how they are looked up.
	// This means that no caching is supported on roles being listed.
	return s.Next().RoleGetAll(ctx, hints...)
}

func (s *LocalCacheSupplier) RoleGetByName(ctx context.Context, name string, hints ...LayeredStoreHint) (*model.Role, *model.AppError) {
	if result := s.doStandardReadCache(ctx, s.roleCache, name, hints...); result != nil {
		return result.Data.(*model.Role), nil
	}

	role, err := s.Next().RoleGetByName(ctx, name, hints...)
	if err != nil {
		return nil, err
	}

	result := NewSupplierResult()
	result.Data = role
	s.doStandardAddToCache(ctx, s.roleCache, name, result, hints...)

	return role, nil
}

func (s *LocalCacheSupplier) RoleGetByNames(ctx context.Context, roleNames []string, hints ...LayeredStoreHint) ([]*model.Role, *model.AppError) {
	var foundRoles []*model.Role
	var rolesToQuery []string

	for _, roleName := range roleNames {
		if result := s.doStandardReadCache(ctx, s.roleCache, roleName, hints...); result != nil {
			foundRoles = append(foundRoles, result.Data.(*model.Role))
		} else {
			rolesToQuery = append(rolesToQuery, roleName)
		}
	}

	rolesFound, err := s.Next().RoleGetByNames(ctx, rolesToQuery, hints...)

	for _, role := range rolesFound {
		res := NewSupplierResult()
		res.Data = role
		s.doStandardAddToCache(ctx, s.roleCache, role.Name, res, hints...)
	}
	foundRoles = append(foundRoles, rolesFound...)

	return foundRoles, err
}

func (s *LocalCacheSupplier) RoleDelete(ctx context.Context, roleId string, hints ...LayeredStoreHint) (*model.Role, *model.AppError) {
	role, err := s.Next().RoleDelete(ctx, roleId, hints...)
	if err != nil {
		return nil, err
	}

	s.doInvalidateCacheCluster(s.roleCache, role.Name)

	return role, nil
}

func (s *LocalCacheSupplier) RolePermanentDeleteAll(ctx context.Context, hints ...LayeredStoreHint) *model.AppError {
	defer s.roleCache.Purge()
	defer s.doClearCacheCluster(s.roleCache)

	return s.Next().RolePermanentDeleteAll(ctx, hints...)
}