Firebase Firestore batch more than 500 docs or operations

batchWrapper.js

const batchWrapper = async (documentRef, action, update) => {
  const batchArray = [];
  batchArray.push(db.batch());
  let operationCounter = 0;
  let batchIndex = 0;

  documentRef.forEach(doc => {
    console.log('Org cleanup: deleting notifications', doc.id);
    if (action === 'delete') {
      batchArray[batchIndex].delete(doc.ref);
    }
    if (action === 'update') {
      batchArray[batchIndex].update(doc.ref, update);
    }
    operationCounter++;

    if (operationCounter === 499) {
      batchArray.push(db.batch());
      batchIndex++;
      operationCounter = 0;
    }
  });

  batchArray.forEach(async batch => await batch.commit());

  return;
};

const clearOrganizationData = async organizationId => {
  const usersRef = await db
    .collection('users')
    .where('organizationId', '==', organizationId)
    .get();
  const notificationsRef = await db
    .collection('notifications')
    .where('organizationId', '==', organizationId)
    .get();
  const rewardsRef = await db
    .collection('rewards')
    .where('organizationId', '==', organizationId)
    .get();

  await batchWrapper(usersRef, 'delete');
  await batchWrapper(rewardsRef, 'delete');
  await batchWrapper(notificationsRef, 'delete');
};