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 }