76 lines
2.6 KiB
Go
Executable File
76 lines
2.6 KiB
Go
Executable File
package model
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
)
|
|
|
|
var _ NhTaskNftStakeModel = (*customNhTaskNftStakeModel)(nil)
|
|
|
|
type (
|
|
// NhTaskNftStakeModel is an interface to be customized, add more methods here,
|
|
// and implement the added methods in customNhTaskNftStakeModel.
|
|
NhTaskNftStakeModel interface {
|
|
nhTaskNftStakeModel
|
|
withSession(session sqlx.Session) NhTaskNftStakeModel
|
|
StakeNft(ctx context.Context, uid uint, roleId uint64, tokens []string) error
|
|
UnStakeNft(ctx context.Context, uid uint, token string) error
|
|
FindByUid(ctx context.Context, uid uint) ([]NhTaskNftStake, error)
|
|
AllStakeNft(ctx context.Context) ([]NhTaskNftStake, error)
|
|
}
|
|
|
|
customNhTaskNftStakeModel struct {
|
|
*defaultNhTaskNftStakeModel
|
|
}
|
|
)
|
|
|
|
func (m *customNhTaskNftStakeModel) AllStakeNft(ctx context.Context) ([]NhTaskNftStake, error) {
|
|
query := fmt.Sprintf("SELECT %s FROM %s WHERE `state` = 1", nhTaskNftStakeRows, m.table)
|
|
var result []NhTaskNftStake
|
|
err := m.conn.QueryRowsCtx(ctx, &result, query)
|
|
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
|
|
return nil, err
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (m *customNhTaskNftStakeModel) FindByUid(ctx context.Context, uid uint) ([]NhTaskNftStake, error) {
|
|
query := fmt.Sprintf("SELECT %s FROM %s WHERE `uid` = ?", nhTaskNftStakeRows, m.table)
|
|
var list []NhTaskNftStake
|
|
err := m.conn.QueryRowsCtx(ctx, &list, query, uid)
|
|
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
|
|
return nil, err
|
|
}
|
|
return list, nil
|
|
}
|
|
|
|
func (m *customNhTaskNftStakeModel) UnStakeNft(ctx context.Context, uid uint, token string) error {
|
|
update := fmt.Sprintf("UPDATE %s SET `state` = 0 WHERE `uid` = ? AND `token_id` = ?", m.table)
|
|
_, err := m.conn.ExecCtx(ctx, update, uid, token)
|
|
return err
|
|
}
|
|
|
|
func (m *customNhTaskNftStakeModel) StakeNft(ctx context.Context, uid uint, roleId uint64, tokens []string) error {
|
|
insertOrUpdate := fmt.Sprintf("INSERT INTO %s (`uid`, `token_id`, `role_id`, `state`) VALUES (?, ?, ?, 1) ON DUPLICATE KEY UPDATE `role_id` = ?, `state` = 1", m.table)
|
|
for _, token := range tokens {
|
|
_, err := m.conn.ExecCtx(ctx, insertOrUpdate, uid, token, roleId, roleId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// NewNhTaskNftStakeModel returns a model for the database table.
|
|
func NewNhTaskNftStakeModel(conn sqlx.SqlConn) NhTaskNftStakeModel {
|
|
return &customNhTaskNftStakeModel{
|
|
defaultNhTaskNftStakeModel: newNhTaskNftStakeModel(conn),
|
|
}
|
|
}
|
|
|
|
func (m *customNhTaskNftStakeModel) withSession(session sqlx.Session) NhTaskNftStakeModel {
|
|
return NewNhTaskNftStakeModel(sqlx.NewSqlConnFromSession(session))
|
|
}
|