70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
package payment
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type customPaymentLogicModel interface {
|
|
FindOneByPaymentToken(ctx context.Context, token string) (*Payment, error)
|
|
FindAll(ctx context.Context) ([]*Payment, error)
|
|
FindListByPage(ctx context.Context, page, size int, req *Filter) (int64, []*Payment, error)
|
|
FindAvailableMethods(ctx context.Context) ([]*Payment, error)
|
|
}
|
|
|
|
// NewModel returns a model for the database table.
|
|
func NewModel(conn *gorm.DB, c *redis.Client) Model {
|
|
return &customPaymentModel{
|
|
defaultPaymentModel: newPaymentModel(conn, c),
|
|
}
|
|
}
|
|
|
|
func (m *customPaymentModel) FindOneByPaymentToken(ctx context.Context, token string) (*Payment, error) {
|
|
var resp *Payment
|
|
key := cachePaymentTokenPrefix + token
|
|
err := m.QueryCtx(ctx, &resp, key, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Payment{}).Where("token = ?", token).First(v).Error
|
|
})
|
|
return resp, err
|
|
}
|
|
|
|
func (m *customPaymentModel) FindAll(ctx context.Context) ([]*Payment, error) {
|
|
var resp []*Payment
|
|
err := m.QueryNoCacheCtx(ctx, &resp, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Payment{}).Find(v).Error
|
|
})
|
|
return resp, err
|
|
}
|
|
|
|
func (m *customPaymentModel) FindAvailableMethods(ctx context.Context) ([]*Payment, error) {
|
|
var resp []*Payment
|
|
err := m.QueryNoCacheCtx(ctx, &resp, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Payment{}).Where("enable = ?", true).Find(v).Error
|
|
})
|
|
return resp, err
|
|
}
|
|
|
|
func (m *customPaymentModel) FindListByPage(ctx context.Context, page, size int, req *Filter) (int64, []*Payment, error) {
|
|
var resp []*Payment
|
|
var total int64
|
|
err := m.QueryNoCacheCtx(ctx, &resp, func(conn *gorm.DB, v interface{}) error {
|
|
conn = conn.Model(&Payment{})
|
|
if req != nil {
|
|
if req.Enable != nil {
|
|
conn = conn.Where("`enable` = ?", *req.Enable)
|
|
}
|
|
if req.Mark != "" {
|
|
conn = conn.Where("`mark` = ?", req.Mark)
|
|
}
|
|
if req.Search != "" {
|
|
conn = conn.Where("`name` LIKE ?", "%"+req.Search+"%")
|
|
}
|
|
}
|
|
|
|
return conn.Count(&total).Offset((page - 1) * size).Limit(size).Find(v).Error
|
|
})
|
|
return total, resp, err
|
|
}
|