100 lines
3.3 KiB
Go
Executable File
100 lines
3.3 KiB
Go
Executable File
package model
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlc"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
"strings"
|
|
)
|
|
|
|
var _ NhNftHolderModel = (*customNhNftHolderModel)(nil)
|
|
|
|
type (
|
|
// NhNftHolderModel is an interface to be customized, add more methods here,
|
|
// and implement the added methods in customNhNftHolderModel.
|
|
NhNftHolderModel interface {
|
|
nhNftHolderModel
|
|
withSession(session sqlx.Session) NhNftHolderModel
|
|
FindOtherUpdateSeq(ctx context.Context, updateSeq int) ([]*NhNftHolder, error)
|
|
DeleteOtherUpdateSeq(ctx context.Context, updateSeq int) error
|
|
FindTokensByAddress(ctx context.Context, address string) ([]string, error)
|
|
HoldNft(ctx context.Context, uid uint) bool
|
|
FindTokensByAddresses(ctx context.Context, addresses []string) ([]*NhNftHolder, error)
|
|
}
|
|
|
|
customNhNftHolderModel struct {
|
|
*defaultNhNftHolderModel
|
|
}
|
|
)
|
|
|
|
func (m *customNhNftHolderModel) HoldNft(ctx context.Context, uid uint) bool {
|
|
query := fmt.Sprintf("select count(1) from %s h join nh_wallet w on h.address = w.address where w.`uid` = ?", m.table)
|
|
var count int
|
|
err := m.conn.QueryRowCtx(ctx, &count, query, uid)
|
|
if err != nil {
|
|
if !errors.Is(err, sqlc.ErrNotFound) {
|
|
logx.Errorw("HoldNft query failed", logx.Field("query", query), logx.Field("args", uid), logx.Field("err", err))
|
|
}
|
|
return false
|
|
}
|
|
return count > 0
|
|
}
|
|
|
|
func (m *customNhNftHolderModel) FindTokensByAddress(ctx context.Context, address string) ([]string, error) {
|
|
query := fmt.Sprintf("select `token_id` from %s where `address` = ?", m.table)
|
|
var tokens []string
|
|
err := m.conn.QueryRowsCtx(ctx, &tokens, query, address)
|
|
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
|
|
return nil, err
|
|
}
|
|
return tokens, nil
|
|
}
|
|
|
|
func (m *customNhNftHolderModel) FindTokensByAddresses(ctx context.Context, addresses []string) ([]*NhNftHolder, error) {
|
|
// 生成占位符
|
|
placeholders := make([]string, len(addresses))
|
|
args := make([]interface{}, len(addresses))
|
|
|
|
for i, val := range addresses {
|
|
placeholders[i] = "?"
|
|
args[i] = val
|
|
}
|
|
query := fmt.Sprintf("select %s from %s where `address` in (%s)", nhNftHolderRows, m.table, strings.Join(placeholders, ","))
|
|
var tokens []*NhNftHolder
|
|
err := m.conn.QueryRowsCtx(ctx, &tokens, query, args...)
|
|
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
|
|
return nil, err
|
|
}
|
|
return tokens, nil
|
|
}
|
|
|
|
func (m *customNhNftHolderModel) DeleteOtherUpdateSeq(ctx context.Context, updateSeq int) error {
|
|
delSql := fmt.Sprintf("delete from %s where `update_seq` != ?", m.table)
|
|
_, err := m.conn.ExecCtx(ctx, delSql, updateSeq)
|
|
return err
|
|
}
|
|
|
|
func (m *customNhNftHolderModel) FindOtherUpdateSeq(ctx context.Context, updateSeq int) ([]*NhNftHolder, error) {
|
|
query := fmt.Sprintf("select %s from %s where `update_seq` != ?", nhNftHolderRows, m.table)
|
|
var resp []*NhNftHolder
|
|
err := m.conn.QueryRowsCtx(ctx, &resp, query, updateSeq)
|
|
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
|
|
return nil, err
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
// NewNhNftHolderModel returns a model for the database table.
|
|
func NewNhNftHolderModel(conn sqlx.SqlConn) NhNftHolderModel {
|
|
return &customNhNftHolderModel{
|
|
defaultNhNftHolderModel: newNhNftHolderModel(conn),
|
|
}
|
|
}
|
|
|
|
func (m *customNhNftHolderModel) withSession(session sqlx.Session) NhNftHolderModel {
|
|
return NewNhNftHolderModel(sqlx.NewSqlConnFromSession(session))
|
|
}
|