feat: 重构前后端架构,添加Go后端和Python Agent服务
- 新增 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>
This commit is contained in:
48
server/internal/repository/agent_repo.go
Normal file
48
server/internal/repository/agent_repo.go
Normal file
@@ -0,0 +1,48 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"x-agents/server/internal/model"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type AgentRepository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewAgentRepository(db *gorm.DB) *AgentRepository {
|
||||
return &AgentRepository{db: db}
|
||||
}
|
||||
|
||||
func (r *AgentRepository) Create(agent *model.Agent) error {
|
||||
return r.db.Create(agent).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindByID(id string) (*model.Agent, error) {
|
||||
var agent model.Agent
|
||||
err := r.db.First(&agent, "id = ?", id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &agent, nil
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindByOwnerID(ownerID string) ([]model.Agent, error) {
|
||||
var agents []model.Agent
|
||||
err := r.db.Where("owner_id = ?", ownerID).Find(&agents).Error
|
||||
return agents, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindAll() ([]model.Agent, error) {
|
||||
var agents []model.Agent
|
||||
err := r.db.Where("is_active = ?", true).Find(&agents).Error
|
||||
return agents, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) Update(agent *model.Agent) error {
|
||||
return r.db.Save(agent).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) Delete(id string) error {
|
||||
return r.db.Delete(&model.Agent{}, "id = ?", id).Error
|
||||
}
|
||||
56
server/internal/repository/audit_repo.go
Normal file
56
server/internal/repository/audit_repo.go
Normal file
@@ -0,0 +1,56 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"x-agents/server/internal/model"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type AuditRepository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewAuditRepository(db *gorm.DB) *AuditRepository {
|
||||
return &AuditRepository{db: db}
|
||||
}
|
||||
|
||||
func (r *AuditRepository) Create(log *model.AuditLog) error {
|
||||
return r.db.Create(log).Error
|
||||
}
|
||||
|
||||
func (r *AuditRepository) FindByUserID(userID string, limit int) ([]model.AuditLog, error) {
|
||||
var logs []model.AuditLog
|
||||
err := r.db.Where("user_id = ?", userID).Order("created_at DESC").Limit(limit).Find(&logs).Error
|
||||
return logs, err
|
||||
}
|
||||
|
||||
func (r *AuditRepository) FindByAgentID(agentID string, limit int) ([]model.AuditLog, error) {
|
||||
var logs []model.AuditLog
|
||||
err := r.db.Where("agent_id = ?", agentID).Order("created_at DESC").Limit(limit).Find(&logs).Error
|
||||
return logs, err
|
||||
}
|
||||
|
||||
// ToolApproval 工具审批仓储
|
||||
|
||||
func (r *AuditRepository) CreateApproval(req *model.ToolApprovalRequest) error {
|
||||
return r.db.Create(req).Error
|
||||
}
|
||||
|
||||
func (r *AuditRepository) FindApprovalByID(id string) (*model.ToolApprovalRequest, error) {
|
||||
var req model.ToolApprovalRequest
|
||||
err := r.db.First(&req, "id = ?", id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
|
||||
func (r *AuditRepository) FindPendingApprovals() ([]model.ToolApprovalRequest, error) {
|
||||
var reqs []model.ToolApprovalRequest
|
||||
err := r.db.Where("status = ?", model.ApprovalStatusPending).Order("created_at ASC").Find(&reqs).Error
|
||||
return reqs, err
|
||||
}
|
||||
|
||||
func (r *AuditRepository) UpdateApproval(req *model.ToolApprovalRequest) error {
|
||||
return r.db.Save(req).Error
|
||||
}
|
||||
47
server/internal/repository/database_repo.go
Normal file
47
server/internal/repository/database_repo.go
Normal file
@@ -0,0 +1,47 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"x-agents/server/internal/model"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type DatabaseRepository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewDatabaseRepository(db *gorm.DB) *DatabaseRepository {
|
||||
return &DatabaseRepository{db: db}
|
||||
}
|
||||
|
||||
// Create 创建数据库信息
|
||||
func (r *DatabaseRepository) Create(info *model.DatabaseInfo) error {
|
||||
return r.db.Create(info).Error
|
||||
}
|
||||
|
||||
// FindByID 根据ID查询
|
||||
func (r *DatabaseRepository) FindByID(id string) (*model.DatabaseInfo, error) {
|
||||
var info model.DatabaseInfo
|
||||
err := r.db.First(&info, "id = ?", id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &info, nil
|
||||
}
|
||||
|
||||
// FindAll 查询所有
|
||||
func (r *DatabaseRepository) FindAll() ([]model.DatabaseInfo, error) {
|
||||
var list []model.DatabaseInfo
|
||||
err := r.db.Order("created_at DESC").Find(&list).Error
|
||||
return list, err
|
||||
}
|
||||
|
||||
// Update 更新
|
||||
func (r *DatabaseRepository) Update(id string, info *model.DatabaseInfo) error {
|
||||
return r.db.Model(&model.DatabaseInfo{}).Where("id = ?", id).Updates(info).Error
|
||||
}
|
||||
|
||||
// Delete 删除
|
||||
func (r *DatabaseRepository) Delete(id string) error {
|
||||
return r.db.Delete(&model.DatabaseInfo{}, "id = ?", id).Error
|
||||
}
|
||||
53
server/internal/repository/sub_table_repo.go
Normal file
53
server/internal/repository/sub_table_repo.go
Normal file
@@ -0,0 +1,53 @@
|
||||
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
|
||||
}
|
||||
66
server/internal/repository/user_repo.go
Normal file
66
server/internal/repository/user_repo.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"x-agents/server/internal/model"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type UserRepository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewUserRepository(db *gorm.DB) *UserRepository {
|
||||
return &UserRepository{db: db}
|
||||
}
|
||||
|
||||
func (r *UserRepository) Create(user *model.User) error {
|
||||
return r.db.Create(user).Error
|
||||
}
|
||||
|
||||
func (r *UserRepository) FindByID(id string) (*model.User, error) {
|
||||
var user model.User
|
||||
err := r.db.Preload("Role").First(&user, "id = ?", id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &user, nil
|
||||
}
|
||||
|
||||
func (r *UserRepository) FindByUsername(username string) (*model.User, error) {
|
||||
var user model.User
|
||||
err := r.db.Preload("Role").First(&user, "username = ?", username).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &user, nil
|
||||
}
|
||||
|
||||
func (r *UserRepository) FindAll() ([]model.User, error) {
|
||||
var users []model.User
|
||||
err := r.db.Preload("Role").Find(&users).Error
|
||||
return users, err
|
||||
}
|
||||
|
||||
func (r *UserRepository) Update(user *model.User) error {
|
||||
return r.db.Save(user).Error
|
||||
}
|
||||
|
||||
func (r *UserRepository) Delete(id string) error {
|
||||
return r.db.Delete(&model.User{}, "id = ?", id).Error
|
||||
}
|
||||
|
||||
// FindRoleByID 根据ID查找角色
|
||||
func (r *UserRepository) FindRoleByID(id string) (*model.Role, error) {
|
||||
var role model.Role
|
||||
err := r.db.First(&role, "id = ?", id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &role, nil
|
||||
}
|
||||
|
||||
// CreateRole 创建角色
|
||||
func (r *UserRepository) CreateRole(role *model.Role) error {
|
||||
return r.db.Create(role).Error
|
||||
}
|
||||
Reference in New Issue
Block a user