mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-09-23 00:55:59 +08:00
90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package repo
|
|
|
|
import (
|
|
"context"
|
|
"github.com/1Panel-dev/1Panel/app/model"
|
|
"github.com/1Panel-dev/1Panel/global"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type AppInstallRepo struct{}
|
|
|
|
func (a AppInstallRepo) WithDetailIdsIn(detailIds []uint) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("app_detail_id in (?)", detailIds)
|
|
}
|
|
}
|
|
func (a AppInstallRepo) WithAppId(appId uint) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("app_id = ?", appId)
|
|
}
|
|
}
|
|
func (a AppInstallRepo) WithStatus(status string) DBOption {
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
return g.Where("status = ?", status)
|
|
}
|
|
}
|
|
|
|
func (a AppInstallRepo) GetBy(opts ...DBOption) ([]model.AppInstall, error) {
|
|
db := global.DB.Model(&model.AppInstall{})
|
|
for _, opt := range opts {
|
|
db = opt(db)
|
|
}
|
|
var install []model.AppInstall
|
|
err := db.Preload("App").Preload("Containers").Find(&install).Error
|
|
return install, err
|
|
}
|
|
|
|
func (a AppInstallRepo) GetFirst(opts ...DBOption) (model.AppInstall, error) {
|
|
db := global.DB.Model(&model.AppInstall{})
|
|
for _, opt := range opts {
|
|
db = opt(db)
|
|
}
|
|
var install model.AppInstall
|
|
err := db.Preload("App").Preload("Containers").First(&install).Error
|
|
return install, err
|
|
}
|
|
|
|
func (a AppInstallRepo) Create(ctx context.Context, install *model.AppInstall) error {
|
|
db := ctx.Value("db").(*gorm.DB).Model(&model.AppInstall{})
|
|
return db.Create(&install).Error
|
|
}
|
|
|
|
func (a AppInstallRepo) Save(install model.AppInstall) error {
|
|
db := global.DB
|
|
return db.Save(&install).Error
|
|
}
|
|
|
|
func (a AppInstallRepo) DeleteBy(opts ...DBOption) error {
|
|
db := global.DB.Model(&model.AppInstall{})
|
|
for _, opt := range opts {
|
|
db = opt(db)
|
|
}
|
|
return db.Delete(&model.AppInstall{}).Error
|
|
}
|
|
|
|
func (a AppInstallRepo) Delete(ctx context.Context, install model.AppInstall) error {
|
|
db := ctx.Value("db").(*gorm.DB).Model(&model.AppInstall{})
|
|
return db.Delete(&install).Error
|
|
}
|
|
|
|
func (a AppInstallRepo) Page(page, size int, opts ...DBOption) (int64, []model.AppInstall, error) {
|
|
var apps []model.AppInstall
|
|
db := global.DB.Model(&model.AppInstall{})
|
|
for _, opt := range opts {
|
|
db = opt(db)
|
|
}
|
|
count := int64(0)
|
|
db = db.Count(&count)
|
|
err := db.Debug().Limit(size).Offset(size * (page - 1)).Preload("App").Preload("Containers").Find(&apps).Error
|
|
return count, apps, err
|
|
}
|
|
|
|
func (a AppInstallRepo) BatchUpdateBy(update model.AppInstall, opts ...DBOption) error {
|
|
db := global.DB.Model(model.AppInstall{})
|
|
for _, opt := range opts {
|
|
db = opt(db)
|
|
}
|
|
return db.Updates(update).Error
|
|
}
|