64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package transfercastile
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"nova_task/internal/model"
|
|
"nova_task/internal/pkg/errs"
|
|
"nova_task/internal/pkg/utils"
|
|
|
|
"nova_task/internal/svc"
|
|
"nova_task/internal/types"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type GetCastileBalanceLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// 查询castile代币余额
|
|
func NewGetCastileBalanceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetCastileBalanceLogic {
|
|
return &GetCastileBalanceLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetCastileBalanceLogic) GetCastileBalance() (resp *types.UserCastileBalanceResp, err error) {
|
|
//获取JWT中的UID
|
|
uid := utils.GetUidUint(l.ctx)
|
|
|
|
u, err := l.svcCtx.UserModel.FindOne(l.ctx, uid)
|
|
if err != nil {
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
return nil, errs.New(errs.ErrUserNotFound, "user not found")
|
|
}
|
|
l.Errorw("find user error", logx.Field("err", err), logx.Field("uid", uid))
|
|
return nil, errs.New(errs.ErrUserNotFound, err)
|
|
}
|
|
|
|
res := &types.UserCastileBalanceResp{
|
|
TotalCastile: 0,
|
|
TransferAmount: 0,
|
|
TotalBalance: 0,
|
|
}
|
|
|
|
cToken, err := l.svcCtx.CastileTokenModel.FindOneByEmail(l.ctx, u.Email)
|
|
if err != nil {
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
return res, nil
|
|
}
|
|
return nil, errs.New(errs.ErrDatabaseOperate, err)
|
|
}
|
|
|
|
return &types.UserCastileBalanceResp{
|
|
TotalCastile: int(cToken.Total),
|
|
TransferAmount: int(cToken.Transfer),
|
|
TotalBalance: int(cToken.Total) - int(cToken.Transfer),
|
|
}, nil
|
|
}
|