TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-03 23:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

在各个地方支持IPv6

Browse Source
This commit is contained in:
GoEdgeLab
2021-07-20 10:55:34 +08:00
parent 1dbfb11dd8
commit 4d3625fc31
3 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/installers/installer_base.go
View File

@@ -2,6 +2,7 @@ package installers
import (
"errors"
"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
"github.com/iwind/TeaGo/Tea"
stringutil "github.com/iwind/TeaGo/utils/string"
"golang.org/x/crypto/ssh"
@@ -17,7 +18,7 @@ type BaseInstaller struct {
client *SSHClient
}
// 登录SSH服务
// Login 登录SSH服务
func (this *BaseInstaller) Login(credentials *Credentials) error {
var hostKeyCallback ssh.HostKeyCallback = nil
@@ -78,7 +79,7 @@ func (this *BaseInstaller) Login(credentials *Credentials) error {
Timeout: 5 * time.Second, // TODO 后期可以设置这个超时时间
}
sshClient, err := ssh.Dial("tcp", credentials.Host+":"+strconv.Itoa(credentials.Port), config)
sshClient, err := ssh.Dial("tcp", configutils.QuoteIP(credentials.Host)+":"+strconv.Itoa(credentials.Port), config)
if err != nil {
return err
}
@@ -90,7 +91,7 @@ func (this *BaseInstaller) Login(credentials *Credentials) error {
return nil
}
// 关闭SSH服务
// Close 关闭SSH服务
func (this *BaseInstaller) Close() error {
if this.client != nil {
return this.client.Close()
@@ -99,7 +100,7 @@ func (this *BaseInstaller) Close() error {
return nil
}
// 查找最新的版本的文件
// LookupLatestInstaller 查找最新的版本的文件
func (this *BaseInstaller) LookupLatestInstaller(filePrefix string) (string, error) {
matches, err := filepath.Glob(Tea.Root + Tea.DS + "deploy" + Tea.DS + "*.zip")
if err != nil {
@@ -131,7 +132,7 @@ func (this *BaseInstaller) LookupLatestInstaller(filePrefix string) (string, err
return result, nil
}
// 上传安装助手
// InstallHelper 上传安装助手
func (this *BaseInstaller) InstallHelper(targetDir string) (env *Env, err error) {
uname, _, err := this.client.Exec("uname -a")
if err != nil {

12
internal/installers/ssh_client.go
View File

@@ -30,7 +30,7 @@ func NewSSHClient(raw *ssh.Client) (*SSHClient, error) {
return c, nil
}
// 执行shell命令
// Exec 执行shell命令
func (this *SSHClient) Exec(cmd string) (stdout string, stderr string, err error) {
session, err := this.raw.NewSession()
if err != nil {
@@ -86,7 +86,7 @@ func (this *SSHClient) Chmod(path string, mode os.FileMode) error {
return this.sftp.Chmod(path, mode)
}
// 拷贝文件
// Copy 拷贝文件
func (this *SSHClient) Copy(localPath string, remotePath string, mode os.FileMode) error {
localFp, err := os.Open(localPath)
if err != nil {
@@ -110,12 +110,12 @@ func (this *SSHClient) Copy(localPath string, remotePath string, mode os.FileMod
return this.Chmod(remotePath, mode)
}
// 获取新Session
// NewSession 获取新Session
func (this *SSHClient) NewSession() (*ssh.Session, error) {
return this.raw.NewSession()
}
// 读取文件内容
// ReadFile 读取文件内容
func (this *SSHClient) ReadFile(path string) ([]byte, error) {
fp, err := this.sftp.OpenFile(path, 0444)
if err != nil {
@@ -134,7 +134,7 @@ func (this *SSHClient) ReadFile(path string) ([]byte, error) {
return buffer.Bytes(), nil
}
// 写入文件内容
// WriteFile 写入文件内容
func (this *SSHClient) WriteFile(path string, data []byte) (n int, err error) {
fp, err := this.sftp.OpenFile(path, os.O_CREATE|os.O_TRUNC|os.O_WRONLY)
if err != nil {
@@ -148,7 +148,7 @@ func (this *SSHClient) WriteFile(path string, data []byte) (n int, err error) {
return
}
// 删除文件
// Remove 删除文件
func (this *SSHClient) Remove(path string) error {
return this.sftp.Remove(path)
}

3
internal/rpc/services/service_node_grant.go
View File

@@ -6,6 +6,7 @@ import (
"fmt"
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeAPI/internal/utils/numberutils"
"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
"golang.org/x/crypto/ssh"
"net"
@@ -252,7 +253,7 @@ func (this *NodeGrantService) TestNodeGrant(ctx context.Context, req *pb.TestNod
Timeout: 5 * time.Second, // TODO 后期可以设置这个超时时间
}
sshClient, err := ssh.Dial("tcp", req.Host+":"+fmt.Sprintf("%d", req.Port), config)
sshClient, err := ssh.Dial("tcp", configutils.QuoteIP(req.Host)+":"+fmt.Sprintf("%d", req.Port), config)
if err != nil {
resp.Error = "connect failed: " + err.Error()
return resp, nil