76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package admin
|
|
|
|
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 StakeByAddressLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// 根据地址修复质押
|
|
func NewStakeByAddressLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StakeByAddressLogic {
|
|
return &StakeByAddressLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *StakeByAddressLogic) StakeByAddress(req *types.StakeByAddressReq) error {
|
|
tokens, err := l.svcCtx.NftHolderModel.FindTokensByAddresses(l.ctx, req.Address)
|
|
if err != nil {
|
|
l.Errorw("find tokens by address failed", logx.Field("err", err), logx.Field("address", req.Address))
|
|
return errs.New(errs.ErrDatabaseOperate, err)
|
|
}
|
|
for _, tk := range tokens {
|
|
uid, err := l.svcCtx.WalletModel.FindUidByAddress(l.ctx, tk.Address)
|
|
if err != nil {
|
|
l.Errorw("find uid by address failed", logx.Field("err", err), logx.Field("address", tk.Address))
|
|
continue
|
|
}
|
|
st, err := l.svcCtx.StakeNftModel.FindOneByTokenId(l.ctx, tk.TokenId)
|
|
if err != nil {
|
|
l.Errorw("find stake nft by token id failed", logx.Field("err", err), logx.Field("tokenId", tk.TokenId))
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
var ty int8
|
|
if utils.IsBigTarot(tk.TokenId) {
|
|
ty = 1
|
|
}
|
|
st = &model.NhTaskNftStake{
|
|
Uid: uid,
|
|
Type: ty,
|
|
TokenId: tk.TokenId,
|
|
State: 1,
|
|
}
|
|
_, err = l.svcCtx.StakeNftModel.Insert(l.ctx, st)
|
|
if err != nil {
|
|
l.Errorw("insert stake nft failed", logx.Field("err", err), logx.Field("stakeNft", st))
|
|
}
|
|
}
|
|
} else {
|
|
if st.Uid != uid {
|
|
st.Uid = uid
|
|
st.State = 1
|
|
err = l.svcCtx.StakeNftModel.Update(l.ctx, st)
|
|
if err != nil {
|
|
l.Errorw("update stake nft failed", logx.Field("err", err), logx.Field("stakeNft", st.TokenId), logx.Field("uid", uid))
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
return errs.Success()
|
|
}
|