From 62f6d857b34deb0c87db4fb8296173b8f3006c90 Mon Sep 17 00:00:00 2001
From: c9s <yoan@maicoin.com>
Date: Tue, 19 Jan 2021 23:31:13 +0800
Subject: [PATCH] update compiled migrations package for margin columns

---
 .../20210119232826_add_margin_columns.go      | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 pkg/migrations/20210119232826_add_margin_columns.go

diff --git a/pkg/migrations/20210119232826_add_margin_columns.go b/pkg/migrations/20210119232826_add_margin_columns.go
new file mode 100644
index 000000000..778876cad
--- /dev/null
+++ b/pkg/migrations/20210119232826_add_margin_columns.go
@@ -0,0 +1,43 @@
+package migrations
+
+import (
+	"context"
+
+	"github.com/c9s/rockhopper"
+)
+
+func init() {
+	rockhopper.AddMigration(upAddMarginColumns, downAddMarginColumns)
+}
+
+func upAddMarginColumns(ctx context.Context, tx rockhopper.SQLExecutor) (err error) {
+	// This code is executed when the migration is applied.
+
+	_, err = tx.ExecContext(ctx, "ALTER TABLE `trades`\n    ADD COLUMN `is_margin` BOOLEAN NOT NULL DEFAULT FALSE,\n    ADD COLUMN `is_isolated` BOOLEAN NOT NULL DEFAULT FALSE\n    ;")
+	if err != nil {
+		return err
+	}
+
+	_, err = tx.ExecContext(ctx, "ALTER TABLE `orders`\n    ADD COLUMN `is_margin` BOOLEAN NOT NULL DEFAULT FALSE,\n    ADD COLUMN `is_isolated` BOOLEAN NOT NULL DEFAULT FALSE\n    ;")
+	if err != nil {
+		return err
+	}
+
+	return err
+}
+
+func downAddMarginColumns(ctx context.Context, tx rockhopper.SQLExecutor) (err error) {
+	// This code is executed when the migration is rolled back.
+
+	_, err = tx.ExecContext(ctx, "ALTER TABLE `trades`\n    DROP COLUMN `is_margin`,\n    DROP COLUMN `is_isolated`;")
+	if err != nil {
+		return err
+	}
+
+	_, err = tx.ExecContext(ctx, "ALTER TABLE `orders`\n    DROP COLUMN `is_margin`,\n    DROP COLUMN `is_isolated`;")
+	if err != nil {
+		return err
+	}
+
+	return err
+}