jyResourcesCenter/service/resourceManageService.go

package service

import (
	"app.yhyue.com/moapp/jyResourcesCenter/entity"
	"app.yhyue.com/moapp/jyResourcesCenter/rpc/resourcesCenter"
)

type ResourceManageService struct{}

//查询用户资源权限
func (service *ResourceManageService) FindResourcesAuth(data *resourcesCenter.ResourcesAuthReq) ([]*resourcesCenter.ResourcesAuth, error) {
	orm := entity.Engine.NewSession()
	var authArr []*entity.ResourcesAuth
	var authList []*resourcesCenter.ResourcesAuth
	err := orm.Table("resources_authorization").Where("accountId = ?", data.EmployeeId).Find(&authArr)
	if err != nil {
		return authList, err
	}
	for _, v := range authArr {
		authList = append(authList,&resourcesCenter.ResourcesAuth{
			Id:v.Id,
			EmployeeId:v.EmployeeId,
			ResourcesId:v.ResourcesId,
			Quota:v.Quota,
			State:v.State,
			Ratio:v.Ratio,
		})
	}
	return authList, nil
}

//查询账户资源余额
func (service *ResourceManageService) FindAccountBalance(in *resourcesCenter.AccountBalanceReq)([]*resourcesCenter.ResourceBalance,error){
	orm := entity.Engine.NewSession()
	var accountBalanceArr []*entity.AccountBalance
	var dataList []*resourcesCenter.ResourceBalance
	err := orm.Table("account_resources").Where("employeeId = ?", in.EmployeeId).Find(&accountBalanceArr)
	if err != nil {
		return dataList, err
	}
	for _, v := range accountBalanceArr {
		dataList = append(dataList,&resourcesCenter.ResourceBalance{
			Id:v.Id,
			EmployeeId:v.EmployeeId,
			Name:v.Name,
			Code:v.Code,
			Number:v.Number,
			Spec:v.Spec,
		})
	}
	return dataList, nil
}

//查询账户资源余额
func (service *ResourceManageService) FindConsumeRecord(in *resourcesCenter.ConsumeRecordReq)([]*resourcesCenter.ConsumeRecord,error){
	orm := entity.Engine.NewSession()
	var recordArr []*entity.ConsumeRecord
	var dataList []*resourcesCenter.ConsumeRecord
	err := orm.Table("consume_record").Where("employeeId = ?", in.EmployeeId).Find(&recordArr)
	if err != nil {
		return dataList, err
	}
	for _, v := range recordArr {
		dataList = append(dataList,&resourcesCenter.ConsumeRecord{
			Id:v.Id,

		})
	}
	return dataList, nil
}