增加统一加资产接口

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{