148 lines
3.3 KiB
Go
148 lines
3.3 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"io"
|
||
"log"
|
||
"os"
|
||
"path"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/yeka/zip"
|
||
|
||
"github.com/pkg/sftp"
|
||
"golang.org/x/crypto/ssh"
|
||
)
|
||
|
||
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(),
|
||
}
|
||
|
||
// 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
|
||
}
|
||
|
||
var FORMAT = "2006-01-02"
|
||
var (
|
||
err error
|
||
sftpClient *sftp.Client
|
||
)
|
||
var dbnames = []string{"tower", "mengyin", "pingyi", "shizhi", "tancheng", "yinan", "yishui", "feixian", "gaoxinqu", "hedong", "jingkaiqu", "junan", "luozhuang", "lanling", "lanshan", "lingang", "linshu"}
|
||
|
||
func main() {
|
||
|
||
// 这里换成实际的 SSH 连接的 用户名，密码，主机名或IP，SSH端口
|
||
// sftpClient, err = connect("root", "sdlylshl871016", "111.235.181.127", 443)
|
||
sftpClient, err = connect("root", "HR2018!!", "15.14.12.150", 22)
|
||
if err != nil {
|
||
log.Println(err)
|
||
}
|
||
defer sftpClient.Close()
|
||
// 用来测试的远程文件路径 和 本地文件夹
|
||
// fmt.Println(shizhi)
|
||
// var localDir = "."
|
||
var dsts []string
|
||
tmp := os.TempDir()
|
||
date_dir := path.Join(tmp, "db_"+time.Now().Format(FORMAT))
|
||
os.Mkdir(date_dir, 0755)
|
||
var lzkpbi = "/docker/backup/" + time.Now().Format(FORMAT) + "_lzkp_bi_inner.zip"
|
||
Down(lzkpbi, date_dir)
|
||
dsts = append(dsts, path.Join(date_dir, time.Now().Format(FORMAT)+"_lzkp_bi_inner.zip"))
|
||
for _, n := range dbnames {
|
||
p := "/docker/backup/" + time.Now().Format(FORMAT) + "_" + n + "_inner.zip"
|
||
// fmt.Println(p)
|
||
|
||
Down(p, date_dir)
|
||
dsts = append(dsts, path.Join(date_dir, time.Now().Format(FORMAT)+"_"+n+"_inner.zip"))
|
||
|
||
}
|
||
|
||
zippass("", dsts...)
|
||
// fmt.Scanln()
|
||
for _, v := range dsts {
|
||
// fmt.Println(v)
|
||
// ioutil.WriteFile(v, []byte("aaa"), 755)
|
||
os.Remove(v)
|
||
}
|
||
|
||
}
|
||
|
||
func Down(src, dst string) {
|
||
// fmt.Println(src, "数据正在复制中，请耐心等待...")
|
||
srcFile, err := sftpClient.Open(src)
|
||
if err != nil {
|
||
// log.Println(err)
|
||
return
|
||
}
|
||
defer srcFile.Close()
|
||
|
||
var localFileName = path.Base(src)
|
||
dstFile, err := os.Create(path.Join(dst, localFileName))
|
||
if err != nil {
|
||
// log.Println(err)
|
||
return
|
||
}
|
||
defer dstFile.Close()
|
||
|
||
if _, err = srcFile.WriteTo(dstFile); err != nil {
|
||
// log.Println(err)
|
||
return
|
||
}
|
||
|
||
// fmt.Println(src, "数据复制完成!")
|
||
|
||
}
|
||
|
||
func zippass(dst string, src ...string) {
|
||
fzip, err := os.Create(`D:/待测试数据.zip`)
|
||
if err != nil {
|
||
log.Fatalln(err)
|
||
}
|
||
zipw := zip.NewWriter(fzip)
|
||
defer zipw.Close()
|
||
for i, n := range src {
|
||
w, err := zipw.Encrypt(strconv.Itoa(i), `hangruan2017`, zip.AES256Encryption)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
f, err := os.Open(n)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
_, err = io.Copy(w, f)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
}
|
||
zipw.Flush()
|
||
}
|