DESKTOP-72TV0V4\caoxiaozhu
b2bc9988a9
- 新增 Go 语言后端服务(server/)，包含用户认证、Agent管理、数据库连接等API - 新增 Python Agent 服务(agent/)，实现Agent核心逻辑和工具集 - 前端从原生HTML迁移到Vue.js框架(web/src/) - 添加 Docker Compose 支持(docker-compose.yml) - 添加项目架构文档(docs/ARCHITECTURE.md) - 添加环境变量示例(.env.example)和本地启动脚本(start-local.ps1) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
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
|
|
}
|