增加统一加资产接口

internal/model/nh_nft_holder_model.go

@@ -4,6 +4,8 @@ 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"
 )
 
@@ -18,6 +20,7 @@ type (
 		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
 	}
 
 	customNhNftHolderModel struct {
@@ -25,6 +28,19 @@ type (
 	}
 )
 
+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

internal/model/nh_task_ambassador_model.go

@@ -1,6 +1,11 @@
 package model
 
-import "github.com/zeromicro/go-zero/core/stores/sqlx"
+import (
+	"context"
+	"errors"
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
 
 var _ NhTaskAmbassadorModel = (*customNhTaskAmbassadorModel)(nil)
 
@@ -10,6 +15,7 @@ type (
 	NhTaskAmbassadorModel interface {
 		nhTaskAmbassadorModel
 		withSession(session sqlx.Session) NhTaskAmbassadorModel
+		IsAmbassador(ctx context.Context, uid uint) bool
 	}
 
 	customNhTaskAmbassadorModel struct {
@@ -17,6 +23,19 @@ type (
 	}
 )
 
+func (m *customNhTaskAmbassadorModel) IsAmbassador(ctx context.Context, uid uint) bool {
+	query := "select t.id from nh_task_ambassador t join nh_user u on t.email = u.email where u.id = ? limit 1"
+	var id uint
+	err := m.conn.QueryRowCtx(ctx, &id, query, uid)
+	if err != nil {
+		if !errors.Is(err, sqlx.ErrNotFound) {
+			logx.Errorw("IsAmbassador query failed", logx.Field("query", query), logx.Field("args", uid), logx.Field("err", err))
+		}
+		return false
+	}
+	return id > 0
+}
+
 // NewNhTaskAmbassadorModel returns a model for the database table.
 func NewNhTaskAmbassadorModel(conn sqlx.SqlConn) NhTaskAmbassadorModel {
 	return &customNhTaskAmbassadorModel{