第一次提交

entity/resources.go

@@ -0,0 +1,32 @@
+package entity
+
+type ResourcesAuth struct {
+	Id          int64 `xorm:"pk autoincr id" form:"id" json:"id"`
+	EmployeeId  int64 `xorm:"employeeId" form:"employeeId" json:"employeeId"`    //用户标识
+	ResourcesId int64 `xorm:"resourcesId" form:"resourcesId" json:"resourcesId"` //资源标识
+	Quota       int64 `xorm:"quota" form:"quota" json:"quota"`                   //定额
+	Ratio       int64 `xorm:"ratio" form:"ratio" json:"ratio"`                   //比例
+	State       int64 `xorm:"state" form:"state" json:"state"`                   //状态
+}
+
+type AccountBalance struct {
+	Id         int64  `xorm:"pk autoincr id" form:"id" json:"id"`
+	EmployeeId int64  `xorm:"employeeId" form:"employeeId" json:"employeeId"` //人员标识
+	Name       string `xorm:"name" form:"name" json:"name"`                   //资源名称
+	Number     int64  `xorm:"number" form:"number" json:"number"`             //余额
+	Code       string `xorm:"code" form:"code" json:"code"`                   //资源代码
+	Spec       string `xorm:"spec" form:"spec" json:"spec"`                   //规格
+}
+
+type ConsumeRecord struct {
+	Id            int64  `xorm:"pk autoincr id" form:"id" json:"id"`
+	EmployeeId    int64  `xorm:"employeeId" form:"employeeId" json:"employeeId"`       //人员标识; //用户标识
+	CompanyId     int64  `xorm:"companyId" form:"companyId" json:"companyId"`          //用户标识
+	ResourceType  string `xorm:"resourceType" form:"resourceType" json:"resourceType"` //资源类型
+	ExportNum     int64  `xorm:"exportNum" form:"exportNum" json:"exportNum"`//导出数量
+	DeductionType int64  `xorm:"deductionType" form:"deductionType" json:"deductionType"` //扣费类型(1高级字段包)
+	RuleId        int64  `xorm:"ruleId" form:"ruleId" json:"ruleId"`
+	ExportTime    string `xorm:"exportTime" form:"exportTime" json:"exportTime"`
+	UserType      int64  `xorm:"userType" form:"userType" json:"userType"`
+	DepartmentId  int64  `xorm:"departmentId" form:"departmentId" json:"departmentId"`
+}

rpc/internal/logic/findaccountbalancelogic.go

@@ -0,0 +1,41 @@
+package logic
+
+import (
+	"context"
+
+	"app.yhyue.com/moapp/jyResourcesCenter/rpc/internal/svc"
+	"app.yhyue.com/moapp/jyResourcesCenter/rpc/resourcesCenter"
+
+	"github.com/tal-tech/go-zero/core/logx"
+)
+
+type FindAccountBalanceLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewFindAccountBalanceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *FindAccountBalanceLogic {
+	return &FindAccountBalanceLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *FindAccountBalanceLogic) FindAccountBalance(in *resourcesCenter.AccountBalanceReq) (*resourcesCenter.AccountBalanceRes, error) {
+	// todo: add your logic here and delete this line
+	dataList, err := resourceManageService.FindAccountBalance(in)
+	if len(dataList) == 0 && err != nil {
+		return &resourcesCenter.AccountBalanceRes{
+			Code:    0,
+			Message: "查询出错",
+			Data:    dataList,
+		}, err
+	}
+	return &resourcesCenter.AccountBalanceRes{
+		Code:    1,
+		Message: "查询成功",
+		Data:    dataList,
+	}, nil
+}

rpc/internal/logic/findconsumerecordlogic.go

@@ -0,0 +1,41 @@
+package logic
+
+import (
+	"context"
+
+	"app.yhyue.com/moapp/jyResourcesCenter/rpc/internal/svc"
+	"app.yhyue.com/moapp/jyResourcesCenter/rpc/resourcesCenter"
+
+	"github.com/tal-tech/go-zero/core/logx"
+)
+
+type FindConsumeRecordLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewFindConsumeRecordLogic(ctx context.Context, svcCtx *svc.ServiceContext) *FindConsumeRecordLogic {
+	return &FindConsumeRecordLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *FindConsumeRecordLogic) FindConsumeRecord(in *resourcesCenter.ConsumeRecordReq) (*resourcesCenter.ConsumeRecordRes, error) {
+	// todo: add your logic here and delete this line
+	recordArr, err := resourceManageService.FindConsumeRecord(in)
+	if len(recordArr) == 0 && err != nil {
+		return &resourcesCenter.ConsumeRecordRes{
+			Code:    0,
+			Message: "查询出错",
+			Data:    recordArr,
+		}, err
+	}
+	return &resourcesCenter.ConsumeRecordRes{
+		Code:    1,
+		Message: "查询成功",
+		Data:    recordArr,
+	}, nil
+}

rpc/internal/logic/findresourcesauthlogic.go

@@ -0,0 +1,44 @@
+package logic
+
+import (
+	"app.yhyue.com/moapp/jyResourcesCenter/service"
+	"context"
+
+	"app.yhyue.com/moapp/jyResourcesCenter/rpc/internal/svc"
+	"app.yhyue.com/moapp/jyResourcesCenter/rpc/resourcesCenter"
+
+	"github.com/tal-tech/go-zero/core/logx"
+)
+
+var resourceManageService = service.ResourceManageService{}
+
+type FindResourcesAuthLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewFindResourcesAuthLogic(ctx context.Context, svcCtx *svc.ServiceContext) *FindResourcesAuthLogic {
+	return &FindResourcesAuthLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *FindResourcesAuthLogic) FindResourcesAuth(in *resourcesCenter.ResourcesAuthReq) (*resourcesCenter.ResourcesAuthRes, error) {
+	// todo: add your logic here and delete this line
+	authArr, err := resourceManageService.FindResourcesAuth(in)
+	if len(authArr) == 0 && err != nil {
+		return &resourcesCenter.ResourcesAuthRes{
+			Code:    0,
+			Message: "查询出错",
+			Data:    authArr,
+		}, err
+	}
+	return &resourcesCenter.ResourcesAuthRes{
+		Code:    1,
+		Message: "查询成功",
+		Data:    authArr,
+	}, nil
+}

rpc/internal/server/resourcescenterserver.go

@@ -26,3 +26,21 @@ func (s *ResourcesCenterServer) FindCompanyOrgan(ctx context.Context, in *resour
 	l := logic.NewFindCompanyOrganLogic(ctx, s.svcCtx)
 	return l.FindCompanyOrgan(in)
 }
+
+// 查询账户资源权限
+func (s *ResourcesCenterServer) FindResourcesAuth(ctx context.Context, in *resourcesCenter.ResourcesAuthReq) (*resourcesCenter.ResourcesAuthRes, error) {
+	l := logic.NewFindResourcesAuthLogic(ctx, s.svcCtx)
+	return l.FindResourcesAuth(in)
+}
+
+// 查询账户资源余额
+func (s *ResourcesCenterServer) FindAccountBalance(ctx context.Context, in *resourcesCenter.AccountBalanceReq) (*resourcesCenter.AccountBalanceRes, error) {
+	l := logic.NewFindAccountBalanceLogic(ctx, s.svcCtx)
+	return l.FindAccountBalance(in)
+}
+
+// 查询流水账
+func (s *ResourcesCenterServer) FindConsumeRecord(ctx context.Context, in *resourcesCenter.ConsumeRecordReq) (*resourcesCenter.ConsumeRecordRes, error) {
+	l := logic.NewFindConsumeRecordLogic(ctx, s.svcCtx)
+	return l.FindConsumeRecord(in)
+}

rpc/resourcesCenter.proto

@@ -1,51 +1,126 @@
 syntax = "proto3";
 
 package resourcesCenter;
-option go_package="resourcesCenter/";
+//option go_package = "resourcesCenter/";
 
-message OrganRequest{
-   string   name = 1; //组织名称
-   string   shortName = 2;  //组织简称
-   string   contactPerson = 3;  //组织联系人
-   string   contactPhone= 4;//组织联系人手机号
-   int64    parentId=5;//上级部门ID
-   int64    companyId=6;//企业信息ID
-   int64    id = 7;     //组织ID
+message OrganRequest {
+    string name = 1; //组织名称
+    string shortName = 2; //组织简称
+    string contactPerson = 3; //组织联系人
+    string contactPhone = 4; //组织联系人手机号
+    int64 parentId = 5; //上级部门ID
+    int64 companyId = 6; //企业信息ID
+    int64 id = 7; //组织ID
 }
 message CompanyOrganResponse {
-   int64    code =1;            //响应代码
-   string   message=2;          //响应消息
-   repeated CompanyOrgan data=3;
+    int64 code = 1; //响应代码
+    string message = 2; //响应消息
+    repeated CompanyOrgan data = 3;
 }
 message CompanyOrgan {
-   string  name = 1;         //组织名称
-   string  shortName = 2;    //组织简称
-   string  contactPerson = 3;//组织联系人
-   string  contactPhone= 4;  //组织联系人手机号
-   int64   parentId=5;       //上级部门ID
-   int64   companyId=6;      //企业信息ID
-   bool    state = 7;
-   string  createAt = 8;
-   string  updateAt = 9;
-   string  deleteAt = 10;
-   int64   id=11;
- }
- message Organ {
-   string  name = 1;         //组织名称
-   string  shortName = 2;    //组织简称
-   string  contactPerson = 3;//组织联系人
-   string  contactPhone= 4;  //组织联系人手机号
-   int64   parentId=5;       //上级部门ID
-   int64   companyId=6;      //企业信息ID
-   bool    state = 7;
-   string  createAt = 8;
-   string  updateAt = 9;
-   string  deleteAt = 10;
-   int64   id=11;
-   repeated Organ data=12;   //列表
- }
+    string name = 1; //组织名称
+    string shortName = 2; //组织简称
+    string contactPerson = 3; //组织联系人
+    string contactPhone = 4; //组织联系人手机号
+    int64 parentId = 5; //上级部门ID
+    int64 companyId = 6; //企业信息ID
+    bool state = 7;
+    string createAt = 8;
+    string updateAt = 9;
+    string deleteAt = 10;
+    int64 id = 11;
+}
+message Organ {
+    string name = 1; //组织名称
+    string shortName = 2; //组织简称
+    string contactPerson = 3; //组织联系人
+    string contactPhone = 4; //组织联系人手机号
+    int64 parentId = 5; //上级部门ID
+    int64 companyId = 6; //企业信息ID
+    bool state = 7;
+    string createAt = 8;
+    string updateAt = 9;
+    string deleteAt = 10;
+    int64 id = 11;
+    repeated Organ data = 12; //列表
+}
+
+message ResourcesAuth {
+    int64 id = 1;
+    int64 employeeId = 2; //用户标识
+    int64 resourcesId = 3; //资源标识
+    int64 quota = 4; //定额
+    int64 ratio = 5; //比例
+    int64 state = 6; //状态
+}
+
+message ResourceBalance {
+    int64 id = 1;
+    int64 employeeId = 2; //人员标识
+    string name = 3; //资源名称
+    int64 number = 4; //剩余数量
+    string code = 5; //资源代码
+    string spec = 6; //规格
+}
+
+message ConsumeRecord {
+    int64 id = 1;
+    int64 employeeId = 2; //用户标识
+    int64 companyId = 3; //用户标识
+    string resourceType = 4;
+    int64 exportNum = 5;
+    int64 deductionType = 6;
+    int64 ruleId = 7;
+    string exportTime = 8;
+    int64 userType = 9;
+    int64 departmentId = 10;
+}
+
+//根据账户标识查询资源权限请求参数
+message ResourcesAuthReq {
+    int64 employeeId = 1;
+}
+
+message ResourcesAuthRes {
+    int64 code = 1; //响应代码
+    string message = 2; //响应消息
+    repeated ResourcesAuth data = 3; //账号拥有的资源权限
+}
+
+//根据账户标识查询账户余额
+message AccountBalanceReq {
+    int64 employeeId = 1; //用户标识
+    int64 companyId = 2; //公司标识
+    int64 departmentId = 3; //部门标识
+}
+
+message AccountBalanceRes {
+    int64 code = 1;
+    string message = 2;
+    repeated ResourceBalance data = 3;
+}
+
+//查询流水账
+message ConsumeRecordReq {
+    int64 employeeId = 1;
+    int64 companyId = 2;
+}
+
+message ConsumeRecordRes {
+    int64 code = 1;
+    string message = 2;
+    repeated ConsumeRecord data = 3;
+}
+
+
 service ResourcesCenter {
 
-  //查询用户组织信息
-  rpc findCompanyOrgan(OrganRequest) returns(CompanyOrganResponse);
+    //查询用户组织信息
+    rpc findCompanyOrgan (OrganRequest) returns (CompanyOrganResponse);
+    //查询账户资源权限
+    rpc findResourcesAuth (ResourcesAuthReq) returns (ResourcesAuthRes);
+    //查询账户资源余额
+    rpc findAccountBalance (AccountBalanceReq) returns (AccountBalanceRes);
+    //查询流水账
+    rpc findConsumeRecord (ConsumeRecordReq) returns (ConsumeRecordRes);
 }

rpc/resourcesCenterclient/resourcescenter.go

@@ -1,9 +1,9 @@
 // Code generated by goctl. DO NOT EDIT!
 // Source: resourcesCenter.proto
 
-//go:generate mockgen -destination ./resourcescenter_mock.go -package resourcescenter -source $GOFILE
+//go:generate mockgen -destination ./resourcescenter_mock.go -package resourcescenterclient -source $GOFILE
 
-package resourcesCenterclient
+package resourcescenterclient
 
 import (
 	"context"
@@ -14,14 +14,29 @@ import (
 )
 
 type (
-	OrganRequest         = resourcesCenter.OrganRequest
-	CompanyOrganResponse = resourcesCenter.CompanyOrganResponse
 	CompanyOrgan         = resourcesCenter.CompanyOrgan
 	Organ                = resourcesCenter.Organ
+	ResourcesAuth        = resourcesCenter.ResourcesAuth
+	ResourceBalance      = resourcesCenter.ResourceBalance
+	ConsumeRecord        = resourcesCenter.ConsumeRecord
+	ResourcesAuthReq     = resourcesCenter.ResourcesAuthReq
+	ResourcesAuthRes     = resourcesCenter.ResourcesAuthRes
+	AccountBalanceRes    = resourcesCenter.AccountBalanceRes
+	ConsumeRecordReq     = resourcesCenter.ConsumeRecordReq
+	OrganRequest         = resourcesCenter.OrganRequest
+	CompanyOrganResponse = resourcesCenter.CompanyOrganResponse
+	AccountBalanceReq    = resourcesCenter.AccountBalanceReq
+	ConsumeRecordRes     = resourcesCenter.ConsumeRecordRes
 
 	ResourcesCenter interface {
 		// 查询用户组织信息
 		FindCompanyOrgan(ctx context.Context, in *OrganRequest) (*CompanyOrganResponse, error)
+		// 查询账户资源权限
+		FindResourcesAuth(ctx context.Context, in *ResourcesAuthReq) (*ResourcesAuthRes, error)
+		// 查询账户资源余额
+		FindAccountBalance(ctx context.Context, in *AccountBalanceReq) (*AccountBalanceRes, error)
+		// 查询流水账
+		FindConsumeRecord(ctx context.Context, in *ConsumeRecordReq) (*ConsumeRecordRes, error)
 	}
 
 	defaultResourcesCenter struct {
@@ -40,3 +55,21 @@ func (m *defaultResourcesCenter) FindCompanyOrgan(ctx context.Context, in *Organ
 	client := resourcesCenter.NewResourcesCenterClient(m.cli.Conn())
 	return client.FindCompanyOrgan(ctx, in)
 }
+
+// 查询账户资源权限
+func (m *defaultResourcesCenter) FindResourcesAuth(ctx context.Context, in *ResourcesAuthReq) (*ResourcesAuthRes, error) {
+	client := resourcesCenter.NewResourcesCenterClient(m.cli.Conn())
+	return client.FindResourcesAuth(ctx, in)
+}
+
+// 查询账户资源余额
+func (m *defaultResourcesCenter) FindAccountBalance(ctx context.Context, in *AccountBalanceReq) (*AccountBalanceRes, error) {
+	client := resourcesCenter.NewResourcesCenterClient(m.cli.Conn())
+	return client.FindAccountBalance(ctx, in)
+}
+
+// 查询流水账
+func (m *defaultResourcesCenter) FindConsumeRecord(ctx context.Context, in *ConsumeRecordReq) (*ConsumeRecordRes, error) {
+	client := resourcesCenter.NewResourcesCenterClient(m.cli.Conn())
+	return client.FindConsumeRecord(ctx, in)
+}

rpc/test/resourcesRecord_test.go

@@ -0,0 +1,6 @@
+package test
+
+/*func Test_FindResourcesAuth(t *testing.T){
+	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
+	FileSystem := resourcesCenterclient.NewUserCenter(zrpc.MustNewClient(c.FileSystemConf))
+}*/

service/resourceManageService.go

@@ -0,0 +1,70 @@
+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
+}