ERP_user/models/Account/Menu.go

package Account

import (
	"ERP_user/conf"
	"encoding/json"
	"fmt"
	menulibs "git.baozhida.cn/ERP_libs/Menu"
	"git.baozhida.cn/ERP_libs/lib"
	"github.com/astaxie/beego/cache"
	_ "github.com/astaxie/beego/cache/redis"
	"github.com/astaxie/beego/logs"
	"github.com/beego/beego/v2/adapter/orm"
	"time"
)

var redisCache_API cache.Cache
var redisCache_Menu cache.Cache

func init() {
	//注册模型
	orm.RegisterModel(new(menulibs.Menu))
	orm.RegisterModel(new(menulibs.API))

	var err error

	menuConfig := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
		"redis_User_Menu", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
	fmt.Println(menuConfig)
	redisCache_Menu, err = cache.NewCache("redis", menuConfig)
	if err != nil || redisCache_Menu == nil {
		errMsg := "failed to init redis menu"
		logs.Error(errMsg, err)
		panic(errMsg)
	}

	apiConfig := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
		"redis_ERP_USER_API", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
	fmt.Println(apiConfig)
	redisCache_API, err = cache.NewCache("redis", apiConfig)
	if err != nil || redisCache_API == nil {
		errMsg := "failed to init redis api"
		logs.Error(errMsg, err)
		panic(errMsg)
	}
}

func Redis_Munu_Set(T_power_id string, r []menulibs.Menu) (err error) {
	//json序列化
	str, err := json.Marshal(r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	err = redisCache_Menu.Put(T_power_id, str, 24*time.Hour)
	if err != nil {
		logs.Error("set key:", T_power_id+",value:", str, err)
	}
	return
}

func Redis_Menu_Get(T_power_id string) (r []menulibs.Menu, is bool) {
	if redisCache_Menu.IsExist(T_power_id) {
		//println("找到key:",key)
		v := redisCache_Menu.Get(T_power_id)

		err := json.Unmarshal(v.([]byte), &r)
		if err != nil {
			logs.Error(lib.FuncName(), err)
			return r, false
		}
		return r, true
	}
	return r, false
}

func Redis_Menu_DelK(T_power_id string) (err error) {
	err = redisCache_Menu.Delete(T_power_id)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}
	return
}

func Read_Menu_List() ([]menulibs.Menu, error) {
	o := orm.NewOrm()
	MenuDao := menulibs.NewMenu(o)
	m, err := MenuDao.Read_Menu_List()
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return nil, err
	}

	return m, nil
}

// Read_User_Bind_Menu_List 获取用户绑定的菜单列表
func Read_User_Bind_Menu_List(T_Menu string) ([]menulibs.Menu, error) {
	o := orm.NewOrm()
	MenuDao := menulibs.NewMenu(o)
	m, err := MenuDao.Read_Menu_List_ByPower_T_Menu(T_Menu)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return nil, err
	}

	return m, nil
}

// 获取列表
func Read_API_List_ByPower(T_power_id string, Menu_Bind string) (maps []menulibs.API) {
	if len(Menu_Bind) == 0 {
		return
	}
	o := orm.NewOrm()
	APIDao := menulibs.NewAPI(o, redisCache_API)
	maps, err := APIDao.Read_API_List_ByPower(T_power_id, Menu_Bind)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}
	return maps
}