70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"github.com/go-sql-driver/mysql"
|
|
"github.com/spf13/cast"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"nova_task/internal/model"
|
|
"nova_task/internal/pkg/errs"
|
|
"nova_task/internal/pkg/utils"
|
|
"nova_task/internal/svc"
|
|
"nova_task/internal/types"
|
|
)
|
|
|
|
type StakeNftLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewStakeNftLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StakeNftLogic {
|
|
return &StakeNftLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *StakeNftLogic) StakeNft(req *types.StakeNftList) error {
|
|
uid := utils.GetUidUint(l.ctx)
|
|
err := l.svcCtx.StakeNftModel.StakeNft(l.ctx, uid, req.RoleId, req.TokenIds)
|
|
if err != nil {
|
|
l.Errorw("stake nft failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("tokenIds", req.TokenIds))
|
|
return errs.New(errs.ErrDatabaseOperate, err)
|
|
}
|
|
|
|
for _, tokenId := range req.TokenIds {
|
|
var sns []int8
|
|
var propertyId string
|
|
if utils.IsBigTarot(tokenId) {
|
|
sns = []int8{1, 2}
|
|
propertyId = "402505"
|
|
} else {
|
|
sns = []int8{1}
|
|
propertyId = "402605"
|
|
}
|
|
for _, sn := range sns {
|
|
_, err = l.svcCtx.StakePropertyModel.Insert(l.ctx, &model.NhNftStakeProperty{
|
|
Uid: uid,
|
|
RoleId: int64(req.RoleId),
|
|
TokenId: cast.ToUint(tokenId),
|
|
PropertyId: "",
|
|
Sn: sn,
|
|
})
|
|
if err != nil {
|
|
errMySQL := new(mysql.MySQLError)
|
|
if errors.As(err, &errMySQL) {
|
|
switch errMySQL.Number {
|
|
case 1062:
|
|
default:
|
|
l.Errorw("insert stake property failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("tokenId", tokenId), logx.Field("propertyId", propertyId))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return errs.Success()
|
|
}
|