Add randomOrSeed() to Seedable

alchemy-swift · Apr 20, 2022 · da80749 · da80749
1 parent 0396c02
commit da80749
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Sources/Alchemy/SQL/Database/Seeding/Seeder.swift b/Sources/Alchemy/SQL/Database/Seeding/Seeder.swift
@@ -23,6 +23,14 @@ extension Seedable where Self: Model {
 
         return try await rows.insertReturnAll()
     }
+
+    public static func randomOrSeed() async throws -> Self {
+        guard let random = try await random() else {
+            return try await seed()
+        }
+
+        return random
+    }
 }
 
 extension Faker {