node实现FTP上传

3,251 阅读1分钟
const fs = require('fs');
const FTPClient = require('ftp');

const dirPath = 'C:\\Users\\Ai_power04\\Desktop\\monit\\dist'; // 本地FTP目录
const remoteFtpPath = 'yc-monit'; // 远程FTP目录
const localFiles = []; // 本地文件
let localFileLength = 0; // 上传文件个数
let uploadTime = 0; // 上传耗时
let ftp = null;

readingFile().then(()=>{
    // console.log(localFiles);
    if(localFileLength) {
        // 上传文件 - 覆盖
        console.log('连接192.168.10.107...');
        ftp = new FTPClient();
        ftp.on('ready', () => {
            // 清理文件
            console.log('连接成功,清理旧文件...');
            ftp.rmdir(remoteFtpPath,true,err=>{
                if(err) console.log('目录不存在,无需清理');
                console.log('开始上传...');
                uploadingFile().then(()=>{
                    console.log(`上传结束,本次上传${localFiles.length}个文件`);
                }).catch(err=>{
                    console.log(err);
                })
            });
        });
        // 连接服务器
        ftp.connect({
            host:'192.168.10.107',
            port:21,
            user:'Administrator',
            password:'1234',
            connTimeout: 1000*10, // 连接超时时间
            pasvTimeout: 1000*10, // PASV data 连接超时时间
            keepalive: 1000* 10, // 多久发送一次请求,以保持连接
        });
    } else {
        console.log('本地文件为空,取消上传');
    }
}).catch(err=>{
    console.log(err);
});

// 正在获取本地文件
function readingFile() {
    return new Promise((resolve,reject)=>{
        readFiles(dirPath).catch(err=>reject(err));
        // 判断是否读取完毕(1s内 localFiles长度不再变化)
        console.log('正在读取本地文件,请稍后...');
        let timer = setInterval(() => {
            if(localFileLength == localFiles.length) {
                clearInterval(timer);
                resolve();
                return console.log('读取完成');
            }
        }, 1000);
    })
}
// 递归读取文件
function readFiles(filepath) {
    return new Promise((resolve,reject)=>{
        fs.readdir(filepath,{withFileTypes:true},(err,files)=>{
            if(err) throw err;
            if(files.length>0) {
                files.map(file=>{
                    if(file.isFile()) { // 文件
                        const child_filepath = filepath + '\\' + file.name;
                        fs.readFile(child_filepath,(err,data)=>{
                            if(err) throw err;
                            const dir = remoteFtpPath + filepath.replace(dirPath,'').replace('\\','/') ;
                            localFiles.push({
                                dir,
                                filepath: dir + '/' +file.name,
                                filedata: data
                            })
                            localFileLength = localFiles.length;
                        });
                    } else { // 目录
                        const child_filepath = filepath + '\\' + file.name;
                        readFiles(child_filepath);
                    }
                });
            }
        });
    });
}

// 正在上传文件
function uploadingFile() {
    return new Promise((resolve,reject)=>{
        localFiles.map(file=>{
            uploadFiles(file.dir,file.filepath,file.filedata).catch(err=>reject(err));
        })
        console.log('正在上传,请稍后...');
        let timer = setInterval(() => {
            if(localFileLength == 0) {
                clearInterval(timer);
                resolve();
                return console.log('上传完成');
            }
            uploadTime++;
            console.log(`已用时:${uploadTime}s`);
        }, 1000);
    })
}
// 覆盖上传文件
function uploadFiles(dir,filepath,filedata) {
    return new Promise((resolve,reject)=>{
        ftp.mkdir(dir,true,err1=>{
            if(err1) reject(err1);
            ftp.put(filedata,filepath,err2 => {
                if (err2) reject(err2);
                localFileLength--;
                ftp.end();
                resolve();
            });
        })
    })
}