forked from pytool/ssh
/
main.go
147 lines (124 loc) · 3.28 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
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()
}