v1.1

app/main/model/agentShortLinkModel.go

@@ -1,6 +1,10 @@
 package model
 
 import (
+	"context"
+	"database/sql"
+
+	"github.com/pkg/errors"
 	"github.com/zeromicro/go-zero/core/stores/cache"
 	"github.com/zeromicro/go-zero/core/stores/sqlx"
 )
@@ -12,6 +16,7 @@ type (
 	// and implement the added methods in customAgentShortLinkModel.
 	AgentShortLinkModel interface {
 		agentShortLinkModel
+		FindOneByInviteCodeTypeDelState(ctx context.Context, inviteCode sql.NullString, tp int64, delState int64) (*AgentShortLink, error)
 	}
 
 	customAgentShortLinkModel struct {
@@ -25,3 +30,17 @@ func NewAgentShortLinkModel(conn sqlx.SqlConn, c cache.CacheConf) AgentShortLink
 		defaultAgentShortLinkModel: newAgentShortLinkModel(conn, c),
 	}
 }
+
+func (m *customAgentShortLinkModel) FindOneByInviteCodeTypeDelState(ctx context.Context, inviteCode sql.NullString, tp int64, delState int64) (*AgentShortLink, error) {
+	builder := m.SelectBuilder().
+		Where("invite_code = ? AND type = ? AND del_state = ?", inviteCode, tp, delState).
+		Limit(1)
+	rows, err := m.FindAll(ctx, builder, "")
+	if err != nil {
+		return nil, errors.Wrapf(err, "query short link by invite_code failed")
+	}
+	if len(rows) == 0 {
+		return nil, ErrNotFound
+	}
+	return rows[0], nil
+}