新增：连接管理、数据查询等功能

2026-01-22 18:34:59 +08:00
parent 95d3a20292
commit 652f5e5d60
87 changed files with 15082 additions and 162 deletions
--- a/internal/api/connection_api.go
+++ b/internal/api/connection_api.go
@@ -0,0 +1,109 @@
+package api
+
+import (
+	"go-desk/internal/service"
+	"go-desk/internal/storage/models"
+)
+
+// ConnectionAPI 连接管理API
+type ConnectionAPI struct {
+	connService *service.ConnectionService
+}
+
+// NewConnectionAPI 创建连接管理API
+func NewConnectionAPI() (*ConnectionAPI, error) {
+	connService, err := service.NewConnectionService()
+	if err != nil {
+		return nil, err
+	}
+	return &ConnectionAPI{connService}, nil
+}
+
+// SaveConnectionRequest 保存连接请求结构体
+type SaveConnectionRequest struct {
+	ID       uint   `json:"id"`
+	Name     string `json:"name"`
+	Type     string `json:"type"`
+	Host     string `json:"host"`
+	Port     int    `json:"port"`
+	Username string `json:"username"`
+	Password string `json:"password"`
+	Database string `json:"database"`
+	Options  string `json:"options"`
+}
+
+// SaveDbConnection 保存数据库连接配置
+func (api *ConnectionAPI) SaveDbConnection(req SaveConnectionRequest) error {
+	conn := &models.DbConnection{
+		ID:       req.ID,
+		Name:     req.Name,
+		Type:     req.Type,
+		Host:     req.Host,
+		Port:     req.Port,
+		Username: req.Username,
+		Password: req.Password,
+		Database: req.Database,
+		Options:  req.Options,
+	}
+	return api.connService.SaveConnection(conn)
+}
+
+// ListDbConnections 获取连接列表
+func (api *ConnectionAPI) ListDbConnections() ([]map[string]interface{}, error) {
+	connections, err := api.connService.ListConnections()
+	if err != nil {
+		return nil, err
+	}
+
+	result := make([]map[string]interface{}, len(connections))
+	timeFormat := "2006-01-02 15:04:05"
+	for i, conn := range connections {
+		result[i] = map[string]interface{}{
+			"id":         conn.ID,
+			"name":       conn.Name,
+			"type":       conn.Type,
+			"host":       conn.Host,
+			"port":       conn.Port,
+			"username":   conn.Username,
+			"database":   conn.Database,
+			"options":    conn.Options,
+			"created_at": conn.CreatedAt.Format(timeFormat),
+			"updated_at": conn.UpdatedAt.Format(timeFormat),
+		}
+	}
+	return result, nil
+}
+
+func (api *ConnectionAPI) DeleteDbConnection(id uint) error {
+	return api.connService.DeleteConnection(id)
+}
+
+func (api *ConnectionAPI) TestDbConnection(id uint) error {
+	return api.connService.TestConnection(id)
+}
+
+// TestConnectionRequest 测试连接请求结构体（不保存数据）
+type TestConnectionRequest struct {
+	ID       uint   `json:"id"` // 编辑模式下的连接ID（用于获取已保存的密码）
+	Type     string `json:"type"`
+	Host     string `json:"host"`
+	Port     int    `json:"port"`
+	Username string `json:"username"`
+	Password string `json:"password"`
+	Database string `json:"database"`
+	Options  string `json:"options"`
+}
+
+// TestDbConnectionWithParams 测试数据库连接（直接传入参数，不保存数据）
+func (api *ConnectionAPI) TestDbConnectionWithParams(req TestConnectionRequest) error {
+	return api.connService.TestConnectionWithParams(
+		req.Type,
+		req.Host,
+		req.Port,
+		req.Username,
+		req.Password,
+		req.Database,
+		req.Options,
+		req.ID,
+	)
+}