jyResourcesCenter/api/internal/logic/updateuserbalancelogic.go

package logic

import (
	"app.yhyue.com/moapp/jyResourcesCenter/rpc/resourcesCenterclient"
	"context"
	"fmt"

	"app.yhyue.com/moapp/jyResourcesCenter/api/internal/svc"
	"app.yhyue.com/moapp/jyResourcesCenter/api/internal/types"

	"github.com/tal-tech/go-zero/core/logx"
)

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

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

func (l *UpdateUserBalanceLogic) UpdateUserBalance(req types.ResourcesReq) (*types.ResourcesRes, error) {
	// todo: add your logic here and delete this line
	result := &types.ResourcesRes{}
	fmt.Println(req)
	lsi := l.svcCtx.ResourcesCenter
	var err error
	resp := &resourcesCenterclient.Response{}
	if req.Model == 0 {
		resp, err = lsi.UseUserDetailed(l.ctx, &resourcesCenterclient.Resources{
			AccountId:        req.AccountId,
			CompanyId:        req.CompanyId,
			DepartmentId:     req.DepartmentId,
			Name:             req.Name,
			ResourceType:     req.ResourceType,
			Number:           req.Number,
			Spec:             req.Spec,
			AppId:            req.AppId,
			Model:            req.Model,
			Url:              req.Url,
			SearchCriteria:   req.SearchCriteria,
			Source:           req.Source,
			InfoId:           req.InfoId,
			DuplicateRemoval: req.DuplicateRemoval,
			RuleId:           req.RuleId,
			UserId:           req.UserId,
		})
	} else {
		resp, err = lsi.PurchaseUserBalance(l.ctx, &resourcesCenterclient.Resources{
			AccountId:    req.AccountId,
			CompanyId:    req.CompanyId,
			DepartmentId: req.DepartmentId,
			Name:         req.Name,
			ResourceType: req.ResourceType,
			Number:       req.Number,
			Spec:         req.Spec,
			AppId:        req.AppId,
			EndTime:      req.EndTime,
		})
	}
	if err != nil {
		return nil, err
	}
	result.Code = resp.Code
	result.Message = resp.Message
	return result, nil
}