opms_backend/opms_admin/app/service/sys_dict_data.go

package service

import (
	"context"
	"dashoo.cn/micro/app/common/global"
	comModel "dashoo.cn/micro/app/common/model"
	comService "dashoo.cn/micro/app/common/service"
	"dashoo.cn/micro/app/dao"
	"dashoo.cn/micro/app/model"
	"dashoo.cn/opms_libary/micro_srv"
	"database/sql"
	"github.com/gogf/gf/errors/gerror"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/os/glog"
	"github.com/gogf/gf/text/gstr"
	"github.com/gogf/gf/util/gconv"
)

type dictDataService struct {
	Dao *dao.SysDictTypeDao
}

func NewDictDataService(ctx context.Context) (*dictDataService, error) {
	dict := new(dictDataService)
	// 获取租户码
	tenant, err := micro_srv.GetTenant(ctx)
	if err != nil {
		return nil, err
	}
	reqMethod, _ := micro_srv.GetReqMethod(ctx)
	glog.Info("Received " + reqMethod + " request @ " + tenant)
	dict.Dao = dao.NewSysDictTypeDao(tenant)
	return dict, err
}

func (s *dictDataService) GetDictDataList(req *model.SelectDictPageReq) (total int, list []*model.SysDictData, err error) {
	d := dao.SysDictData.Ctx(req.Ctx)
	if req != nil {
		if req.DictLabel != "" {
			d = d.Where(dao.SysDictData.Columns.DictLabel+" like ?", "%"+req.DictLabel+"%")
		}
		if req.Status != "" {
			d = d.Where(dao.SysDictData.Columns.Status+" = ", gconv.Int(req.Status))
		}
		if req.DictType != "" {
			d = d.Where(dao.SysDictData.Columns.DictType+" = ?", req.DictType)
		}
		total, err = d.Count()
		if err != nil {
			g.Log().Error(err)
			err = gerror.New("获取总行数失败")
			return
		}
		if req.PageNum == 0 {
			req.PageNum = 1
		}
	}
	if req.PageSize == 0 {
		req.PageSize = comModel.PageSize
	}
	list, err = d.Page(req.PageNum, req.PageSize).Order(dao.SysDictData.Columns.DictSort + " asc," +
		dao.SysDictData.Columns.DictCode + " asc").All()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取数据失败")
		return
	}
	return
}

// GetDictDataById 通过字典数据id获取字典数据
func (s *dictDataService) GetDictDataById(id int64) (data *model.SysDictData, err error) {
	data, err = dao.SysDictData.FindOne(dao.SysDictData.Columns.DictCode, id)
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取字典数据失败")
		return
	}
	if data == nil {
		err = gerror.New("获取字典数据失败")
	}
	return
}

// GetDictWithDataByType 通过字典键类型获取选项
func (s *dictDataService) GetDictWithDataByType(req *model.GetDictReq) (dict *model.DictRes, err error) {
	cache := comService.Cache.New()
	cacheKey := global.SysDict + "_" + req.DictType
	//从缓存获取
	iDict := cache.Get(cacheKey)
	if iDict != nil {
		err = gconv.Struct(iDict, &dict)
		if err != nil {
			return
		}
	} else {
		//从数据库获取
		dict = &model.DictRes{}
		//获取类型数据
		err = dao.SysDictType.Ctx(req.Ctx).Where(dao.SysDictType.Columns.DictType, req.DictType).
			And(dao.SysDictType.Columns.Status, 1).Fields(model.DictTypeRes{}).Scan(&dict.Info)
		if err != nil {
			g.Log().Error(err)
			err = gerror.New("获取字典类型失败")
		}
		err = dao.SysDictData.Ctx(req.Ctx).Fields(model.DictDataRes{}).
			Where(dao.SysDictData.Columns.DictType, req.DictType).
			Order(dao.SysDictData.Columns.DictSort + " asc," +
				dao.SysDictData.Columns.DictCode + " asc").
			Scan(&dict.Values)
		if err != nil {
			g.Log().Error(err)
			err = gerror.New("获取字典数据失败")
		}
		//缓存菜单
		if dict.Info != nil && dict.Values != nil {
			cache.Set(cacheKey, dict, 0, global.SysDictTag)
		}
	}
	//设置给定的默认值
	for _, v := range dict.Values {
		if req.DefaultValue != "" {
			if gstr.Equal(req.DefaultValue, v.DictValue) {
				v.IsDefault = 1
			} else {
				v.IsDefault = 0
			}
		}
	}
	return
}

// CheckDictTypeUniqueAll 检查字典类型是否唯一
func (s *dictDataService) CheckDictTypeUniqueAll(dictType string) bool {
	dict, err := dao.SysDictData.FindOne(dao.SysDictData.Columns.DictType+"=?", dictType)
	if err != nil {
		g.Log().Error(err)
		return false
	}
	if dict != nil {
		return false
	}
	return true
}

// Create 添加保存字典数据
func (s *dictDataService) Create(req *model.DictDataAddReq) (id int64, err error) {
	var res sql.Result
	data := new(model.SysDictData)
	if err := gconv.Struct(req, data); err != nil {
		return -1, err
	}
	SetCreatedInfo(data, 1, "")
	res, err = dao.SysDictData.Data(data).Insert()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("添加字典数据失败")
		return
	}
	id, err = res.LastInsertId()
	return
}

// UpdateByDict 修改字典数据
func (s *dictDataService) UpdateByDict(req *model.EditDictDataReq) (err error) {
	_, err = dao.SysDictData.FieldsEx(dao.SysDictData.Columns.DictCode, dao.SysDictData.Columns.CreatedBy).
		WherePri(req.DictCode).Update(req)
	return
}

// DeleteDictDataByIds 删除字典数据
func (s *dictDataService) DeleteDictDataByIds(ids []int64) error {
	_, err := dao.SysDictData.Where(dao.SysDictData.Columns.DictCode+" in(?)", ids).Delete()
	if err != nil {
		g.Log().Error(err)
		return gerror.New("删除失败")
	}
	return nil
}