Add Rollback button for user to quickly rollback change

We currently only support rollback when issue is in the closed state and the issue type is schema update. We will create a rollback issue in the reverse order of the original issue for the schema updated tasks having the rollback statement
bytebase · Aug 30, 2021 · b124888 · b124888
1 parent 2ebd43b
commit b124888
Show file tree

Hide file tree

Showing 4 changed files with 270 additions and 60 deletions.
diff --git a/frontend/src/components/IssueStatusTransitionButtonGroup.vue b/frontend/src/components/IssueStatusTransitionButtonGroup.vue
@@ -85,6 +85,11 @@
             {{ transition.buttonName }}
           </button>
         </template>
+        <template v-if="allowRollback">
+          <button type="button" class="btn-primary" @click.prevent="doRollback">
+            Rollback
+          </button>
+        </template>
       </div>
     </template>
   </template>
@@ -170,12 +175,16 @@ export type IssueContext = {
 
 export default {
   name: "IssueStatusTransitionButtonGroup",
-  emits: ["create", "change-issue-status", "change-task-status"],
+  emits: ["create", "rollback", "change-issue-status", "change-task-status"],
   props: {
     create: {
       required: true,
       type: Boolean,
     },
+    allowRollback: {
+      required: true,
+      type: Boolean,
+    },
     issue: {
       required: true,
       type: Object as PropType<Issue | IssueCreate>,
@@ -424,6 +433,10 @@ export default {
       emit("create");
     };
 
+    const doRollback = () => {
+      emit("rollback");
+    };
+
     return {
       updateStatusModalState,
       applicableTaskStatusTransitionList,
@@ -435,6 +448,7 @@ export default {
       doIssueStatusTransition,
       allowCreate,
       doCreate,
+      doRollback,
     };
   },
 };

diff --git a/frontend/src/plugins/issue/DatabaseSchemaUpdateTemplate.ts b/frontend/src/plugins/issue/DatabaseSchemaUpdateTemplate.ts
@@ -10,7 +10,7 @@ const template: IssueTemplate = {
     const stageList: StageCreate[] = [];
     for (let i = 0; i < ctx.databaseList.length; i++) {
       stageList.push({
-        name: `[${ctx.databaseList[i].instance.environment.name}] ${ctx.databaseList[i].name}`,
+        name: `[${ctx.environmentList[i].name}] ${ctx.databaseList[i].name}`,
         environmentId: ctx.environmentList[i].id,
         taskList: [
           {
@@ -22,8 +22,10 @@ const template: IssueTemplate = {
             type: "bb.task.database.schema.update",
             instanceId: ctx.databaseList[i].instance.id,
             databaseId: ctx.databaseList[i].id,
-            statement: "",
-            rollbackStatement: "",
+            statement: ctx.statementList ? ctx.statementList[i] : "",
+            rollbackStatement: ctx.rollbackStatementList
+              ? ctx.rollbackStatementList[i]
+              : "",
           },
         ],
       });

diff --git a/frontend/src/plugins/types.ts b/frontend/src/plugins/types.ts
@@ -1,5 +1,5 @@
 import { Store } from "vuex";
-import { Environment, Principal, IssueCreate, Issue, Database } from "../types";
+import { Database, Environment, Issue, IssueCreate, Principal } from "../types";
 
 // Issue
 // It has to be string type because the id for stage field contain multiple parts.
@@ -86,6 +86,8 @@ export type TemplateContext = {
   databaseList: Database[];
   environmentList: Environment[];
   currentUser: Principal;
+  statementList?: string[];
+  rollbackStatementList?: string[];
 };
 
 export type IssueTemplate = {