tyler-trie 0.2.0 → 0.2.1

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
-patch: 0
+patch: 1
 major: 0
 minor: 2

data/ext/trie/trie.c

@@ -95,7 +95,7 @@ static VALUE trie_add(VALUE self, VALUE args) {
 
     VALUE key;
     key = RARRAY(args)->ptr[0];
-    short value = size == 2 ? NUM2INT(RARRAY(args)->ptr[1]) : TRIE_DATA_ERROR;
+    int32 value = size == 2 ? NUM2INT(RARRAY(args)->ptr[1]) : TRIE_DATA_ERROR;
     
     const TrieChar *sb_key = stringToTrieChar(key);
     
@@ -117,6 +117,16 @@ static VALUE trie_delete(VALUE self, VALUE key) {
 	return Qnil;
 }
 
+static VALUE trie_save(VALUE self) {
+    SBTrie *sb_trie;
+    Data_Get_Struct(self, SBTrie, sb_trie);
+    
+    if(sb_trie_save(sb_trie) == -1)
+	return Qnil;
+    else
+	return Qtrue;
+}
+
 static VALUE walk_all_paths(VALUE children, SBTrieState *state, char *prefix) {
     int c;
     for(c = 1; c < TRIE_CHAR_MAX; c++) {
@@ -326,6 +336,7 @@ void Init_trie() {
     rb_define_method(cTrie, "children", trie_children, 1);
     rb_define_method(cTrie, "walk_to_terminal", trie_walk_to_terminal, -2);
     rb_define_method(cTrie, "root", trie_root, 0);
+    rb_define_method(cTrie, "save", trie_save, 0);
 
     cTrieNode = rb_define_class("TrieNode", rb_cObject);
     rb_define_alloc_func(cTrieNode, trie_node_alloc);

data/spec/trie_spec.rb

@@ -53,6 +53,11 @@ describe Trie do
       @trie.add('chicka',123).should == true
       @trie.get('chicka').should == 123
     end
+
+    it 'adds values greater than 16-bit allows' do
+      @trie.add('chicka', 72_000).should == true
+      @trie.get('chicka').should == 72_000
+    end
   end
 
   describe :delete do
@@ -109,6 +114,16 @@ describe Trie do
       @trie.root.should_not == @trie.root
     end
   end
+
+  describe :save do
+    it 'saves the trie to disk such that another trie can be spawned which will read succesfully' do
+      @trie.add('omgwtf',123)
+      @trie.save
+
+      trie2 = Trie.new(TRIE_PATH)
+      trie2.get('omgwtf').should == 123
+    end
+  end
 end
 
 describe TrieNode do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: tyler-trie
 version: !ruby/object:Gem::Version 
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors: 
 - Tyler McMullen
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-03-10 00:00:00 -07:00
+date: 2009-03-11 00:00:00 -07:00
 default_executable: 
 dependencies: []