Thomas Demirdjian
89 lines
2.8 KiB
JavaScript
89 lines
2.8 KiB
JavaScript
const admin = require("firebase-admin");
|
|
const { BATCH_TYPE } = require("../config/types");
|
|
|
|
async function deleteFolder(path) {
|
|
try {
|
|
console.log(`Deleting folder: ${path}`);
|
|
const bucket = admin.storage().bucket();
|
|
await bucket.deleteFiles({ prefix: path, force: true });
|
|
console.log(`Folder ${path} deleted successfully`);
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
}
|
|
|
|
async function batchFirestore({
|
|
path = "", // ADD only
|
|
docs = [], // not for ADD
|
|
data = {}, // not for DELETE
|
|
type = BATCH_TYPE.UPDATE,
|
|
}) {
|
|
try {
|
|
if (docs?.length === 0 && type !== BATCH_TYPE.ADD) {
|
|
console.log(`Aucun document trouvé dans la collection ${path}.`);
|
|
return;
|
|
}
|
|
|
|
// Préparer des lots pour les opérations
|
|
const batches = [];
|
|
let currentBatch = admin.firestore().batch();
|
|
let operationCounter = 0;
|
|
|
|
docs.forEach((doc) => {
|
|
const ref =
|
|
doc?.ref ||
|
|
(typeof doc?.path === "string"
|
|
? admin.firestore().doc(doc.path)
|
|
: null);
|
|
const payload =
|
|
doc && typeof doc.data === "object" && doc.data !== null && !Array.isArray(doc.data)
|
|
? doc.data
|
|
: data;
|
|
|
|
if (type === BATCH_TYPE.ADD) {
|
|
// Pour ADD, créer un nouveau document avec ID automatique
|
|
const newDocRef = admin.firestore().collection(path).doc();
|
|
currentBatch.set(newDocRef, data);
|
|
} else if (type === BATCH_TYPE.UPDATE) {
|
|
if (!ref) {
|
|
throw new Error("batchFirestore UPDATE nécessite une référence de document.");
|
|
}
|
|
if (!payload || Object.keys(payload).length === 0) {
|
|
throw new Error("batchFirestore UPDATE nécessite des données à écrire.");
|
|
}
|
|
currentBatch.set(ref, payload, { merge: true });
|
|
} else if (type === BATCH_TYPE.DELETE) {
|
|
if (!ref) {
|
|
throw new Error("batchFirestore DELETE nécessite une référence de document.");
|
|
}
|
|
currentBatch.delete(ref);
|
|
} else {
|
|
throw new Error(`Opération non supportée : ${type}`);
|
|
}
|
|
operationCounter++;
|
|
// Si le lot atteint la limite de 500 opérations, le sauvegarder et en créer un nouveau
|
|
if (operationCounter === 500) {
|
|
batches.push(currentBatch);
|
|
currentBatch = admin.firestore().batch();
|
|
operationCounter = 0;
|
|
}
|
|
});
|
|
|
|
// Ajouter le dernier lot si des opérations y sont présentes
|
|
if (operationCounter > 0) {
|
|
batches.push(currentBatch);
|
|
}
|
|
|
|
// Exécuter tous les lots en parallèle
|
|
await Promise.all(batches.map((batch) => batch.commit()));
|
|
} catch (error) {
|
|
console.error(
|
|
`Erreur lors des opérations ${type} pour la collection ${path} :`,
|
|
error,
|
|
);
|
|
}
|
|
}
|
|
|
|
exports.batchFirestore = batchFirestore;
|
|
exports.deleteFolder = deleteFolder;
|