kami_gateway/internal/models/merchant/merchant_info.go

package merchant

import (
	"context"
	"gateway/internal/otelTrace"

	"time"

	"github.com/beego/beego/v2/client/orm"
	"go.uber.org/zap"
)

type MerchantInfo struct {
	Id                   int
	Status               string
	BelongAgentUid       string
	BelongAgentName      string
	MerchantName         string
	MerchantUid          string
	MerchantKey          string
	MerchantSecret       string
	LoginPassword        string
	LoginAccount         string
	WhiteIps             string
	Remark               string
	SinglePayForRoadUid  string
	SinglePayForRoadName string
	RollPayForRoadCode   string
	RollPayForRoadName   string
	PayforFee            float64
	CreateTime           time.Time
	UpdateTime           time.Time
}

const MERCHANT_INFO = "merchant_info"

func IsExistByMerchantName(merchantName string) bool {
	o := orm.NewOrm()
	exist := o.QueryTable(MERCHANT_INFO).Filter("merchant_name", merchantName).Exist()

	return exist
}

func IsExistByMerchantUid(uid string) bool {
	o := orm.NewOrm()
	exist := o.QueryTable(MERCHANT_INFO).Filter("merchant_uid", uid).Exist()

	return exist
}

func IsExistMerchantByAgentUid(uid string) bool {
	o := orm.NewOrm()
	exist := o.QueryTable(MERCHANT_INFO).Filter("belong_agent_uid", uid).Exist()

	return exist
}

func IsExistByMerchantPhone(phone string) bool {
	o := orm.NewOrm()
	exist := o.QueryTable(MERCHANT_INFO).Filter("LoginAccount", phone).Exist()

	return exist
}

func GetMerchantByPhone(ctx context.Context, phone string) (m MerchantInfo) {
	o := orm.NewOrm()
	_, err := o.QueryTable(MERCHANT_INFO).Filter("LoginAccount", phone).Limit(1).All(&m)
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("GetMerchantByPhone merchant fail: ", zap.Error(err))
	}
	return m
}

func InsertMerchantInfo(ctx context.Context, merchantInfo MerchantInfo) bool {
	o := orm.NewOrm()
	_, err := o.Insert(&merchantInfo)
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("insert merchant fail: ", zap.Error(err))
		return false
	}
	return true
}

func GetMerchantLenByMap(ctx context.Context, params map[string]string) int {
	o := orm.NewOrm()
	qs := o.QueryTable(MERCHANT_INFO)
	for k, v := range params {
		if len(v) > 0 {
			qs = qs.Filter(k, v)
		}
	}
	cnt, err := qs.Count()
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("get merchant len by map fail: ", zap.Error(err))
	}
	return int(cnt)
}

func GetMerchantListByMap(ctx context.Context, params map[string]string, displayCount, offset int) []MerchantInfo {
	o := orm.NewOrm()
	qs := o.QueryTable(MERCHANT_INFO)
	for k, v := range params {
		if len(v) > 0 {
			qs = qs.Filter(k, v)
		}
	}
	var merchantList []MerchantInfo
	_, err := qs.Limit(displayCount, offset).OrderBy("-update_time").All(&merchantList)
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("get merchant list by map fail: ", zap.Error(err))
	}
	return merchantList
}

func GetAllMerchant(ctx context.Context) []MerchantInfo {
	o := orm.NewOrm()
	var merchantList []MerchantInfo

	_, err := o.QueryTable(MERCHANT_INFO).Limit(-1).All(&merchantList)
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("get all merchant fail：", zap.Error(err))
	}

	return merchantList
}

func GetMerchantByParams(ctx context.Context, params map[string]string, displayCount, offset int) []MerchantInfo {
	o := orm.NewOrm()
	var merchantList []MerchantInfo
	qs := o.QueryTable(MERCHANT_INFO)
	for k, v := range params {
		if len(v) > 0 {
			qs = qs.Filter(k, v)
		}
	}

	qs.Limit(displayCount, offset).All(&merchantList)

	return merchantList
}

func GetMerchantLenByParams(ctx context.Context, params map[string]string) int {
	o := orm.NewOrm()
	qs := o.QueryTable(MERCHANT_INFO)
	for k, v := range params {
		if len(v) > 0 {
			qs = qs.Filter(k, v)
		}
	}

	cnt, err := qs.Limit(-1).Count()
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("get merchant len by params fail: ", zap.Error(err))
	}
	return int(cnt)
}

func GetMerchantByUid(ctx context.Context, merchantUid string) MerchantInfo {
	o := orm.NewOrm()
	var merchantInfo MerchantInfo
	_, err := o.QueryTable(MERCHANT_INFO).Filter("merchant_uid", merchantUid).Limit(1).All(&merchantInfo)
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("get merchant info fail: ", zap.Error(err))
	}
	return merchantInfo
}

func GetMerchantByPasskey(ctx context.Context, payKey string) MerchantInfo {
	o := orm.NewOrm()
	var merchantInfo MerchantInfo
	_, err := o.QueryTable(MERCHANT_INFO).Filter("merchant_key", payKey).Limit(1).All(&merchantInfo)
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("get merchant by merchantKey fail: ", zap.Error(err))
	}
	return merchantInfo
}

func UpdateMerchant(ctx context.Context, merchantInfo MerchantInfo) bool {
	o := orm.NewOrm()
	_, err := o.Update(&merchantInfo)
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("update merchant fail: ", zap.Error(err))
		return false
	}

	return true
}

func DeleteMerchantByUid(ctx context.Context, merchantUid string) bool {
	o := orm.NewOrm()
	_, err := o.QueryTable(MERCHANT_INFO).Filter("merchant_uid", merchantUid).Delete()
	if err != nil {
		otelTrace.Logger.WithContext(ctx).Error("delete merchant fail: ", zap.Error(err))
		return false
	}
	return true
}