From 8ed444cbc904c6750c6a57845c4afaeb56a1d577 Mon Sep 17 00:00:00 2001
From: cuixiaorui <cui_xiaorui@126.com>
Date: Sun, 7 Jul 2024 12:37:14 +0800
Subject: [PATCH] chore: Another way to create enumerations

---
 .github/workflows/ci.yaml  | 4 ----
 packages/db/init.sql       | 1 -
 packages/db/src/migrate.ts | 4 +++-
 3 files changed, 3 insertions(+), 6 deletions(-)
 delete mode 100644 packages/db/init.sql

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index c67e8051f..e31492aa3 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -49,10 +49,6 @@ jobs:
       - name: Install dependencies
         run: pnpm install
 
-      - name: Execute SQL script
-        run: |
-          psql -U test -d earthworm_test -h localhost -p 5480 -f /packages/db/init.sql
-
       - name: Run database initialization command
         run: pnpm db:init:test:ci
 
diff --git a/packages/db/init.sql b/packages/db/init.sql
deleted file mode 100644
index be8d714fb..000000000
--- a/packages/db/init.sql
+++ /dev/null
@@ -1 +0,0 @@
-CREATE TYPE membership_type AS ENUM ('regular', 'premium');
\ No newline at end of file
diff --git a/packages/db/src/migrate.ts b/packages/db/src/migrate.ts
index 522fdda19..8d19187f6 100644
--- a/packages/db/src/migrate.ts
+++ b/packages/db/src/migrate.ts
@@ -1,10 +1,12 @@
+import { sql } from "drizzle-orm";
 import { migrate } from "drizzle-orm/postgres-js/migrator";
 
 import { db } from "./db";
 
 async function main() {
   console.log("Running your migrations...");
-
+  // 创建一个枚举类型
+  await db.execute(sql`CREATE TYPE membership_type AS ENUM ('regular', 'premium');`);
   await migrate(db, { migrationsFolder: "drizzle" });
   console.log("Woohoo! Migrations completed!");
   return;