Merge pull request #91 from aice030/feat/77

Feat/77: 实现发布系统各内部方法

Author: Erickw87

docs/deploy/README.md

@@ -126,9 +126,9 @@ type InstanceManager interface {
 func ValidatePackageURL(packageURL string) error
 ```
 
-### 5.2 **GetServiceInstanceIDs**: 根据服务名和版本获取实例ID列表
+### 5.2 **GetServiceInstanceInfos**: 根据服务名和版本获取实例信息列表
 ```go
-func GetServiceInstanceIDs(serviceName string, version ...string) ([]string, error)
+func GetServiceInstanceInfos(serviceName string, version ...string) ([]*InstanceInfo, error)
 ```
 
 ### 5.3 **GetInstanceIP**: 根据实例ID获取实例的IP地址
@@ -143,7 +143,7 @@ func GetInstancePort(instanceID string) (int, error)
 
 ### 5.5 **CheckInstanceHealth**: 检查实例的健康状态
 ```go
-func CheckInstanceHealth(instanceID string) (bool, error)
+func CheckInstanceHealth(instanceIP string, instancePort int) (bool, error)
 ```
 
 ### 5.6 **GetAvailableHosts**: 获取所有可用的主机列表

docs/deploy/interface.md

@@ -120,10 +120,10 @@ type InstanceManager interface {
 **InstanceInfo结构体**:
 ```go
 type InstanceInfo struct {
-    InstanceID  string `json:"instance_id"`  // 实例唯一标识符
-    ServiceName string `json:"service_name"` // 所属服务名称
-    Version     string `json:"version"`      // 当前运行的版本号
-    Status      string `json:"status"`       // 实例运行状态 - 'active'运行中；'pending'发布中；'error'出现故障
+    InstanceID     string `json:"instance_id"`     // 实例唯一标识符
+    ServiceName    string `json:"service_name"`   // 所属服务名称
+    ServiceVersion string `json:"service_version"` // 当前运行的版本号
+    Status         string `json:"status"`          // 实例运行状态 - 'active'运行中；'pending'发布中；'error'出现故障
 }
 ```
 
@@ -187,13 +187,13 @@ packageURL string // 必填，包下载URL
 
 **返回结果**: `error` - 验证失败时返回错误信息
 
-### 4.2 GetServiceInstanceIDs函数
+### 4.2 GetServiceInstanceInfos函数
 
-**函数描述**: 根据服务名和版本获取实例ID列表，用于内部批量操作
+**函数描述**: 根据服务名和版本获取实例信息列表，用于内部批量操作
 
 **函数签名**:
 ```go
-func GetServiceInstanceIDs(serviceName string, version ...string) ([]string, error)
+func GetServiceInstanceInfos(serviceName string, version ...string) ([]*InstanceInfo, error)
 ```
 
 **输入参数**:
@@ -202,7 +202,7 @@ serviceName string   // 必填，服务名称
 version     ...string // 选填，指定版本号进行过滤，未输入则默认获取全部版本的运行实例
 ```
 
-**返回结果**: `[]string` - 实例ID数组
+**返回结果**: `[]*InstanceInfo` - 实例信息数组
 
 ### 4.3 GetInstanceIP函数
 
@@ -242,12 +242,13 @@ instanceID string // 必填，实例ID
 
 **函数签名**:
 ```go
-func CheckInstanceHealth(instanceID string) (bool, error)
+func CheckInstanceHealth(instanceIP string, instancePort int) (bool, error)
 ```
 
 **输入参数**:
 ```go
-instanceID string // 必填，实例ID
+instanceIP string // 必填，实例IP地址
+instancePort int  // 必填，实例端口号
 ```
 
 **返回结果**: `bool` - 健康检查结果，true表示实例有响应，false表示无响应

internal/deploy/database/database.go

@@ -0,0 +1,64 @@
+package database
+
+import (
+	"database/sql"
+	"fmt"
+	"sync"
+
+	_ "github.com/lib/pq" // PostgreSQL驱动
+	"github.com/qiniu/zeroops/internal/deploy/config"
+)
+
+// Database 数据库连接管理器
+type Database struct {
+	db *sql.DB
+	mu sync.RWMutex
+}
+
+// NewDatabase 创建新的数据库连接
+func NewDatabase(cfg *config.DatabaseConfig) (*Database, error) {
+	dsn := cfg.GetDSN()
+
+	db, err := sql.Open("postgres", dsn)
+	if err != nil {
+		return nil, fmt.Errorf("failed to open database connection: %w", err)
+	}
+
+	// 测试连接
+	if err := db.Ping(); err != nil {
+		db.Close()
+		return nil, fmt.Errorf("failed to ping database: %w", err)
+	}
+
+	database := &Database{
+		db: db,
+	}
+
+	return database, nil
+}
+
+// GetDB 获取数据库连接（供repo使用）
+func (d *Database) GetDB() *sql.DB {
+	d.mu.RLock()
+	defer d.mu.RUnlock()
+	return d.db
+}
+
+// Close 关闭数据库连接
+func (d *Database) Close() error {
+	d.mu.Lock()
+	defer d.mu.Unlock()
+
+	if d.db != nil {
+		return d.db.Close()
+	}
+	return nil
+}
+
+// Ping 测试数据库连接
+func (d *Database) Ping() error {
+	d.mu.RLock()
+	defer d.mu.RUnlock()
+
+	return d.db.Ping()
+}

internal/deploy/database/host_repo.go

@@ -0,0 +1,55 @@
+package database
+
+import (
+	"database/sql"
+	"fmt"
+	"strconv"
+
+	"github.com/qiniu/zeroops/internal/deploy/model"
+)
+
+// HostRepo 主机数据访问层
+type HostRepo struct {
+	db *sql.DB
+}
+
+// NewHostRepo 创建主机仓库
+func NewHostRepo(db *Database) *HostRepo {
+	return &HostRepo{
+		db: db.GetDB(),
+	}
+}
+
+// GetAvailableHostInfos 获取所有可用的主机信息列表
+func (r *HostRepo) GetAvailableHostInfos() ([]*model.HostInfo, error) {
+	query := `
+		SELECT id, name, ip_address, is_stopped
+		FROM hosts 
+		WHERE is_stopped = false
+		ORDER BY id
+	`
+
+	rows, err := r.db.Query(query)
+	if err != nil {
+		return nil, fmt.Errorf("failed to query available hosts: %w", err)
+	}
+	defer rows.Close()
+
+	var hostInfos []*model.HostInfo
+	for rows.Next() {
+		hostInfo := new(model.HostInfo)
+		var id int
+		err := rows.Scan(&id, &hostInfo.HostName, &hostInfo.HostIPAddress, &hostInfo.IsStopped)
+		if err != nil {
+			return nil, fmt.Errorf("failed to scan host: %w", err)
+		}
+		hostInfo.HostID = strconv.Itoa(id)
+		hostInfos = append(hostInfos, hostInfo)
+	}
+
+	if err := rows.Err(); err != nil {
+		return nil, fmt.Errorf("error iterating hosts: %w", err)
+	}
+
+	return hostInfos, nil
+}