overleaf/tools/migrations/20200120163346_atlas_recommended_indexes.mjs

/* eslint-disable no-unused-vars */

import Helpers from './lib/helpers.mjs'

const tags = ['server-ce', 'server-pro', 'saas']

const indexes = {
  docSnapshots: [{ key: { project_id: 1, ts: 1 }, name: 'project_id_1_ts_1' }],
  deletedProjects: [
    {
      key: { 'deleterData.deletedProjectOwnerId': 1 },
      name: 'deleterdata_deletedProjectOwnerId_1',
    },
  ],
  docs: [{ key: { project_id: 1, inS3: 1 }, name: 'project_id_1_inS3_1' }],
}

const migrate = async client => {
  const { db } = client

  await Promise.all(
    Object.keys(indexes).map(key =>
      Helpers.addIndexesToCollection(db[key], indexes[key])
    )
  )
}

const rollback = async client => {
  const { db } = client

  await Promise.all(
    Object.keys(indexes).map(key =>
      Helpers.dropIndexesFromCollection(db[key], indexes[key])
    )
  )
}

export default {
  tags,
  migrate,
  rollback,
}