feat: 数据库、redis、mongo支持ssh隧道等

server/internal/devops/domain/entity/db.go

@@ -1,6 +1,7 @@
 package entity
 
 import (
+	"fmt"
 	"mayfly-go/pkg/model"
 )
 
@@ -21,9 +22,24 @@ type Db struct {
 	EnvId     uint64
 	Env       string
 
-	EnableSSH int    `orm:"column(enable_ssh)" json:"enable_ssh"`
-	SSHHost   string `orm:"column(ssh_host)" json:"ssh_host"`
-	SSHPort   int    `orm:"column(ssh_port)" json:"ssh_port"`
-	SSHUser   string `orm:"column(ssh_user)" json:"ssh_user"`
-	SSHPass   string `orm:"column(ssh_pass)" json:"-"`
+	EnableSshTunnel    int8   `orm:"column(enable_ssh_tunnel)" json:"enableSshTunnel"`        // 是否启用ssh隧道
+	SshTunnelMachineId uint64 `orm:"column(ssh_tunnel_machine_id)" json:"sshTunnelMachineId"` // ssh隧道机器id
 }
+
+// 获取数据库连接网络, 若没有使用ssh隧道，则直接返回。否则返回拼接的网络需要注册至指定dial
+func (d Db) GetNetwork() string {
+	network := d.Network
+	if d.EnableSshTunnel == -1 {
+		if network == "" {
+			return "tcp"
+		} else {
+			return network
+		}
+	}
+	return fmt.Sprintf("%s+ssh:%d", d.Type, d.SshTunnelMachineId)
+}
+
+const (
+	DbTypeMysql    = "mysql"
+	DbTypePostgres = "postgres"
+)

server/internal/devops/domain/entity/machine.go

@@ -9,8 +9,9 @@ type Machine struct {
 	ProjectId   uint64 `json:"projectId"`
 	ProjectName string `json:"projectName"`
 	Name        string `json:"name"`
-	Ip          string `json:"ip"`       // IP地址
-	Username    string `json:"username"` // 用户名
+	Ip          string `json:"ip"`         // IP地址
+	Username    string `json:"username"`   // 用户名
+	AuthMethod  int8   `json:"authMethod"` // 授权认证方式
 	Password    string `json:"-"`
 	Port        int    `json:"port"`   // 端口号
 	Status      int8   `json:"status"` // 状态 1:启用；2:停用
@@ -18,6 +19,8 @@ type Machine struct {
 }
 
 const (
-	MachineStatusEnable  int8 = 1  // 启用状态
-	MachineStatusDisable int8 = -1 // 禁用状态
+	MachineStatusEnable        int8 = 1  // 启用状态
+	MachineStatusDisable       int8 = -1 // 禁用状态
+	MachineAuthMethodPassword  int8 = 1  // 密码登录
+	MachineAuthMethodPublicKey int8 = 2  // 公钥免密登录
 )

server/internal/devops/domain/entity/mongo.go

@@ -5,10 +5,12 @@ import "mayfly-go/pkg/model"
 type Mongo struct {
 	model.Model
 
-	Name      string `orm:"column(name)" json:"name"`
-	Uri       string `orm:"column(uri)" json:"uri"`
-	ProjectId uint64 `json:"projectId"`
-	Project   string `json:"project"`
-	EnvId     uint64 `json:"envId"`
-	Env       string `json:"env"`
+	Name               string `orm:"column(name)" json:"name"`
+	Uri                string `orm:"column(uri)" json:"uri"`
+	EnableSshTunnel    int8   `orm:"column(enable_ssh_tunnel)" json:"enableSshTunnel"`        // 是否启用ssh隧道
+	SshTunnelMachineId uint64 `orm:"column(ssh_tunnel_machine_id)" json:"sshTunnelMachineId"` // ssh隧道机器id
+	ProjectId          uint64 `json:"projectId"`
+	Project            string `json:"project"`
+	EnvId              uint64 `json:"envId"`
+	Env                string `json:"env"`
 }

server/internal/devops/domain/entity/redis.go

@@ -7,15 +7,17 @@ import (
 type Redis struct {
 	model.Model
 
-	Host      string `orm:"column(host)" json:"host"`
-	Mode      string `json:"mode"`
-	Password  string `orm:"column(password)" json:"-"`
-	Db        int    `orm:"column(database)" json:"db"`
-	Remark    string
-	ProjectId uint64
-	Project   string
-	EnvId     uint64
-	Env       string
+	Host               string `orm:"column(host)" json:"host"`
+	Mode               string `json:"mode"`
+	Password           string `orm:"column(password)" json:"-"`
+	Db                 int    `orm:"column(database)" json:"db"`
+	EnableSshTunnel    int8   `orm:"column(enable_ssh_tunnel)" json:"enableSshTunnel"`        // 是否启用ssh隧道
+	SshTunnelMachineId uint64 `orm:"column(ssh_tunnel_machine_id)" json:"sshTunnelMachineId"` // ssh隧道机器id
+	Remark             string
+	ProjectId          uint64
+	Project            string
+	EnvId              uint64
+	Env                string
 }
 
 const (