chore: 并入 iotlcm-app 历史到 app/(原提交 f40bda9)
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <li@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
// 内存缓存管理
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
)
|
||||
|
||||
// cache 缓存驱动
|
||||
var (
|
||||
mcache *gcache.Cache
|
||||
mu sync.Mutex
|
||||
)
|
||||
|
||||
// Instance 缓存实例
|
||||
func Instance() *gcache.Cache {
|
||||
if mcache == nil {
|
||||
mu.Lock()
|
||||
if mcache == nil {
|
||||
mcache = gcache.NewWithAdapter(gcache.NewAdapterMemory())
|
||||
}
|
||||
mu.Unlock()
|
||||
}
|
||||
return mcache
|
||||
}
|
||||
|
||||
func Set(ctx context.Context, key string, value interface{}, timeout time.Duration) error {
|
||||
return Instance().Set(ctx, key, value, timeout)
|
||||
}
|
||||
|
||||
func Get(ctx context.Context, key string) (interface{}, error) {
|
||||
return Instance().Get(ctx, key)
|
||||
}
|
||||
|
||||
func Remove(ctx context.Context, key string) error {
|
||||
_, err := Instance().Remove(ctx, key)
|
||||
return err
|
||||
}
|
||||
@@ -0,0 +1,97 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <li@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/service"
|
||||
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
)
|
||||
|
||||
func init() {
|
||||
event.EventsInstance().Register(consts.EventKeyUserLogout, func(ctx context.Context, eventKey string, args ...interface{}) {
|
||||
if len(args) == 0 {
|
||||
return
|
||||
}
|
||||
userId := args[0].(int64)
|
||||
RemoveUserAccessCodeList(ctx, userId)
|
||||
g.Log().Infof(ctx, "mcache_access_list.go EventKeyUserLogout userId: %d", userId)
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
type requestAccessKey struct{}
|
||||
|
||||
type requestAccessSnapshot struct {
|
||||
userId int64
|
||||
codes []string
|
||||
scopes []string
|
||||
}
|
||||
|
||||
// WithFreshUserAccess 为小程序请求读取当前权限并保留请求内快照。
|
||||
// 控制台清理本地缓存不会清理小程序进程内的权限缓存。
|
||||
func WithFreshUserAccess(ctx context.Context, userId int64) (context.Context, error) {
|
||||
codes, scopes, err := service.SysAuth().GetUserAccessCodeList(ctx, userId)
|
||||
if err != nil {
|
||||
return ctx, err
|
||||
}
|
||||
snapshot := requestAccessSnapshot{userId: userId, codes: ensureCurrentUserAccessCode(codes), scopes: scopes}
|
||||
return context.WithValue(ctx, requestAccessKey{}, snapshot), nil
|
||||
}
|
||||
|
||||
func GetUserAccessCodeList(ctx context.Context, userId int64) ([]string, []string, error) {
|
||||
if snapshot, ok := ctx.Value(requestAccessKey{}).(requestAccessSnapshot); ok && snapshot.userId == userId {
|
||||
return snapshot.codes, snapshot.scopes, nil
|
||||
}
|
||||
accessCodeList, err := Instance().Get(ctx, fmt.Sprintf(consts.MemCacheUserAccessCodeList, userId))
|
||||
menuRoleDataAccessCodeList, err := Instance().Get(ctx, fmt.Sprintf(consts.MemCacheUserRoleDataAccessCodeList, userId))
|
||||
if err != nil || accessCodeList == nil {
|
||||
accessCodeCurrList, menuRoleDataAccessCodeList, err := service.SysAuth().GetUserAccessCodeList(ctx, userId)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
accessCodeCurrList = ensureCurrentUserAccessCode(accessCodeCurrList)
|
||||
if len(accessCodeCurrList) == 0 {
|
||||
accessCodeCurrList = []string{"null"}
|
||||
}
|
||||
Instance().Set(ctx, fmt.Sprintf(consts.MemCacheUserAccessCodeList, userId), accessCodeCurrList, time.Hour*24)
|
||||
Instance().Set(ctx, fmt.Sprintf(consts.MemCacheUserRoleDataAccessCodeList, userId), menuRoleDataAccessCodeList, time.Hour*24)
|
||||
return accessCodeCurrList, menuRoleDataAccessCodeList, nil
|
||||
}
|
||||
return ensureCurrentUserAccessCode(accessCodeList.Strings()), menuRoleDataAccessCodeList.Strings(), nil
|
||||
}
|
||||
|
||||
func ensureCurrentUserAccessCode(accessCodeList []string) []string {
|
||||
for _, code := range accessCodeList {
|
||||
if code == consts.SysCheckPermissionCurrentUser {
|
||||
return accessCodeList
|
||||
}
|
||||
}
|
||||
return append(accessCodeList, consts.SysCheckPermissionCurrentUser)
|
||||
}
|
||||
|
||||
func SetUserAccessCodeList(ctx context.Context, userId int64, accessCodeList []string) error {
|
||||
return Instance().Set(ctx, fmt.Sprintf(consts.MemCacheUserAccessCodeList, userId), accessCodeList, time.Hour*24)
|
||||
}
|
||||
|
||||
func SetUserRoleDataAccessCodeList(ctx context.Context, userId int64, roleDataAccessCodeList []string) error {
|
||||
return Instance().Set(ctx, fmt.Sprintf(consts.MemCacheUserRoleDataAccessCodeList, userId), roleDataAccessCodeList, time.Hour*24)
|
||||
}
|
||||
|
||||
func RemoveUserRoleDataAccessCodeList(ctx context.Context, userId int64) error {
|
||||
_, err := Instance().Remove(ctx, fmt.Sprintf(consts.MemCacheUserRoleDataAccessCodeList, userId))
|
||||
return err
|
||||
}
|
||||
|
||||
func RemoveUserAccessCodeList(ctx context.Context, userId int64) error {
|
||||
_, err := Instance().Remove(ctx, fmt.Sprintf(consts.MemCacheUserAccessCodeList, userId))
|
||||
return err
|
||||
}
|
||||
@@ -0,0 +1,19 @@
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestRequestAccessSnapshotOverridesProcessCache(t *testing.T) {
|
||||
for _, codes := range [][]string{{"cpm:iot:tSimCardAllocate:batch:submit"}, {}} {
|
||||
ctx := context.WithValue(context.Background(), requestAccessKey{}, requestAccessSnapshot{
|
||||
userId: 123, codes: ensureCurrentUserAccessCode(codes), scopes: []string{"scope-current"},
|
||||
})
|
||||
got, scopes, err := GetUserAccessCodeList(ctx, 123)
|
||||
if err != nil || !reflect.DeepEqual(got, ensureCurrentUserAccessCode(codes)) || !reflect.DeepEqual(scopes, []string{"scope-current"}) {
|
||||
t.Fatalf("request snapshot was not respected: codes=%v scopes=%v err=%v", got, scopes, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,77 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <li@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/contexts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/model"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/model/request"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/service"
|
||||
)
|
||||
|
||||
func init() {
|
||||
event.EventsInstance().Register([]string{consts.EventKeyDBSysConfigUpdate, consts.EventKeyDBSysConfigCreate, consts.EventKeyDBSysConfigDelete}, func(ctx context.Context, eventKey string, args ...interface{}) {
|
||||
if len(args) < 1 {
|
||||
return
|
||||
}
|
||||
// tenantId := args[0].(string)
|
||||
Instance().Remove(ctx, getKey(ctx))
|
||||
})
|
||||
}
|
||||
func getKey(ctx context.Context) string {
|
||||
return fmt.Sprintf(consts.MemCacheSystemConfig, contexts.GetTenantId(ctx))
|
||||
}
|
||||
|
||||
func GetSystemConfig(ctx context.Context, key string, defaultValue string) (string, error) {
|
||||
mapV, err := Instance().Get(ctx, getKey(ctx))
|
||||
if err != nil {
|
||||
data, _, err := service.SysConfig().List(ctx, &model.SysConfigListParam{
|
||||
PageInfo: request.PageInfo{
|
||||
Page: 1,
|
||||
PageSize: 9999,
|
||||
},
|
||||
})
|
||||
if err != nil {
|
||||
return defaultValue, err
|
||||
}
|
||||
mapConfig := make(map[string]string)
|
||||
for _, item := range data {
|
||||
mapConfig[item.ConfigKey] = item.ConfigValue
|
||||
}
|
||||
Instance().Set(ctx, getKey(ctx), mapConfig, time.Hour*24)
|
||||
return mapConfig[key], nil
|
||||
}
|
||||
mapConfig := mapV.MapStrStr()
|
||||
if v, ok := mapConfig[key]; ok {
|
||||
return v, nil
|
||||
}
|
||||
return defaultValue, nil
|
||||
}
|
||||
|
||||
func SetSystemConfig(ctx context.Context, key string, value string) error {
|
||||
mapV, err := Instance().Get(ctx, getKey(ctx))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
mapConfig := mapV.MapStrStr()
|
||||
mapConfig[key] = value
|
||||
return Instance().Set(ctx, getKey(ctx), mapConfig, time.Hour*24)
|
||||
}
|
||||
|
||||
func RemoveSystemConfig(ctx context.Context, key string) error {
|
||||
mapV, err := Instance().Get(ctx, getKey(ctx))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
mapConfig := mapV.MapStrStr()
|
||||
delete(mapConfig, key)
|
||||
return Instance().Set(ctx, getKey(ctx), mapConfig, time.Hour*24)
|
||||
}
|
||||
@@ -0,0 +1,49 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <li@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/service"
|
||||
)
|
||||
|
||||
func init() {
|
||||
event.EventsInstance().Register([]string{consts.EventKeyDBSysDeptUpdate,
|
||||
consts.EventKeyDBSysDeptCreate,
|
||||
consts.EventKeyDBSysDeptDelete}, func(ctx context.Context, eventKey string, args ...interface{}) {
|
||||
if len(args) == 0 {
|
||||
return
|
||||
}
|
||||
deptId := args[0].(int64)
|
||||
RemoveDeptName(ctx, deptId)
|
||||
})
|
||||
}
|
||||
|
||||
func GetDeptName(ctx context.Context, deptId int64) (string, error) {
|
||||
deptName, err := Instance().Get(ctx, fmt.Sprintf(consts.MemCacheDeptName, deptId))
|
||||
if err != nil {
|
||||
dept, err := service.SysDept().GetDeptById(ctx, deptId)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
Instance().Set(ctx, fmt.Sprintf(consts.MemCacheDeptName, deptId), dept.DeptName, time.Hour*24)
|
||||
return dept.DeptName, nil
|
||||
}
|
||||
return deptName.String(), nil
|
||||
}
|
||||
|
||||
func SetDeptName(ctx context.Context, deptId int64, deptName string) error {
|
||||
return Instance().Set(ctx, fmt.Sprintf(consts.MemCacheDeptName, deptId), deptName, time.Hour*24)
|
||||
}
|
||||
|
||||
func RemoveDeptName(ctx context.Context, deptId int64) error {
|
||||
_, err := Instance().Remove(ctx, fmt.Sprintf(consts.MemCacheDeptName, deptId))
|
||||
return err
|
||||
}
|
||||
@@ -0,0 +1,58 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <li@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/model"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-app/internal/service"
|
||||
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
)
|
||||
|
||||
func init() {
|
||||
event.EventsInstance().Register(consts.EventKeyUserUpdate, func(ctx context.Context, eventKey string, args ...interface{}) {
|
||||
if len(args) == 0 {
|
||||
return
|
||||
}
|
||||
userId := args[0].(int64)
|
||||
RemoveUserInfo(ctx, userId)
|
||||
})
|
||||
}
|
||||
|
||||
func GetUserInfo(ctx context.Context, userId int64) (*model.SysUserMiniModel, error) {
|
||||
userInfo, err := Instance().Get(ctx, fmt.Sprintf(consts.MemCacheUserInfo, userId))
|
||||
if err != nil {
|
||||
user, err := service.SysUser().GetUserById(ctx, userId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
miniUser := &model.SysUserMiniModel{}
|
||||
gconv.Struct(miniUser, user)
|
||||
Instance().Set(ctx, fmt.Sprintf(consts.MemCacheUserInfo, userId), miniUser, time.Hour*24)
|
||||
return miniUser, nil
|
||||
}
|
||||
miniUser := &model.SysUserMiniModel{}
|
||||
err = userInfo.Struct(miniUser)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return miniUser, nil
|
||||
}
|
||||
|
||||
func SetUserInfo(ctx context.Context, userId int64, userInfo *model.SysUserMiniModel) error {
|
||||
return Instance().Set(ctx, fmt.Sprintf(consts.MemCacheUserInfo, userId), userInfo, time.Hour*24)
|
||||
}
|
||||
|
||||
func RemoveUserInfo(ctx context.Context, userId int64) error {
|
||||
_, err := Instance().Remove(ctx, fmt.Sprintf(consts.MemCacheUserInfo, userId))
|
||||
return err
|
||||
}
|
||||
Reference in New Issue
Block a user