Merge pull request #7 from hsharghi/async

Fix import a FluentPostgresDriver without checking
hsharghi · Jun 20, 2023 · b438cd9 · b438cd9
2 parents 3890977 + 5fbd025
commit b438cd9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Sources/VaporWallet/WalletRepository.swift b/Sources/VaporWallet/WalletRepository.swift
@@ -7,7 +7,9 @@
 
 import Vapor
 import Fluent
+#if canImport(FluentPostgresDriver)
 import FluentPostgresDriver
+#endif
 
 /// This calss gives access to wallet methods for a `HasWallet` model.
 /// Creating multiple wallets, accessing them and getting balance of each wallet,
@@ -75,6 +77,7 @@ extension WalletsRepository {
         if withUnconfirmed {
             // (1) Temporary workaround for sum and average aggregates,
             var balance: Double
+            #if canImport(FluentPostgresDriver)
             if let _ = self.db as? PostgresDatabase {
                 let balanceOptional = try? await wallet.$transactions
                     .query(on: self.db)
@@ -88,6 +91,13 @@ extension WalletsRepository {
 
                 balance = intBalance == nil ? 0.0 : Double(intBalance!)
             }
+            #else
+            let intBalance = try await wallet.$transactions
+                .query(on: self.db)
+                .sum(\.$amount)
+
+            balance = intBalance == nil ? 0.0 : Double(intBalance!)
+            #endif
             return asDecimal ? balance.toDecimal(with: wallet.decimalPlaces) : balance
         }
         return asDecimal ? Double(wallet.balance).toDecimal(with: wallet.decimalPlaces) : Double(wallet.balance)