1Panel/agent/app/repo/app_install_resource.go

package repo

import (
	"context"

	"github.com/1Panel-dev/1Panel/agent/app/model"
	"github.com/1Panel-dev/1Panel/agent/global"
	"gorm.io/gorm"
)

type AppInstallResourceRpo struct {
}

type IAppInstallResourceRpo interface {
	WithAppInstallId(appInstallId uint) DBOption
	WithLinkId(linkId uint) DBOption
	WithResourceId(resourceId uint) DBOption
	GetBy(opts ...DBOption) ([]model.AppInstallResource, error)
	GetFirst(opts ...DBOption) (model.AppInstallResource, error)
	Create(ctx context.Context, resource *model.AppInstallResource) error
	DeleteBy(ctx context.Context, opts ...DBOption) error
	BatchUpdateBy(maps map[string]interface{}, opts ...DBOption) error
}

func NewIAppInstallResourceRpo() IAppInstallResourceRpo {
	return &AppInstallResourceRpo{}
}

func (a AppInstallResourceRpo) WithAppInstallId(appInstallId uint) DBOption {
	return func(db *gorm.DB) *gorm.DB {
		return db.Where("app_install_id = ?", appInstallId)
	}
}

func (a AppInstallResourceRpo) WithLinkId(linkId uint) DBOption {
	return func(db *gorm.DB) *gorm.DB {
		return db.Where("link_id = ?", linkId)
	}
}

func (a AppInstallResourceRpo) WithResourceId(resourceId uint) DBOption {
	return func(db *gorm.DB) *gorm.DB {
		return db.Where("resource_id = ?", resourceId)
	}
}

func (a AppInstallResourceRpo) GetBy(opts ...DBOption) ([]model.AppInstallResource, error) {
	db := global.DB.Model(&model.AppInstallResource{})
	var resources []model.AppInstallResource
	for _, opt := range opts {
		db = opt(db)
	}
	err := db.Find(&resources).Error
	return resources, err
}

func (a AppInstallResourceRpo) GetFirst(opts ...DBOption) (model.AppInstallResource, error) {
	db := global.DB.Model(&model.AppInstallResource{})
	var resources model.AppInstallResource
	for _, opt := range opts {
		db = opt(db)
	}
	err := db.First(&resources).Error
	return resources, err
}

func (a AppInstallResourceRpo) Create(ctx context.Context, resource *model.AppInstallResource) error {
	db := getTx(ctx).Model(&model.AppInstallResource{})
	return db.Create(&resource).Error
}

func (a AppInstallResourceRpo) DeleteBy(ctx context.Context, opts ...DBOption) error {
	return getTx(ctx, opts...).Delete(&model.AppInstallResource{}).Error
}

func (a *AppInstallResourceRpo) BatchUpdateBy(maps map[string]interface{}, opts ...DBOption) error {
	db := getDb(opts...).Model(&model.AppInstallResource{})
	if len(opts) == 0 {
		db = db.Where("1=1")
	}
	return db.Updates(&maps).Error
}