mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2025-09-12 01:24:38 +08:00
31 lines
620 B
Go
31 lines
620 B
Go
package repo
|
|
|
|
import "gorm.io/gorm"
|
|
|
|
type DBOption func(*gorm.DB) *gorm.DB
|
|
|
|
type ICommonRepo interface {
|
|
WithByID(id uint) DBOption
|
|
WithByName(name string) DBOption
|
|
WithOrderBy(orderStr string) 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) WithOrderBy(orderStr string) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Order(orderStr)
|
|
}
|
|
}
|