54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
|
package repository
|
||
|
|
|
||
|
|
import (
|
||
|
|
"x-agents/server/internal/model"
|
||
|
|
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SubTableRepository struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewSubTableRepository(db *gorm.DB) *SubTableRepository {
|
||
|
|
return &SubTableRepository{db: db}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create 创建子表信息
|
||
|
|
func (r *SubTableRepository) Create(info *model.SubTableInfo) error {
|
||
|
|
return r.db.Create(info).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// FindByID 根据ID查询
|
||
|
|
func (r *SubTableRepository) FindByID(id string) (*model.SubTableInfo, error) {
|
||
|
|
var info model.SubTableInfo
|
||
|
|
if err := r.db.Where("id = ?", id).First(&info).Error; err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &info, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// FindByDatabaseID 根据数据库ID查询所有子表
|
||
|
|
func (r *SubTableRepository) FindByDatabaseID(databaseID string) ([]model.SubTableInfo, error) {
|
||
|
|
var list []model.SubTableInfo
|
||
|
|
if err := r.db.Where("database_id = ?", databaseID).Find(&list).Error; err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return list, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update 更新子表信息
|
||
|
|
func (r *SubTableRepository) Update(id string, info *model.SubTableInfo) error {
|
||
|
|
return r.db.Model(info).Where("id = ?", id).Updates(info).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// Delete 删除子表信息
|
||
|
|
func (r *SubTableRepository) Delete(id string) error {
|
||
|
|
return r.db.Where("id = ?", id).Delete(&model.SubTableInfo{}).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// DeleteByDatabaseID 删除数据库下所有子表信息
|
||
|
|
func (r *SubTableRepository) DeleteByDatabaseID(databaseID string) error {
|
||
|
|
return r.db.Where("database_id = ?", databaseID).Delete(&model.SubTableInfo{}).Error
|
||
|
|
}
|