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

internal/storage/sqlite.go

@@ -0,0 +1,57 @@
+package storage
+
+import (
+	"go-desk/internal/storage/models"
+	"os"
+	"path/filepath"
+	"time"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+var globalDB *gorm.DB
+
+func Init() (*gorm.DB, error) {
+	if globalDB != nil {
+		return globalDB, nil
+	}
+
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		return nil, err
+	}
+
+	dataDir := filepath.Join(homeDir, ".go-desk")
+	os.MkdirAll(dataDir, 0755)
+
+	dbPath := filepath.Join(dataDir, "db-cli.db")
+
+	db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{
+		Logger: logger.Default.LogMode(logger.Silent),
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	sqlDB, _ := db.DB()
+	sqlDB.SetMaxOpenConns(1)
+	sqlDB.SetMaxIdleConns(1)
+	sqlDB.SetConnMaxLifetime(time.Hour)
+
+	if err := db.AutoMigrate(
+		&models.DbConnection{},
+		&models.SqlTab{},
+		&models.SqlResultHistory{},
+	); err != nil {
+		return nil, err
+	}
+
+	globalDB = db
+	return globalDB, nil
+}
+
+func GetDB() *gorm.DB {
+	return globalDB
+}