圆月山庄资源网 Design By www.vgjia.com
使用go语言的第三方包:github.com/pkg/sftp和golang.org/x/crypto/ssh实现文件和文件夹传输。
1、创建connect方法:
func connect(user, password, host string, port int) (*sftp.Client, error) { var ( auth []ssh.AuthMethod addr string clientConfig *ssh.ClientConfig sshClient *ssh.Client sftpClient *sftp.Client err error ) // get auth method auth = make([]ssh.AuthMethod, 0) auth = append(auth, ssh.Password(password)) clientConfig = &ssh.ClientConfig{ User: user, Auth: auth, Timeout: 30 * time.Second, HostKeyCallback: ssh.InsecureIgnoreHostKey(), //ssh.FixedHostKey(hostKey), } // connet to ssh addr = fmt.Sprintf("%s:%d", host, port) if sshClient, err = ssh.Dial("tcp", addr, clientConfig); err != nil { return nil, err } // create sftp client if sftpClient, err = sftp.NewClient(sshClient); err != nil { return nil, err } return sftpClient, nil }
2、上传文件
func uploadFile(sftpClient *sftp.Client, localFilePath string, remotePath string) { srcFile, err := os.Open(localFilePath) if err != nil { fmt.Println("os.Open error : ", localFilePath) log.Fatal(err) } defer srcFile.Close() var remoteFileName = path.Base(localFilePath) dstFile, err := sftpClient.Create(path.Join(remotePath, remoteFileName)) if err != nil { fmt.Println("sftpClient.Create error : ", path.Join(remotePath, remoteFileName)) log.Fatal(err) } defer dstFile.Close() ff, err := ioutil.ReadAll(srcFile) if err != nil { fmt.Println("ReadAll error : ", localFilePath) log.Fatal(err) } dstFile.Write(ff) fmt.Println(localFilePath + " copy file to remote server finished!") }
3、上传文件夹
func uploadDirectory(sftpClient *sftp.Client, localPath string, remotePath string) { localFiles, err := ioutil.ReadDir(localPath) if err != nil { log.Fatal("read dir list fail ", err) } for _, backupDir := range localFiles { localFilePath := path.Join(localPath, backupDir.Name()) remoteFilePath := path.Join(remotePath, backupDir.Name()) if backupDir.IsDir() { sftpClient.Mkdir(remoteFilePath) uploadDirectory(sftpClient, localFilePath, remoteFilePath) } else { uploadFile(sftpClient, path.Join(localPath, backupDir.Name()), remotePath) } } fmt.Println(localPath + " copy directory to remote server finished!") }
4、上传测试
func DoBackup(host string, port int, userName string, password string, localPath string, remotePath string) { var ( err error sftpClient *sftp.Client ) start := time.Now() sftpClient, err = connect(userName, password, host, port) if err != nil { log.Fatal(err) } defer sftpClient.Close() _, errStat := sftpClient.Stat(remotePath) if errStat != nil { log.Fatal(remotePath + " remote path not exists!") } backupDirs, err := ioutil.ReadDir(localPath) if err != nil { log.Fatal(localPath + " local path not exists!") } uploadDirectory(sftpClient, localPath, remotePath) elapsed := time.Since(start) fmt.Println("elapsed time : ", elapsed) }
补充:go实现ssh远程机器并传输文件
核心依赖包:
golang.org/x/crypto/ssh
github.com/pkg/sftp
其中golang.org/x/crypto/ssh 可从github上下载,
下载地址:https://github.com/golang/crypto
ssh连接源码(这里是根据秘钥连接):
var keypath = "key/id_rsa" //获取秘钥 func publicKey(path string) ssh.AuthMethod { keypath, err := homedir.Expand(path) if err != nil { fmt.Println("获取秘钥路径失败", err) } key, err1 := ioutil.ReadFile(keypath) if err1 != nil { fmt.Println("读取秘钥失败", err1) } signer, err2 := ssh.ParsePrivateKey(key) if err2 != nil { fmt.Println("ssh 秘钥签名失败", err2) } return ssh.PublicKeys(signer) } //获取ssh连接 func GetSSHConect(ip, user string, port int) (*ssh.Client) { con := &ssh.ClientConfig{ User: user, Auth: []ssh.AuthMethod{publicKey(keypath)}, HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error { return nil }, } addr := fmt.Sprintf("%s:%d", ip, port) client, err := ssh.Dial("tcp", addr, con) if err != nil { fmt.Println("Dail failed: ", err) panic(err) } return client } // 远程执行脚本 func Exec_Task(ip, user, localpath, remotepath string) int { port := 22 client := GetSSHConect(ip, user, port) UploadFile(ip, user, localpath, remotepath, port) session, err := client.NewSession() if err != nil { fmt.Println("创建会话失败", err) panic(err) } defer session.Close() remoteFileName := path.Base(localpath) dstFile := path.Join(remotepath, remoteFileName) err1 := session.Run(fmt.Sprintf("/usr/bin/sh %s", dstFile)) if err1 != nil { fmt.Println("远程执行脚本失败", err1) return 2 } else { fmt.Println("远程执行脚本成功") return 1 } }
文件传输功能:
//获取ftp连接 func getftpclient(client *ssh.Client) (*sftp.Client) { ftpclient, err := sftp.NewClient(client) if err != nil { fmt.Println("创建ftp客户端失败", err) panic(err) } return ftpclient } //上传文件 func UploadFile(ip, user, localpath, remotepath string, port int) { client := GetSSHConect(ip, user, port) ftpclient := getftpclient(client) defer ftpclient.Close() remoteFileName := path.Base(localpath) fmt.Println(localpath, remoteFileName) srcFile, err := os.Open(localpath) if err != nil { fmt.Println("打开文件失败", err) panic(err) } defer srcFile.Close() dstFile, e := ftpclient.Create(path.Join(remotepath, remoteFileName)) if e != nil { fmt.Println("创建文件失败", e) panic(e) } defer dstFile.Close() buffer := make([]byte, 1024) for { n, err := srcFile.Read(buffer) if err != nil { if err == io.EOF { fmt.Println("已读取到文件末尾") break } else { fmt.Println("读取文件出错", err) panic(err) } } dstFile.Write(buffer[:n]) //注意,由于文件大小不定,不可直接使用buffer,否则会在文件末尾重复写入,以填充1024的整数倍 } fmt.Println("文件上传成功") } //文件下载 func DownLoad(ip, user, localpath, remotepath string, port int) { client := GetSSHConect(ip, user, port) ftpClient := getftpclient(client) defer ftpClient.Close() srcFile, err := ftpClient.Open(remotepath) if err != nil { fmt.Println("文件读取失败", err) panic(err) } defer srcFile.Close() localFilename := path.Base(remotepath) dstFile, e := os.Create(path.Join(localpath, localFilename)) if e != nil { fmt.Println("文件创建失败", e) panic(e) } defer dstFile.Close() if _, err1 := srcFile.WriteTo(dstFile); err1 != nil { fmt.Println("文件写入失败", err1) panic(err1) } fmt.Println("文件下载成功") }
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。如有错误或未考虑完全的地方,望不吝赐教。
圆月山庄资源网 Design By www.vgjia.com
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
圆月山庄资源网 Design By www.vgjia.com
暂无评论...
更新日志
2024年11月01日
2024年11月01日
- [ABC]费玉清《5年10年20年》(德国版HD)【WAV+CUE】
- 「花神诞祭的祝福」DIY活动开启
- MARIA.CORDERO玛俐亚.1987-MARIA.CORDERO【银星】【WAV+CUE】
- 张文浩.2024-跳火山的人【星外星】【FLAC分轨】
- 蔡琴.1985-老歌(2022新世纪MQA限量版)【永声】【WAV+CUE】
- 群星《十五首男声金曲2》DTS-WAV
- 2024年08月05日
- 柏菲·珞叔作品集《金色大厅2》限量开盘母带ORMCD[低速原抓WAV+CUE]
- NS版《人中之龙 极》演员本宫泰风饰演父亲的广告现已公开!
- 英国女性Switch用户占比达到47% 5年内翻了一番!
- 外媒:《死亡岛2》或是今年PS+2档最受欢迎游戏
- 群星.1997-世纪之歌第二辑6CD【宝丽金】【WAV+CUE】
- 邵萱.1997-是是非非【捷登】【WAV+CUE】
- 巫启贤.1998-我是你的【风格】【WAV+CUE】
- 【原神手游】「月草的赐慧」祈愿