mirror of https://github.com/1Panel-dev/1Panel
121 lines
2.5 KiB
Go
121 lines
2.5 KiB
Go
package repo
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/constant"
|
|
"github.com/1Panel-dev/1Panel/backend/global"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type DBOption func(*gorm.DB) *gorm.DB
|
|
|
|
type ICommonRepo interface {
|
|
WithByID(id uint) DBOption
|
|
WithByName(name string) DBOption
|
|
WithByType(tp string) DBOption
|
|
WithOrderBy(orderStr string) DBOption
|
|
WithByGroupID(groupID uint) DBOption
|
|
WithLikeName(name string) DBOption
|
|
WithIdsIn(ids []uint) DBOption
|
|
WithByDate(startTime, endTime time.Time) DBOption
|
|
WithByStartDate(startTime time.Time) DBOption
|
|
}
|
|
|
|
type CommonRepo struct{}
|
|
|
|
func (c *CommonRepo) WithByID(id uint) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("id = ?", id)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithByName(name string) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("name = ?", name)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithByDate(startTime, endTime time.Time) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("start_time > ? AND start_time < ?", startTime, endTime)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithByStartDate(startTime time.Time) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("start_time < ?", startTime)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithByType(tp string) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("type = ?", tp)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithByGroupID(groupID uint) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
if groupID == 0 {
|
|
return g
|
|
}
|
|
return g.Where("group_id = ?", groupID)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithByStatus(status string) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
if len(status) == 0 {
|
|
return g
|
|
}
|
|
return g.Where("status = ?", status)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithLikeName(name string) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
if len(name) == 0 {
|
|
return g
|
|
}
|
|
return g.Where("name like ?", "%"+name+"%")
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithOrderBy(orderStr string) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Order(orderStr)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithIdsIn(ids []uint) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("id in (?)", ids)
|
|
}
|
|
}
|
|
|
|
func (c *CommonRepo) WithIdsNotIn(ids []uint) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("id not in (?)", ids)
|
|
}
|
|
}
|
|
|
|
func getTx(ctx context.Context, opts ...DBOption) *gorm.DB {
|
|
tx, ok := ctx.Value(constant.DB).(*gorm.DB)
|
|
if ok {
|
|
for _, opt := range opts {
|
|
tx = opt(tx)
|
|
}
|
|
return tx
|
|
}
|
|
return getDb(opts...)
|
|
}
|
|
|
|
func getDb(opts ...DBOption) *gorm.DB {
|
|
db := global.DB
|
|
for _, opt := range opts {
|
|
db = opt(db)
|
|
}
|
|
return db
|
|
}
|