support Iterators.take(), same as first()

Author: aplavin

src/readfuncs.jl

@@ -27,7 +27,9 @@ Base.unique(dbc::SQLCollection) = @modify(dbc.query) do q
 	q |> Group(colnames(dbc)...) |> Select(colnames(dbc)...)
 end
 
-Base.first(dbc::SQLCollection, n::Integer) = @modify(dbc.query) do q
+Base.first(dbc::SQLCollection, n::Integer) = Iterators.take(dbc, n)
+
+Iterators.take(dbc::SQLCollection, n::Integer) = @modify(dbc.query) do q
 	q |> Limit(n)
 end
 

test/runtests.jl

@@ -85,7 +85,7 @@ using TestItemRunner
             (@f sort(by=(@o _.i), rev=true)),
             (@f filter(@o _.j ∈ (0.1, 0.5, 0.6) || _.i > 8) sort(by=(@o _.i))),
             (@f sort(by=(@o (_.i, -_.j)), rev=true) first(__, 2)),
-            (@f sort(by=(@o (_.i, -_.j)), rev=true) Iterators.drop(__, 5) first(__, 2)),
+            (@f sort(by=(@o (_.i, -_.j)), rev=true) Iterators.drop(__, 5) Iterators.take(__, 2) first(__, 3)),
         ]
             if f isa Tuple
                 dbs, f = f