powerCheckCenter/rpc/internal/logic/checklogic.go

package logic

import (
	"context"

	"bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/entity"
	"bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/rpc/internal/svc"
	"bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/rpc/pb"
	"bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/service"

	"github.com/zeromicro/go-zero/core/logx"
)

type CheckLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewCheckLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CheckLogic {
	return &CheckLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

// 用户权限判断
func (l *CheckLogic) Check(in *pb.CheckReq) (*pb.CheckResp, error) {
	// todo: add your logic here and delete this line
	resp := &pb.CheckResp{}
	power := service.NewPower(&entity.Conn{
		Mysql:     entity.Mysql,
		BaseMysql: entity.BaseMysql,
		MgoJy:     entity.MgoJy,
	})
	data := power.Power(in.Userid, in.BaseUserId, in.AccountId, in.EntId, in.PositionType, in.PositionId)
	if data == nil {
		resp.ErrorMsg = "暂无数据"
		resp.ErrorCode = -1
	} else {
		resp.Vip = &pb.Vip{
			Status:          data.Vip.Status,
			StartTime:       data.Vip.StartTime,
			EndTime:         data.Vip.EndTime,
			Upgrade:         data.Vip.Upgrade,
			Areacount:       data.Vip.Areacount,
			Buyerclasscount: data.Vip.Buyerclasscount,
			MaxKeyLength:    data.Vip.MaxKeyLength,
			PowerType:       data.Vip.PowerType,
			VipPower:        data.Vip.VipPower,
			FileNum:         data.Vip.FileNum,
			NewCitys:        data.Vip.NewCitys,
		}
		resp.Member = &pb.Member{
			Status:          data.Member.Status,
			StartTime:       data.Member.StartTime,
			EndTime:         data.Member.EndTime,
			MemberName:      data.Member.MemberName,
			MemberPowerList: data.Member.MemberPowerList,
			IsMemberTrial:   data.Member.IsMemberTrial,
			PowerType:       data.Member.PowerType,
			MemberPower:     data.Member.MemberPower,
			IsSubCount:      data.Member.IsSubCount,
			MaxKeyLength:    data.Member.MaxKeyLength,
			EntNum:          data.Member.EntNum,
			ProNum:          data.Member.ProNum,
			DailyNum:        data.Member.DailyNum,
			Customers:       data.Member.Customers,
			Pid:             data.Member.Pid,
			Used:            data.Member.Used,
		}
		resp.Free = &pb.Free{
			IsFree:        data.Free.IsFree,
			FreeHasKey:    data.Free.FreeHasKey,
			FreeEntPort:   data.Free.FreeEntPort,
			FreeBuyerPort: data.Free.FreeBuyerPort,
			FreeFile:      data.Free.FreeFile,
			Registedate:   data.Free.Registedate,
			IsUpgrade:     data.Free.IsUpgrade,
			Mail:          data.Free.Mail,
			Phone:         data.Free.Phone,
		}
		resp.Ent = &pb.Ent{
			Name:                data.Ent.Name,
			EntRoleId:           data.Ent.EntRoleId,
			EntAuthStatus:       data.Ent.EntAuthStatus,
			EntAuthReason:       data.Ent.EntAuthReason,
			EntSubscribe:        data.Ent.EntSubscribe,
			EntSubscribeManager: data.Ent.EntSubscribeManager,
			PowerSource:         data.Entniche.PowerSource,
			DeptSubscribe:       data.Ent.DeptSubscribe,
			BuyVip:              data.Ent.BuyVip,
			BuyMember:           data.Ent.BuyMember,
		}
		resp.Entniche = &pb.Entniche{
			Status:      data.Entniche.Status,
			StartTime:   data.Entniche.StartTime,
			EndTime:     data.Entniche.EndTime,
			PowerSource: data.Entniche.PowerSource,
			IsEntPower:  data.Entniche.IsEntPower,
			IsNew:       data.Entniche.IsNew,
			Model:       data.Entniche.Model,
		}
	}
	return resp, nil
}