trie_matcher 1.1.1 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: dddc71cc0d9fbd56692fecc169fa82a1c3210f05
-  data.tar.gz: 5014bbd465269d7cf3ccd8d32d9271c29da4f2f3
+  metadata.gz: 6826addd0962f940a478110fdd7f434a47576822
+  data.tar.gz: cbc905b0ad75449a9b8fa06794b76b2bfe9b2d6f
 SHA512:
-  metadata.gz: f1c04ca6e84899d4071dd2fc55b0842de91b089104505eff617c95c2936280e93d1fe625d279a45a48831e306d39dea1e05349c5bb234e20d3b109402e28bcda
-  data.tar.gz: 7826761862533ccf710ac0bfbb8e09eb87b5a3dec208638d6db4fb94bff86afc8cf38337b8eeaf6ac14e85591b01a8f71932b0535f3d7812d92f3638efa2772c
+  metadata.gz: 81d3a5f4b73e38debef8881db1a3095c60adbb7b9bedcb3070281a1616419fb13e68198b13e970a74bbe4998126228dc5cd40ba12321307cf7359114746b1ac7
+  data.tar.gz: 05dba91f89e9d004ac456744e12a19263dd2951a56152e2d39556655723c36d165bd9ad5867c145f2cd0ab89bb417ccf1d2a97a011d3d4db045e346f7add3ee3

data/lib/trie_matcher.rb

@@ -44,6 +44,39 @@ class TrieMatcher
     return previous[:value]
   end
 
+  # Perform a prefix search, and return all values in the trie that have this prefix
+  #
+  # @param prefix [String] the prefix to search the trie with
+  # @return [<Object>] the values that start with the given prefix
+  def match(prefix)
+    result = []
+    current = @root
+    current_prefix = prefix
+
+    while current != nil && current_prefix != ""
+      previous, previous_prefix = current, current_prefix
+      current, current_prefix = next_node(current, current_prefix)
+    end
+
+    unless current
+      if current_prefix
+        return []
+      else
+        next_nodes = previous[:nodes].select { |prefix, node| prefix.start_with?(previous_prefix) }.values
+      end
+    else
+      next_nodes = [current]
+    end
+
+    until next_nodes.empty?
+      current = next_nodes.pop
+      result << current[:value]
+      current[:nodes].each { |prefix, node| next_nodes.push(node) }
+    end
+
+    return result.compact
+  end
+
   private
   # get the node for insertion, splitting shared prefixes into subnodes if necessary
   def find_canididate_insertion_node(current, key)

data/lib/trie_matcher/version.rb

@@ -1,3 +1,3 @@
 class TrieMatcher
-  VERSION = "1.1.1"
+  VERSION = "1.2.0"
 end

data/spec/trie_matcher_spec.rb

@@ -1,70 +1,96 @@
 require 'spec_helper'
 
 describe TrieMatcher do
+  before do
+    @t = TrieMatcher.new
+  end
+
   it 'has a version number' do
     expect(TrieMatcher::VERSION).not_to be nil
   end
 
   it 'should store values' do
-    t = TrieMatcher.new
-    t["foo"] = "bar"
+    @t["foo"] = "bar"
   end
 
   it 'should retrieve stored values' do
-    t = TrieMatcher.new
     value = "bar"
-    t["foo"] = value
-    expect(t["foo"]).to be value
+    @t["foo"] = value
+    expect(@t["foo"]).to be value
   end
 
   it 'should return the stored value' do
-    t = TrieMatcher.new
     value = "bar"
-    expect(t["foo"] = value).to be value
+    expect(@t["foo"] = value).to be value
   end
 
   it 'should store values with shared prefixes' do
-    t = TrieMatcher.new
-    t["cat"] = 1
-    t["car"] = 2
-    expect(t["cat"]).to be 1
-    expect(t["car"]).to be 2
+    @t["cat"] = 1
+    @t["car"] = 2
+    expect(@t["cat"]).to be 1
+    expect(@t["car"]).to be 2
   end
 
   it 'should store keys that are a prefix of other keys' do
-    t = TrieMatcher.new
-    t["catch"] = 1
-    t["cat"] = 2
-    expect(t["catch"]).to be 1
-    expect(t["cat"]).to be 2
+    @t["catch"] = 1
+    @t["cat"] = 2
+    expect(@t["catch"]).to be 1
+    expect(@t["cat"]).to be 2
   end
 
   it 'should store keys that have a prefix of another key' do
-    t = TrieMatcher.new
-    t["cat"] = 1
-    t["catch"] = 2
-    expect(t["cat"]).to be 1
-    expect(t["catch"]).to be 2
+    @t["cat"] = 1
+    @t["catch"] = 2
+    expect(@t["cat"]).to be 1
+    expect(@t["catch"]).to be 2
   end
 
   it 'should do prefix searching' do
-    t = TrieMatcher.new
-    t["cat"] = 1
-    expect(t["cats"]).to be 1
+    @t["cat"] = 1
+    expect(@t["cats"]).to be 1
   end
 
   it 'should do partial prefix matching' do
-    t = TrieMatcher.new
-    t["cat"] = 1
-    t["cats in the cradle"] = 2
-    expect(t["cats"]).to be 1
+    @t["cat"] = 1
+    @t["cats in the cradle"] = 2
+    expect(@t["cats"]).to be 1
   end
 
   it 'should return the more specific prefix value' do
-    t = TrieMatcher.new
-    t["cat"] = 1
-    t["catch"] = 2
-    expect(t["catcher"]).to be 2
+    @t["cat"] = 1
+    @t["catch"] = 2
+    expect(@t["catcher"]).to be 2
+  end
+
+  it 'should not give a longer prefix value' do
+    @t["catch"] = 2
+    expect(@t["cat"]).to be nil
+  end
+
+  it 'should return all values that are a prefix' do
+    @t["cats"] = 1
+    @t["foobar"] = 2
+    @t["foobaz"] = 3
+    @t["foobars"] = 4
+    @t["food"] = 5
+
+    expect(@t.match("foo").sort).to eq [2,3,4,5]
+  end
+
+  it 'should return a single value when there is only one match' do
+    @t["cats"] = 1
+    @t["dogs"] = 2
+    @t["birds"] = 3
+    @t["bees"] = 4
+
+    expect(@t.match("dog")).to eq [2]
   end
 
+  it 'should return an empty result when there are no prefix matches' do
+    @t["cat"] = 1
+    @t["dog"] = 2
+    @t["catch"] = 3
+
+    expect(@t.match("catcher")).to eq []
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trie_matcher
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.2.0
 platform: ruby
 authors:
 - Steven Karas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-07 00:00:00.000000000 Z
+date: 2015-08-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler