add keys values

chenkovsky · Mar 26, 2018 · 6547954 · 6547954
1 parent b57325f
commit 6547954
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/preshed.cr b/src/preshed.cr
@@ -87,6 +87,26 @@ module Preshed
       end
     end
 
+    def keys
+      arr = Array(UInt64).new
+      (0...@capacity).each do |i|
+        if (@cells + i).value.key != EMPTY_KEY
+          arr << (@cells + i).value.key
+        end
+      end
+      return arr
+    end
+
+    def values
+      arr = Array(V).new
+      (0...@capacity).each do |i|
+        if (@cells + i).value.key != EMPTY_KEY
+          arr << (@cells + i).value.value
+        end
+      end
+      return arr
+    end
+
     private def resize
       new_capacity = @capacity * 2
       cells = Pointer(Cell(V)).malloc(new_capacity, Cell(V).new(EMPTY_KEY, @default))