集成批处理提交和云Firestore触发器

时间:2019-03-29 07:01:08

标签: google-cloud-firestore

我正在批量提交50块文档到名为“活动”的云Firestore集合,并且我编写了一个云函数,该函数会在创建,更新和删除该集合中的任何文档时触发,但云函数会每个批次仅触发一次,而不是批次中的所有文档触发一次。

我已经计算出记录了多少次调用云函数的日志,它与批量提交的调用次数大致匹配。

正在执行批量提交的代码:

    const activitiesRef = app.firestore.collection(`activities`);
    if (_.isArray(insertObj)) {
        const chunksOf450 = await _.chunk(insertObj, 450);

        for (const singleChunk of chunksOf450) {
            // Get a new write batch
            const batch = app.firestore.batch();
            for (const obj of singleChunk) {
                const tempRef = activitiesRef.doc((obj as any).uid);
                batch.set(tempRef, obj);
            }

            // Commit the batch
            await batch.commit();
        }

        return {
            message: 'All objects written to database successfully'
        };

在onWrite上触发的Cloud Function:

export default functions.firestore.document('activities/{activityId}').onWrite(async (snap, context) => {
const isDocumentExists = snap.after.exists

utils.app.initializeApp(initializeAdminApp.default);
adminUtils.ElasticHelper.init();


if (isDocumentExists) {

    const afterDocData = snap.after.data();
    const afterDocId = snap.after.id;

    console.log('starting..', afterDocData, afterDocId);



    const { contentType } = snap.after.data();

    if (contentType === 'activityTypes') {
        await adminUtils.ElasticHelper.index('activity', afterDocId, afterDocData);
    }
    if (contentType === 'itemTypes') {
        await adminUtils.ElasticHelper.index('item', afterDocId, afterDocData);
    }
    if (contentType === 'questionTypes') {
        await adminUtils.ElasticHelper.index('question', afterDocId, afterDocData);
    }

} else {
    const beforeDocId = snap.before.id;
    const { contentType } = snap.before.data();

    console.log('deleting on elastic ..', beforeDocId);

    if (contentType === 'activityTypes') {
        await adminUtils.ElasticHelper.delete('activity', beforeDocId);
    }
    if (contentType === 'itemTypes') {
        await adminUtils.ElasticHelper.delete('item', beforeDocId);
    }
    if (contentType === 'questionTypes') {
        await adminUtils.ElasticHelper.delete('question', beforeDocId);
    }
}
console.log('function success');
});

对于要批量提交的每个文档,我都希望触发此云函数。

谢谢。

1 个答案:

答案 0 :(得分:0)

如果您完全删除了集合然后进行批处理提交,则效果很好,因为如果集合已经存在并且您尝试覆盖文档,则云触发器将不会启动。您必须先删除所有文档。