Add missing migration file

Signed-off-by: Damon Kaswell <damon.kaswell1@hp.com>
2022-10-27 08:33:51 -07:00
parent ad1beb072a
commit 484be3318e
1 changed files with 86 additions and 0 deletions
@@ -0,0 +1,86 @@
+/*
+ * Copyright 2022 The Backstage Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+const { v4: uuidv4 } = require('uuid');
+
+/**
+ * @param { import("knex").Knex } knex
+ * @returns { Promise<void> }
+ */
+exports.up = async function (knex) {
+  const schema = () => knex.schema.withSchema('ingestion');
+
+  await knex.transaction(async tx => {
+    const providers = await tx('ingestion.ingestions').distinct('provider_name');
+    for (const provider of providers) {
+      const valid = await tx('ingestion.ingestions')
+        .where('provider_name', provider)
+        .andWhere('rest_completed_at', null)
+        .first();
+      const invalid = [];
+
+      const rows = await tx('ingestion.ingestions')
+        .where('provider_name', provider)
+        .andWhere('rest_completed_at', null);
+      for (const row of rows) {
+        if (row.id !== valid.id) {
+          invalid.push(row.id);
+        }
+      }
+
+      if (invalid.length > 0) {
+        await tx('ingestion.ingestions').delete().whereIn('id', invalid);
+        await tx('ingestion.ingestion_mark_entities')
+          .delete()
+          .whereIn('ingestion_mark_id', tx('ingestion.ingestion_marks').select('id').whereIn('ingestion_id', invalid));
+        await tx('ingestion.ingestion_marks').delete().whereIn('ingestion_id', invalid);
+      }
+    }
+  });
+
+  await schema().alterTable('ingestions', t => {
+    t.string('completion_ticket');
+  });
+
+  await knex.transaction(async tx => {
+    await tx('ingestion.ingestions').update('completion_ticket', 'open').where('rest_completed_at', null);
+
+    const rows = await tx('ingestion.ingestions').whereNot('rest_completed_at', null);
+    for (const row of rows) {
+      await tx('ingestion.ingestions').update('completion_ticket', uuidv4()).where('id', row.id);
+    }
+  });
+
+  await schema().alterTable('ingestions', t => {
+    t.string('completion_ticket').notNullable().alter();
+    t.unique(['provider_name', 'completion_ticket'], {
+      indexName: 'ingestion_composite_index',
+      deferrable: 'deferred',
+    });
+  });
+};
+
+/**
+ * @param { import("knex").Knex } knex
+ * @returns { Promise<void> }
+ */
+exports.down = async function (knex) {
+  const schema = () => knex.schema.withSchema('ingestion');
+
+  await schema().alterTable('ingestions', t => {
+    t.dropUnique(['provider_name', 'completion_ticket']);
+  });
+};