/
ftpclient.go
76 lines (66 loc) · 1.52 KB
/
ftpclient.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
package ftpclient
import (
"fmt"
"go-app/lib/conf"
"go-app/lib/logger"
"io"
"os"
"time"
"github.com/jlaffaye/ftp"
"go.uber.org/zap"
)
const LOG_MODULE = "FTP ERROR"
type ftpClient struct {
Conn *ftp.ServerConn
config conf.FtpConf
}
type FileInfo struct {
Entry *ftp.Entry
Dir string
}
func Dail(cfg conf.FtpConf) (ftpclient *ftpClient, err error) {
Conn, err := ftp.Dial(
fmt.Sprintf("%s:%d", cfg.Address, cfg.Port),
ftp.DialWithTimeout(time.Duration(cfg.Timeout)*time.Second),
// ftp.DialWithExplicitTLS(&tls.Config{InsecureSkipVerify: true}),
ftp.DialWithDisabledEPSV(false),
)
if err != nil {
logger.Error(LOG_MODULE, zap.Error(err))
return
}
err = Conn.Login(cfg.User, cfg.Pass)
if err != nil {
logger.Error(LOG_MODULE, zap.Error(err))
return
}
return &ftpClient{Conn: Conn, config: cfg}, nil
}
func (f *ftpClient) Close() (err error) {
err = f.Conn.Quit()
if err != nil {
logger.Error(LOG_MODULE, zap.Error(err))
}
return
}
func (f *ftpClient) Download(entry *ftp.Entry) {
// 如果文件在上传中,则忽略
if entry.Size == 0 {
return
}
localPath := f.config.LocalPath + entry.Name
localFile, err := os.OpenFile(localPath, os.O_RDWR|os.O_CREATE, os.ModePerm)
if err != nil {
logger.Error(LOG_MODULE, zap.Error(err))
}
defer localFile.Close()
remoteFile, err := f.Conn.Retr(entry.Name)
if err != nil {
logger.Error(LOG_MODULE, zap.Error(err))
}
defer remoteFile.Close()
_, err = io.Copy(localFile, remoteFile)
if err != nil {
logger.Error(LOG_MODULE, zap.Error(err))
}
}