jyResourcesCenter/api/internal/logic/findbalancelogic.go

package logic

import (
	"app.yhyue.com/moapp/jyResourcesCenter/api/internal/svc"
	"app.yhyue.com/moapp/jyResourcesCenter/api/internal/types"
	"app.yhyue.com/moapp/jyResourcesCenter/rpc/resourcesCenterclient"
	"context"
	"github.com/zeromicro/go-zero/core/logx"
	"log"
)

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

func NewFindBalanceLogic(ctx context.Context, svcCtx *svc.ServiceContext) FindBalanceLogic {
	return FindBalanceLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *FindBalanceLogic) FindBalance(req types.BalanceReq) (*types.BalanceRes, error) {
	// todo: add your logic here and delete this line
	log.Println(req)
	result := &types.BalanceRes{}
	lsi := l.svcCtx.ResourcesCenter
	resp, err := lsi.FindAccountBalance(l.ctx, &resourcesCenterclient.ResourcesReq{
		AccountId:    req.AccountId,
		ResourceType: req.ResourceType,
		ResourceName: req.ResourceName,
		Vipstate:     req.VipState,
	})
	log.Println(resp)
	if err != nil {
		return nil, err
	}
	data := make([]map[string]interface{}, 0)
	if len(resp.Data) > 0 {
		for _, v := range resp.Data {
			data = append(data, map[string]interface{}{
				"id":           v.Id,
				"name":         v.Name,
				"number":       v.Number,
				"resourceType": v.ResourceType,
				"thirtyNum":    v.ThirtyNum,
				"number1":      v.Number1,
				"number2":      v.Number2,
				"number3":      v.Number3,
				"number4":      v.Number4,
			})
		}
	}
	result.Code = resp.Code
	result.Message = resp.Message
	result.Data = data
	return result, nil
}