node获取指定目录下所有文件的内容生成PDF

有个需求场景是需要把某个目录下的所有指定文件的内容生成pdf,因为是前端开发,所以就用node来干活
首先是pdf的包是用pdfkit, 这里有两个坑,第一个是生成的文件中可能会有系统无法识别的换行和空格,会被替换成莫名其妙的符号,因此在拼接完内容后,要使用 str.replace(/rn|r/g, ‘n’)
进行过滤,第二个坑是中文字符会乱码,因此要引入中文ttf,例如
doc.font(‘./msyh.ttf’) ;
接下来是获取所有文件内容,在一开始我是判断文件的长度来识别是否结束,但是文件获取是异步,递归的时候同时进行这么多读取操作,如何才能判断所有异步操作都已经结束了呢,答案就是我们的Promise.all啦
首先我们将fs模块的几个要用到方法promiseify化,也就是我们可以将异步操作包装在promise函数中,由于文件操作的回调函数也是作为参数传入(万能的JS),所以我们只要获取到arguments中的callback函数,并且根据成功或者失败去resolve或reject就行啦,
那么Promise.all是用来干嘛呢?
Promise.all 接收一个 promise 对象的数组作为参数,当这个数组里的所有 promise 对象全部变为resolve或 有 reject 状态出现的时候,它才会去调用 .then 方法,它们是并发执行的。也就是说我们并行执行的所有文件读取操作,可以用Promise.all来解决啦!

const fs = require('fs')
const yargs = require('yargs')
const PDF = require('pdfkit')
const readdir = promisify(fs.readdir);
const stat = promisify(fs.stat);
const readFile = promisify(fs.readFile);
const doc = new PDF();

function uuid(times = 10) {
    let g = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789',
        str = '';
    let len = g.length;
    for (let i = 0; i < times; i++) {
        str += g[Math.round((Math.random() * (len - 1)))];
    }
    return str;
}
let { path = '', name = uuid(), fileType = '' } = yargs.argv || {};
let str = '';

fileType = (fileType.split(',') || []).filter(item => item !== '')

function promisify(fn) {
    return function() {
        let args = Array.from(arguments);
        return new Promise(function(resolve, reject) {
            args.push(function(err, result) {
                if (err) {
                    reject(err);
                } else {
                    resolve(result);
                }
            });
            fn.apply(null, args);
        });
    }
}

function readDirRecur(file, callback) {
    return readdir(file).then(function(files) {
        files = files.map(function(filename) {
            let fullPath = `${file}/${filename}`;
            return stat(fullPath).then(function(stats) {
                let limmit = fileType.length ? fileType.find(item => filename.includes(`.${item}`)) : true;
                if (stats.isDirectory()) {
                    return readDirRecur(fullPath, callback);
                } else if (limmit) {
                    return readFile(fullPath, 'utf8').then(function(data) {
                        callback(filename, data, fullPath);
                    });
                }
            })
        });
        return Promise.all(files);
    });
}
if (!path) return false;
readDirRecur(path, function(filename, data, fullPath) {
    str += data;
}).then(function() {
    str = str.replace(/\r\n|\r/g, '\n');
    doc.pipe(fs.createWriteStream(`${name}.pdf`))
    doc.fontSize(10);
    doc.font('./msyh.ttf');
    doc.text(str, 50, 50);
    doc.end()
}).catch(function(err) {
    console.log(err);
});
    原文作者:我不会典韦
    原文地址: https://segmentfault.com/a/1190000016497699
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞