@budibase/server/src/workspaceMigrations/migrations/20240604153647_initial_sqs.ts

Budibase Web Server

import { context } from "@budibase/backend-core"
import LinkDocumentImpl from "../../db/linkedRows/LinkDocument"
import { allLinkDocs } from "../../db/utils"
import sdk from "../../sdk"

const migration = async () => {
  const linkDocs = await allLinkDocs()

  const docsToUpdate = []
  for (const linkDoc of linkDocs) {
    if (linkDoc.tableId) {
      // It already had the required data
      continue
    }

    // it already has the junction table ID - no need to migrate
    if (!linkDoc.tableId) {
      const newLink = new LinkDocumentImpl(
        linkDoc.doc1.tableId,
        linkDoc.doc1.fieldName,
        linkDoc.doc1.rowId,
        linkDoc.doc2.tableId,
        linkDoc.doc2.fieldName,
        linkDoc.doc2.rowId
      )
      newLink._id = linkDoc._id!
      newLink._rev = linkDoc._rev
      docsToUpdate.push(newLink)
    }
  }

  const db = context.getWorkspaceDB()
  if (docsToUpdate.length) {
    await db.bulkDocs(docsToUpdate)
  }

  // at the end make sure design doc is ready
  await sdk.tables.sqs.syncDefinition()
}

export default migration