Firebase 功能,将同步/异步混乱变成干净的 ES6

问题描述 投票:0回答:1

我确实编写了一个 firebase 函数,它可以根据请求操作 PDF 文件,对其进行一些操作,将其保存回存储并将哈希值存档到数据库。

我确实设法让它工作,但它一团糟,因为我从未真正学习过如何使用 ES6 之前的 js 回调。我对这一切都很陌生,并且学会了使用箭头函数和承诺。但在这里,我需要使用纯 JavaScript 的包及其以某种方式工作,但我确实需要清理这个同步/异步混乱,以进行干净的错误处理,并在处理函数后向 firebase 返回一个承诺。

我的函数还包括一些奇怪的文件处理,以便为 pdf 库准备不同的文件: 例如,我创建一个 QR 码,将其保存到 tmp 文件,使用另一个库从 png 创建 jpg 并将其再次保存到 tmp。如果我能做更聪明的事情,我愿意接受任何建议或提示。

链出现问题,因为该函数仍在运行时在几毫秒内完成。

我确实在我的代码中添加了一些注释,但我不知道如何将其更改为 ES6,如果你能帮我清理这个大烂摊子,我将非常感激。

const hummus = require('hummus');
const request = require('request');
const fs = require('fs');
const sha1 = require('sha1');
const images = require("images");

exports.handleDocSignRequests = functions.database.ref('/user_writeable/docrequests/{uid}').onWrite(event => {
    var userReq = event.data.val();
    var userRef = event.data.ref;
    if (!userReq) return Promise.resolve();
    if (!userReq.docpath) return Promise.resolve();
    let uid = event.params.uid;
    let filename = userReq.docpath; // File to sign and hash

    return bucket.file(filename).getSignedUrl({  // getting downloadurl from Firebase Storage
        action: 'read'
    }).then(
        (downloadpath) => {
            downloadpath = downloadpath[0];
            //download pdf - how to turn this into a promise?
            download = function (uri, filename, callback) {
                request.head(uri, function (err, res, body) {
                    request(uri).pipe(fs.createWriteStream(filename)).on('close', callback);
                });
            };
            let pdfsourse = LOCAL_TMP_FOLDER + 'downloadedfile.pdf';
            return download(downloadpath, pdfsourse, function () { // download callback turn this into ES6
                console.log('download finished');
                let qrjpg = LOCAL_TMP_FOLDER + 'qrcode.jpg';
                var qrpng = LOCAL_TMP_FOLDER + 'qrcode.png';
                let qrurl = 'https://some.url/' + userReq.docid;
                let pdfdest = LOCAL_TMP_FOLDER + 'newpdf.pdf';
                let logfile = './hummus.log';

                QRCode.toFile(qrpng, qrurl, {  // how to make this a part of the "chain" and go on when finished
                    version: 4, type: "png", scale: 2, margin: 0
                }, function (err) {
                    if (err) throw err;
                    console.log('qr png ready');
                    images(qrpng).save(qrjpg, {operation: 100}); // save png as jpg
                    console.log('qr jpg ready');
                });


                // Doing stuff to PDF with HummusJs
                let pdfWriter = hummus.createWriterToModify(pdfsourse, {
                    modifiedFilePath: pdfdest,
                    log: logfile,
                    userPassword: 'user',
                    ownerPassword: 'owner',
                    userProtectionFlag: 4

                });
                let pdfReader = pdfWriter.getModifiedFileParser();
                let arialFont = pdfWriter.getFontForFile(ariallocal);
                let textOptions = {font: arialFont, size: 5, colorspace: 'gray', color: 0x00};

                for (let i = 0; i < pdfReader.getPagesCount(); ++i) {
                    let pageBox = pdfReader.parsePage(i).getMediaBox();
                    let pageModifier = new hummus.PDFPageModifier(pdfWriter, i, true);
                    let ctx = pageModifier.startContext().getContext();
                    ctx.writeText('Document saved', 5, 110, textOptions);
                    ctx.drawImage(5, 52, qrfile,
                        {
                            transformation: {
                                width: 40,
                                height: 40,
                                fit: 'always'
                            }
                        });
                    pageModifier.endContext().writePage();
                }
                pdfWriter.end();
                // How can I be sure PDF is done and written to tmp file? Or is this given by sync function?
                // Reading finished PDF from file again, to get base64 for hashing - is there a better way?
                let newpdf = fs.readFileSync(pdfdest);
                let base64pdf = newpdf.toString('base64');
                let hash = sha1(base64pdf);
                let signobj = {};
                signobj['hash'] = hash;
                // Check if document already in database, if not write hash to database,
                // upload finished pdf to original place and archive
                // and return remove request
                let sign_ref = docsign_ref.child(userReq.docid);
                return sign_ref.once('value').then(function (snap) {
                    if (!snap.val()) { //Document is new
                        let upload1 = bucket.upload(destcry, {destination: filename}).then
                        (suc => {
                            console.log('uploaded');
                        });
                        //
                        let upload2 = bucket.upload(destcry, {destination: 'signed/' + userReq.docid + '.pdf'}).then
                        (suc => {
                            console.log('uploaded');
                        });
                        return Promise.all([upload1, upload2]).then( // When both uploads are finished go on
                            (suc) => {
                                return sign_ref.set(signobj).then(
                                    (suc) => {
                                        // Remove Request and return Promise
                                        return userRef.remove();
                                    });

                            });
                    }
                    else {
                        //Document already in database, this should never happen, only for seq reasons
                        console.log('doc already in database);
                        return Promise.resolve();
                    }
                });
            });
        });
});
firebase callback ecmascript-6 promise google-cloud-functions
1个回答
0
投票

正如我在评论中所写,你的问题太宽泛了。我将重点关注您在标题中提出的问题。

ES6 Promise 的主要改进是你应该:

  • 将一些基于回调的方案转变为 Promise,并且
  • 展平承诺链,避免嵌套
    then
    调用

最后,当您调用外部函数时,请确保将其视为承诺:一旦进行异步操作,您就需要坚持该模式。所以你可以这样调用你的 main 函数:

handleDocSignRequests().then( .... );

这是一些未经测试(!)的代码重写,以解决上述两点。主要变化用

***
注释。可能会有一些错误,但模式应该很清晰:

const hummus = require('hummus');
const request = require('request');
const fs = require('fs');
const sha1 = require('sha1');
const images = require("images");

exports.handleDocSignRequests = functions.database.ref('/user_writeable/docrequests/{uid}').onWrite(event => {
    var userReq = event.data.val();
    var userRef = event.data.ref;
    if (!userReq) return Promise.resolve();
    if (!userReq.docpath) return Promise.resolve();
    let uid = event.params.uid;
    let filename = userReq.docpath; // File to sign and hash

    // *** Define these variables at this level, so they are accessible throughout
    //     the flattened promise-chain:
    let signobj = {};
    let sign_ref = docsign_ref.child(userReq.docid);

    return bucket.file(filename).getSignedUrl({  // getting downloadurl from Firebase Storage
        action: 'read'
    }).then((downloadpath) => {
        downloadpath = downloadpath[0];
        let pdfsourse = LOCAL_TMP_FOLDER + 'downloadedfile.pdf';
        // *** turned into promise:
        // *** return the promise to the main promise chain
        return new Promise(resolve => {
            request.head(downloadpath, function (err, res, body) {
                request(downloadpath)
                    .pipe(fs.createWriteStream(pdfsourse))
                    .on('close', resolve);
            });
        });
    }).then(function () {
        console.log('download finished');
        let qrjpg = LOCAL_TMP_FOLDER + 'qrcode.jpg';
        var qrpng = LOCAL_TMP_FOLDER + 'qrcode.png';
        let qrurl = 'https://some.url/' + userReq.docid;
        // *** Return a promise to make this a part of the "chain" and go on when finished
        return new Promise( (resolve, reject) => {
            QRCode.toFile(qrpng, qrurl, {  
                version: 4, type: "png", scale: 2, margin: 0
            }, function (err) {
                if (err) reject(err); // *** reject 
                console.log('qr png ready');
                images(qrpng).save(qrjpg, {operation: 100}); // save png as jpg
                console.log('qr jpg ready');
                resolve(); // *** resolve the promise now it's done
            });
        });
    }).then(function () {
        // Doing stuff to PDF with HummusJs
        let pdfdest = LOCAL_TMP_FOLDER + 'newpdf.pdf';
        let logfile = './hummus.log';
        let pdfWriter = hummus.createWriterToModify(pdfsourse, {
            modifiedFilePath: pdfdest,
            log: logfile,
            userPassword: 'user',
            ownerPassword: 'owner',
            userProtectionFlag: 4
        });
        let pdfReader = pdfWriter.getModifiedFileParser();
        let arialFont = pdfWriter.getFontForFile(ariallocal);
        let textOptions = {font: arialFont, size: 5, colorspace: 'gray', color: 0x00};

        for (let i = 0; i < pdfReader.getPagesCount(); ++i) {
            let pageBox = pdfReader.parsePage(i).getMediaBox();
            let pageModifier = new hummus.PDFPageModifier(pdfWriter, i, true);
            let ctx = pageModifier.startContext().getContext();
            ctx.writeText('Document saved', 5, 110, textOptions);
            ctx.drawImage(5, 52, qrfile, {
                transformation: {
                    width: 40,
                    height: 40,
                    fit: 'always'
                }
            });
            pageModifier.endContext().writePage();
        }
        pdfWriter.end();
        // How can I be sure PDF is done and written to tmp file? Or is this given by sync function?
        // Reading finished PDF from file again, to get base64 for hashing - is there a better way?
        let newpdf = fs.readFileSync(pdfdest);
        let base64pdf = newpdf.toString('base64');
        let hash = sha1(base64pdf);
        signobj['hash'] = hash;
        // Check if document already in database, if not write hash to database,
        // upload finished pdf to original place and archive
        // and return remove request
        // *** Only return the basic promise. Perform the `then` on the outer chain.
        return sign_ref.once('value');
    }).then(function (snap) {
        // *** Do simple case first:
        if (snap.val()) { 
            //Document already in database, this should never happen, only for seq reasons
            console.log('doc already in database);
            return Promise.resolve();
        } // *** No need for `else` here. The above `return` is enough
        //Document is new
        // *** use array to replace two blocks of very similar code
        let promises = [filename, 'signed/' + userReq.docid + '.pdf'].map(destination =>
            return bucket.upload(destcry, {destination}).then(suc => {
                console.log('uploaded');
            });
        });
        // *** Return the basic promise. Perform the `then` on the outer chain
        return Promise.all(promises);// Wait for both uploads to finish
    }).then( suc => {
        // *** I suppose this also has to run when not a new document....
        // *** Again: return the promise and perform the `then` on the outer chain
        return sign_ref.set(signobj);
    }).then( suc => {
        // Remove Request and return Promise
        return userRef.remove();
    });
});
© www.soinside.com 2019 - 2024. All rights reserved.