sql row protocol updates (#66)

vapor · Dec 13, 2019 · 9f3a7fb · 9f3a7fb
1 parent f25b68e
commit 9f3a7fb
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/Sources/SQLiteKit/SQLiteRow+SQLRow.swift b/Sources/SQLiteKit/SQLiteRow+SQLRow.swift
@@ -1,12 +1,24 @@
 extension SQLiteRow: SQLRow {
+    public var allColumns: [String] {
+        .init(self.columns.offsets.keys)
+    }
+
+    public func decodeNil(column: String) throws -> Bool {
+        self.columns.offsets.keys.contains(column)
+    }
+
     public func contains(column: String) -> Bool {
         return self.column(column) != nil
     }
 
     public func decode<D>(column: String, as type: D.Type) throws -> D where D : Decodable {
         guard let data = self.column(column) else {
-            fatalError("no value found for \(column)")
+            throw MissingColumn(column: column)
         }
         return try SQLiteDataDecoder().decode(D.self, from: data)
     }
 }
+
+struct MissingColumn: Error {
+    let column: String
+}