64 lines
2.0 KiB
Go
Executable File
64 lines
2.0 KiB
Go
Executable File
package model
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
)
|
|
|
|
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)
|
|
}
|
|
|
|
customNhNftHolderModel struct {
|
|
*defaultNhNftHolderModel
|
|
}
|
|
)
|
|
|
|
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) 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))
|
|
}
|