GithubBackup/MyGoNavi
1
0
Fork 0
mirror of https://github.com/Syngnat/GoNavi.git synced 2026-08-25 02:00:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix(ssh): 为共享端口转发增加引用计数与失败回滚

Browse Source
This commit is contained in:
Syngnat
2026-07-27 00:07:42 +08:00
parent 546b71945d
commit 76249c00b9
28 changed files with 795 additions and 198 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
internal/db/tdengine_impl.go
View File

@@ -56,13 +56,20 @@ func (t *TDengineDB) getDSN(config connection.ConnectionConfig) string {
return dsn + "?" + query
}
func (t *TDengineDB) Connect(config connection.ConnectionConfig) error {
func (t *TDengineDB) Connect(config connection.ConnectionConfig) (err error) {
_ = t.Close()
defer func() {
if err != nil {
_ = t.Close()
}
}()
runConfig := config
if config.UseSSH {
logger.Infof("TDengine 使用 SSH 连接：地址=%s:%d 用户=%s", config.Host, config.Port, config.User)
forwarder, err := ssh.GetOrCreateLocalForwarder(config.SSH, config.Host, config.Port)
forwarder, err := ssh.AcquireLocalForwarder(config.SSH, config.Host, config.Port)
if err != nil {
return fmt.Errorf("创建 SSH 隧道失败：%w", err)
}
@@ -118,7 +125,7 @@ func (t *TDengineDB) Connect(config connection.ConnectionConfig) error {
func (t *TDengineDB) Close() error {
if t.forwarder != nil {
if err := t.forwarder.Close(); err != nil {
if err := t.forwarder.Release(); err != nil {
logger.Warnf("关闭 TDengine SSH 端口转发失败：%v", err)
}
t.forwarder = nil