chore: 并入 iotlcm-admin-svr 历史到 admin-svr/(原提交 49818c6)
This commit is contained in:
@@ -0,0 +1,52 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <xxx@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
|
||||
cacheOnce sync.Once
|
||||
)
|
||||
|
||||
// Instance 缓存实例
|
||||
func Instance() *gcache.Cache {
|
||||
cacheOnce.Do(func() {
|
||||
mcache = gcache.NewWithAdapter(gcache.NewAdapterMemory())
|
||||
})
|
||||
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
|
||||
}
|
||||
|
||||
// Clear invalidates every derived in-process cache entry. It is used after the
|
||||
// authoritative SQLite memory database is rebuilt.
|
||||
func Clear(ctx context.Context) error {
|
||||
keys := Instance().MustKeys(ctx)
|
||||
if len(keys) == 0 {
|
||||
return nil
|
||||
}
|
||||
return Instance().Removes(ctx, keys)
|
||||
}
|
||||
@@ -0,0 +1,62 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <xxx@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
|
||||
"time"
|
||||
|
||||
"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)
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
var loadUserPermissions = func(ctx context.Context, userID int64) ([]string, []string, error) {
|
||||
return service.SysAuth().GetUserAccessCodeList(ctx, userID)
|
||||
}
|
||||
|
||||
func GetUserAccessCodeList(ctx context.Context, userId int64) ([]string, []string, error) {
|
||||
codes, dataCodes, err := loadUserPermissions(ctx, userId)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
if len(codes) == 0 {
|
||||
codes = []string{"null"}
|
||||
}
|
||||
return codes, dataCodes, nil
|
||||
}
|
||||
|
||||
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().Removes(ctx, []interface{}{fmt.Sprintf(consts.MemCacheUserRoleDataAccessCodeList, userId)})
|
||||
return err
|
||||
}
|
||||
|
||||
func RemoveUserAccessCodeList(ctx context.Context, userId int64) error {
|
||||
err := Instance().Removes(ctx, []interface{}{fmt.Sprintf(consts.MemCacheUserAccessCodeList, userId)})
|
||||
return err
|
||||
}
|
||||
@@ -0,0 +1,28 @@
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestPermissionsReloadWithoutRevisionTable(t *testing.T) {
|
||||
original := loadUserPermissions
|
||||
t.Cleanup(func() { loadUserPermissions = original })
|
||||
loads := 0
|
||||
loadUserPermissions = func(context.Context, int64) ([]string, []string, error) {
|
||||
loads++
|
||||
if loads == 1 {
|
||||
return []string{"old"}, nil, nil
|
||||
}
|
||||
return []string{"new"}, nil, nil
|
||||
}
|
||||
ctx := context.Background()
|
||||
if _, _, err := GetUserAccessCodeList(ctx, 99); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
codes, _, err := GetUserAccessCodeList(ctx, 99)
|
||||
if err != nil || !reflect.DeepEqual(codes, []string{"new"}) || loads != 2 {
|
||||
t.Fatalf("stale authorization: codes=%v loads=%d err=%v", codes, loads, err)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,77 @@
|
||||
// Package mcache
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2025 LiXiujie
|
||||
// @Author Lxj <xxx@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/contexts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model/request"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
|
||||
"time"
|
||||
)
|
||||
|
||||
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 <xxx@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
|
||||
"time"
|
||||
)
|
||||
|
||||
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 <xxx@example.com>
|
||||
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
||||
package mcache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/event"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
|
||||
"time"
|
||||
|
||||
"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