Take advantage of the fact that the enum is CaseIterable for Postgres…

… as well as MySQL
vapor-community · Sep 28, 2024 · 198709d · 198709d
1 parent 8f4ee7e
commit 198709d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Sources/QueuesFluentDriver/JobModelMigrate.swift b/Sources/QueuesFluentDriver/JobModelMigrate.swift
@@ -25,11 +25,11 @@ public struct JobModelMigration: AsyncSQLMigration {
         switch database.dialect.enumSyntax {
         case .typeName:
             stateEnumType = "\(self.jobsTableString)_storedjobstatus"
-            try await database.create(enum: stateEnumType)
-                .value("pending")
-                .value("processing")
-                .value("completed")
-                .run()
+            var builder = database.create(enum: stateEnumType)
+            for `case` in StoredJobState.allCases {
+                builder = builder.value(`case`.rawValue)
+            }
+            try await builder.run()
         case .inline:
             stateEnumType = "enum('\(StoredJobState.allCases.map(\.rawValue).joined(separator: "','"))')"
         default: