mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-02 23:40:24 +08:00
57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package entity
|
||
|
||
import (
|
||
"fmt"
|
||
"mayfly-go/internal/common/utils"
|
||
"mayfly-go/pkg/model"
|
||
)
|
||
|
||
type Db struct {
|
||
model.Model
|
||
|
||
Name string `orm:"column(name)" json:"name"`
|
||
Type string `orm:"column(type)" json:"type"` // 类型，mysql oracle等
|
||
Host string `orm:"column(host)" json:"host"`
|
||
Port int `orm:"column(port)" json:"port"`
|
||
Network string `orm:"column(network)" json:"network"`
|
||
Username string `orm:"column(username)" json:"username"`
|
||
Password string `orm:"column(password)" json:"-"`
|
||
Database string `orm:"column(database)" json:"database"`
|
||
Params string `json:"params"`
|
||
ProjectId uint64
|
||
Project string
|
||
EnvId uint64
|
||
Env string
|
||
|
||
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 == 0 || d.EnableSshTunnel == -1 {
|
||
if network == "" {
|
||
return "tcp"
|
||
} else {
|
||
return network
|
||
}
|
||
}
|
||
return fmt.Sprintf("%s+ssh:%d", d.Type, d.SshTunnelMachineId)
|
||
}
|
||
|
||
func (d *Db) PwdEncrypt() {
|
||
// 密码替换为加密后的密码
|
||
d.Password = utils.PwdAesEncrypt(d.Password)
|
||
}
|
||
|
||
func (d *Db) PwdDecrypt() {
|
||
// 密码替换为解密后的密码
|
||
d.Password = utils.PwdAesDecrypt(d.Password)
|
||
}
|
||
|
||
const (
|
||
DbTypeMysql = "mysql"
|
||
DbTypePostgres = "postgres"
|
||
)
|