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

2026-01-22 18:34:59 +08:00
parent 95d3a20292
commit 652f5e5d60
87 changed files with 15082 additions and 162 deletions
--- a/internal/storage/repository/connection_repo.go
+++ b/internal/storage/repository/connection_repo.go
@@ -0,0 +1,70 @@
+package repository
+
+import (
+	"go-desk/internal/storage"
+	"go-desk/internal/storage/models"
+
+	"gorm.io/gorm"
+)
+
+type ConnectionRepository interface {
+	Save(conn *models.DbConnection) error
+	FindAll() ([]models.DbConnection, error)
+	FindByID(id uint) (*models.DbConnection, error)
+	Delete(id uint) error
+	FindByName(name string, excludeID uint) (*models.DbConnection, error)
+}
+
+type connectionRepository struct {
+	db *gorm.DB
+}
+
+func NewConnectionRepository() (ConnectionRepository, error) {
+	db := storage.GetDB()
+	if db == nil {
+		var err error
+		db, err = storage.Init()
+		if err != nil {
+			return nil, err
+		}
+	}
+	return &connectionRepository{db}, nil
+}
+
+func (r *connectionRepository) Save(conn *models.DbConnection) error {
+	if conn.ID > 0 {
+		return r.db.Model(&models.DbConnection{}).Where("id = ?", conn.ID).Updates(conn).Error
+	}
+	return r.db.Create(conn).Error
+}
+
+func (r *connectionRepository) FindAll() ([]models.DbConnection, error) {
+	var connections []models.DbConnection
+	return connections, r.db.Order("created_at DESC").Find(&connections).Error
+}
+
+func (r *connectionRepository) FindByID(id uint) (*models.DbConnection, error) {
+	var conn models.DbConnection
+	err := r.db.First(&conn, id).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	return &conn, err
+}
+
+func (r *connectionRepository) Delete(id uint) error {
+	return r.db.Delete(&models.DbConnection{}, id).Error
+}
+
+func (r *connectionRepository) FindByName(name string, excludeID uint) (*models.DbConnection, error) {
+	var conn models.DbConnection
+	query := r.db.Where("name = ?", name)
+	if excludeID > 0 {
+		query = query.Where("id != ?", excludeID)
+	}
+	err := query.First(&conn).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	return &conn, err
+}
--- a/internal/storage/repository/result_repo.go
+++ b/internal/storage/repository/result_repo.go
@@ -0,0 +1,110 @@
+package repository
+
+import (
+	"encoding/json"
+	"go-desk/internal/storage"
+	"go-desk/internal/storage/models"
+	"gorm.io/gorm"
+	"time"
+)
+
+type ResultRepository interface {
+	Save(connectionID uint, database, sql string, resultType string, data interface{}, columns []string, rowsAffected int, executionTime int64) (*models.SqlResultHistory, error)
+	FindByID(id uint) (*models.SqlResultHistory, error)
+	FindByConnection(connectionID uint, limit int) ([]models.SqlResultHistory, error)
+	Search(connectionID *uint, keyword string, limit, offset int) ([]models.SqlResultHistory, int64, error)
+	Delete(id uint) error
+	DeleteByConnection(connectionID uint) error
+	DeleteOld(keepDays int) error
+}
+
+type resultRepository struct {
+	db *gorm.DB
+}
+
+func NewResultRepository() (ResultRepository, error) {
+	db := storage.GetDB()
+	if db == nil {
+		var err error
+		db, err = storage.Init()
+		if err != nil {
+			return nil, err
+		}
+	}
+	return &resultRepository{db}, nil
+}
+
+func (r *resultRepository) Save(connectionID uint, database, sql string, resultType string, data interface{}, columns []string, rowsAffected int, executionTime int64) (*models.SqlResultHistory, error) {
+	dataJSON, _ := json.Marshal(data)
+	columnsJSON, _ := json.Marshal(columns)
+
+	history := &models.SqlResultHistory{
+		ConnectionID:  connectionID,
+		Database:      database,
+		Sql:           sql,
+		Type:          resultType,
+		Data:          string(dataJSON),
+		Columns:       string(columnsJSON),
+		RowsAffected:  rowsAffected,
+		ExecutionTime: executionTime,
+	}
+
+	return history, r.db.Create(history).Error
+}
+
+func (r *resultRepository) FindByID(id uint) (*models.SqlResultHistory, error) {
+	var history models.SqlResultHistory
+	err := r.db.First(&history, id).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	return &history, err
+}
+
+func (r *resultRepository) FindByConnection(connectionID uint, limit int) ([]models.SqlResultHistory, error) {
+	var histories []models.SqlResultHistory
+	query := r.db.Where("connection_id = ?", connectionID).Order("created_at DESC")
+	if limit > 0 {
+		query = query.Limit(limit)
+	}
+	return histories, query.Find(&histories).Error
+}
+
+func (r *resultRepository) Search(connectionID *uint, keyword string, limit, offset int) ([]models.SqlResultHistory, int64, error) {
+	query := r.db.Model(&models.SqlResultHistory{})
+
+	if connectionID != nil {
+		query = query.Where("connection_id = ?", *connectionID)
+	}
+	if keyword != "" {
+		query = query.Where("sql LIKE ? OR database LIKE ?", "%"+keyword+"%", "%"+keyword+"%")
+	}
+
+	var total int64
+	if err := query.Count(&total).Error; err != nil {
+		return nil, 0, err
+	}
+
+	var histories []models.SqlResultHistory
+	query = query.Order("created_at DESC")
+	if limit > 0 {
+		query = query.Limit(limit)
+	}
+	if offset > 0 {
+		query = query.Offset(offset)
+	}
+
+	return histories, total, query.Find(&histories).Error
+}
+
+func (r *resultRepository) Delete(id uint) error {
+	return r.db.Delete(&models.SqlResultHistory{}, id).Error
+}
+
+func (r *resultRepository) DeleteByConnection(connectionID uint) error {
+	return r.db.Where("connection_id = ?", connectionID).Delete(&models.SqlResultHistory{}).Error
+}
+
+func (r *resultRepository) DeleteOld(keepDays int) error {
+	return r.db.Where("created_at < ?", time.Now().AddDate(0, 0, -keepDays)).Delete(&models.SqlResultHistory{}).Error
+}
--- a/internal/storage/repository/tab_repo.go
+++ b/internal/storage/repository/tab_repo.go
@@ -0,0 +1,55 @@
+package repository
+
+import (
+	"go-desk/internal/storage"
+	"go-desk/internal/storage/models"
+	"gorm.io/gorm"
+)
+
+type TabRepository interface {
+	SaveAll(tabs []models.SqlTab) error
+	FindAll() ([]models.SqlTab, error)
+	Delete(id uint) error
+	DeleteAll() error
+}
+
+type tabRepository struct {
+	db *gorm.DB
+}
+
+func NewTabRepository() (TabRepository, error) {
+	db := storage.GetDB()
+	if db == nil {
+		var err error
+		db, err = storage.Init()
+		if err != nil {
+			return nil, err
+		}
+	}
+	return &tabRepository{db}, nil
+}
+
+func (r *tabRepository) SaveAll(tabs []models.SqlTab) error {
+	return r.db.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Where("1=1").Delete(&models.SqlTab{}).Error; err != nil {
+			return err
+		}
+		if len(tabs) > 0 {
+			return tx.Create(&tabs).Error
+		}
+		return nil
+	})
+}
+
+func (r *tabRepository) FindAll() ([]models.SqlTab, error) {
+	var tabs []models.SqlTab
+	return tabs, r.db.Order("`order` ASC, created_at ASC").Find(&tabs).Error
+}
+
+func (r *tabRepository) Delete(id uint) error {
+	return r.db.Delete(&models.SqlTab{}, id).Error
+}
+
+func (r *tabRepository) DeleteAll() error {
+	return r.db.Where("1=1").Delete(&models.SqlTab{}).Error
+}