绝尘
e5dbe89a6f
- Markdown 编辑器：实时预览、PDF 导出、独立查看器 - 数据库优化：动态连接池、查询缓存、Redis Pipeline - 窗口置顶功能 - 文件系统增强：右键菜单、编辑器集成、收藏夹重构 - 安全修复：XSS 防护、路径穿越、HTML 注入 - 代码质量：正则预编译、缓存锁优化、死代码清理
129 lines
3.9 KiB
Go
129 lines
3.9 KiB
Go
package api
|
||
|
||
import (
|
||
"u-desk/internal/service"
|
||
"u-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"`
|
||
VisibleDatabases string `json:"visible_databases"`
|
||
}
|
||
|
||
// 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,
|
||
VisibleDatabases: req.VisibleDatabases,
|
||
}
|
||
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,
|
||
"visible_databases": conn.VisibleDatabases,
|
||
"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,
|
||
)
|
||
}
|
||
|
||
// LoadAllDatabasesRequest 加载全部数据库请求结构体
|
||
type LoadAllDatabasesRequest struct {
|
||
ID uint `json:"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"`
|
||
}
|
||
|
||
// LoadAllDatabases 加载全部数据库列表
|
||
func (api *ConnectionAPI) LoadAllDatabases(req LoadAllDatabasesRequest) ([]string, error) {
|
||
return api.connService.LoadAllDatabases(
|
||
req.Type, req.Host, req.Port,
|
||
req.Username, req.Password, req.Database,
|
||
req.Options, req.ID,
|
||
)
|
||
}
|