Add CollectionType

noppoMan · Feb 20, 2017 · d337f7d · d337f7d
1 parent 762bd97
commit d337f7d
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/Sources/SwiftKnex/Knex+Query.swift b/Sources/SwiftKnex/Knex+Query.swift
@@ -8,6 +8,13 @@
 
 extension Knex {
 
+    public func fetch<T: CollectionType>(trx: Connection? = nil) throws -> T? {
+        guard let rows = try execute(.select, trx).asResultSet() else {
+            return nil
+        }
+        return try T(rows: rows)
+    }
+
     public func fetch<T: Entity>(trx: Connection? = nil) throws -> [T] {
         let rows = try execute(.select, trx).asResultSet()
         return try rows?.map { try T(row: $0) } ?? []

diff --git a/Sources/SwiftKnex/Protocols/Collection.swift b/Sources/SwiftKnex/Protocols/Collection.swift
@@ -0,0 +1,16 @@
+//
+//  Collection.swift
+//  SwiftJNChatApp
+//
+//  Created by Yuki Takei on 2017/02/20.
+//
+//
+
+public protocol CollectionSerializable {
+    func serialize() throws -> [[String: Any]]
+}
+
+public protocol CollectionType {
+    init(rows: ResultSet) throws
+}
+
diff --git a/Sources/SwiftKnex/Entity/Entity.swift → Sources/SwiftKnex/Protocols/Entity.swift b/Sources/SwiftKnex/Entity/Entity.swift → Sources/SwiftKnex/Protocols/Entity.swift