Merge pull request #422 from mpenet/feature/record-seq

Implement ICounted and ISeqable on records

Author: heyLu

pixie/stdlib.pxi

@@ -11,7 +11,6 @@
 (def printf (ffi-fn libc "printf" [CCharP] CInt :variadic? true))
 (def getenv (ffi-fn libc "getenv" [CCharP] CCharP))
 
-
 (def libedit (ffi-library (str "libedit." pixie.platform/so-ext)))
 (def readline (ffi-fn libedit "readline" [CCharP] CCharP))
 (def rand (ffi-fn libc "rand" [] CInt))
@@ -1286,6 +1285,13 @@ and implements IAssociative, ILookup and IObject."
                                            @acc#
                                            (recur (next fields#) acc#)))
                                        acc#)))
+                        'ICounted
+                        `(-count [self] ~(count fields))
+
+                        'ISeqable
+                        `(-seq [self#]
+                               (map #(map-entry % (get-field self# %))
+                                    ~fields))
 
                         'IMeta
                         `(-with-meta [self# ~meta-gs]

tests/pixie/tests/test-defrecord.pxi

@@ -63,3 +63,11 @@
   (t/assert= [[:one 1] [:two 2] [:three 3]] (reduce conj [] t1))
   (t/assert= [1 2 3] (vals t1))
   (t/assert= [:one :two :three] (keys t1)))
+
+(t/deftest icounted []
+  (t/assert= (count t1) 3))
+
+(t/deftest seqable []
+  (t/assert= (key (first (seq t1))) :one)
+  (t/assert= (val (first (seq t1))) 1)
+  (t/assert (instance? LazySeq (seq t1))))