rocksdb-ruby 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9d4a29a50778a83f758f3ce02bf456fad90700d6
-  data.tar.gz: c38fef9d063b4efc8a3bc198c99bcfc94b5ec165
+  metadata.gz: 7004e4d2e482db223112f32242d96ac95903eca3
+  data.tar.gz: 8c4c17f150b89f4ac103e0cf25efa4248b662db3
 SHA512:
-  metadata.gz: e34957be12c20bde777b651a3fbc78745a115ffc7e6b11cdf6a8589dd4646aefb9518e618b8e143428cdaf7092b0a2c8cc3e1cabed2ab2dd5193d11c1dd74380
-  data.tar.gz: 31881bfefa0ee01bf24f640f7c280cce52b2a4246e65fff61d179fc1438b62588a9619395ece4d4cbc1ef5df2521ef64aebaaac47a7d609285dc7c579c054993
+  metadata.gz: c0fbfd343a345996ef5f399f3de46e4947a1d3d9133ad067ad54e3c77b6f599e77afd3655506ba48d0d6366066d9b2bc46e2d9a3f9f05aa11c6438cdb7b68f4c
+  data.tar.gz: e598affdc5c818af3e9ad5644000d68da9bea2bc079d19c2353b3dad4601f6e601c435a015a7760d56d2a7d88ee96491e671cdc6cef2ea2406a297f70b219a15

data/ext/rocksdb/rocksdb_db_rb.cc

@@ -131,6 +131,48 @@ extern "C" {
     return klass;
   }
 
+
+  VALUE rocksdb_db_each(VALUE self){
+    rocksdb_pointer* db_pointer;
+    Data_Get_Struct(self, rocksdb_pointer, db_pointer);
+    rocksdb::Iterator* it = db_pointer->db->NewIterator(rocksdb::ReadOptions());
+
+    long i;
+    volatile VALUE ary = rb_ary_new();
+
+    i = 0;
+    for (it->SeekToFirst(); it->Valid(); it->Next()) {
+      VALUE v = rb_enc_str_new(it->value().data(), it->value().size(), rb_utf8_encoding());
+      rb_ary_store(ary, i, v);
+      rb_yield(RARRAY_PTR(ary)[i]);
+      i++;
+    }
+    delete it;
+
+    return ary;
+  }
+
+  VALUE rocksdb_db_reverse_each(VALUE self){
+    rocksdb_pointer* db_pointer;
+    Data_Get_Struct(self, rocksdb_pointer, db_pointer);
+    rocksdb::Iterator* it = db_pointer->db->NewIterator(rocksdb::ReadOptions());
+
+    long i;
+    volatile VALUE ary = rb_ary_new();
+
+    i = 0;
+    for (it->SeekToLast(); it->Valid(); it->Prev()) {
+      VALUE v = rb_enc_str_new(it->value().data(), it->value().size(), rb_utf8_encoding());
+      rb_ary_store(ary, i, v);
+      rb_yield(RARRAY_PTR(ary)[i]);
+      i++;
+    }
+    delete it;
+
+    return ary;
+
+  }
+
   VALUE rocksdb_db_debug(VALUE self){
 
     rocksdb_pointer* db_pointer;

data/ext/rocksdb/rocksdb_db_rb.h

@@ -13,5 +13,8 @@ extern "C" {
   VALUE rocksdb_db_close(VALUE self);
   VALUE rocksdb_db_debug(VALUE self);
   VALUE rocksdb_db_new_iterator(VALUE self);
+  VALUE rocksdb_db_each(VALUE self);
+  VALUE rocksdb_db_each_index(VALUE self);
+  VALUE rocksdb_db_reverse_each(VALUE self);
 
 }

data/ext/rocksdb/rocksdb_iterator_rb.cc

@@ -13,6 +13,26 @@ extern "C" {
 
     return Qnil;
   }
+
+  VALUE rocksdb_iterator_seek_to_last(VALUE klass){
+    rocksdb_iterator_pointer* rocksdb_it;
+    Data_Get_Struct(klass, rocksdb_iterator_pointer , rocksdb_it);
+    rocksdb_it->it->SeekToLast();
+
+    return Qnil;
+  }
+
+  VALUE rocksdb_iterator_seek(VALUE klass, VALUE v_target){
+    Check_Type(v_target, T_STRING);
+    rocksdb::Slice target = rocksdb::Slice((char*)RSTRING_PTR(v_target));
+
+    rocksdb_iterator_pointer* rocksdb_it;
+    Data_Get_Struct(klass, rocksdb_iterator_pointer , rocksdb_it);
+    rocksdb_it->it->Seek(target);
+
+    return Qnil;
+  }
+
   VALUE rocksdb_iterator_alloc(VALUE klass){
     rocksdb_iterator_pointer* it = ALLOC(rocksdb_iterator_pointer);
     return Data_Wrap_Struct(klass, 0, -1, it);

data/ext/rocksdb/rocksdb_iterator_rb.h

@@ -8,6 +8,8 @@ extern "C" {
   typedef VALUE (*METHOD)(...);
 
   VALUE rocksdb_iterator_seek_to_first(VALUE klass);
+  VALUE rocksdb_iterator_seek_to_last(VALUE klass);
+  VALUE rocksdb_iterator_seek(VALUE klass, VALUE v_target);
   VALUE rocksdb_iterator_alloc(VALUE klass);
   VALUE rocksdb_iterator_valid(VALUE klass);
   VALUE rocksdb_iterator_key(VALUE klass);

data/ext/rocksdb/rocksdb_rb.cc

@@ -30,6 +30,9 @@ extern "C" {
     rb_define_method(cRocksdb_db, "debug", (METHOD)rocksdb_db_debug, 0);
     rb_define_method(cRocksdb_db, "new_iterator", (METHOD)rocksdb_db_new_iterator, 0);
 
+    rb_define_method(cRocksdb_db, "each", (METHOD)rocksdb_db_each, 0);
+    rb_define_method(cRocksdb_db, "reverse_each", (METHOD)rocksdb_db_reverse_each, 0);
+
     cRocksdb_write_batch = rb_define_class_under(cRocksdb, "Batch", rb_cObject);
     rb_define_alloc_func(cRocksdb_write_batch, batch_alloc);
     rb_define_private_method(cRocksdb_write_batch, "initialize", (METHOD)rocksdb_write_batch_init, 0);
@@ -39,6 +42,8 @@ extern "C" {
     cRocksdb_iterator = rb_define_class_under(cRocksdb, "Iterator", rb_cObject);
     rb_define_alloc_func(cRocksdb_iterator, rocksdb_iterator_alloc);
     rb_define_method(cRocksdb_iterator, "seek_to_first", (METHOD)rocksdb_iterator_seek_to_first, 0);
+    rb_define_method(cRocksdb_iterator, "seek_to_last", (METHOD)rocksdb_iterator_seek_to_last, 0);
+    rb_define_method(cRocksdb_iterator, "seek", (METHOD)rocksdb_iterator_seek, 1);
     rb_define_method(cRocksdb_iterator, "valid", (METHOD)rocksdb_iterator_valid, 0);
     rb_define_method(cRocksdb_iterator, "key", (METHOD)rocksdb_iterator_key, 0);
     rb_define_method(cRocksdb_iterator, "value", (METHOD)rocksdb_iterator_value, 0);

data/lib/rocksdb/ruby/version.rb

@@ -1,5 +1,5 @@
 module Rocksdb
   module Ruby
-    VERSION = "0.0.5"
+    VERSION = "0.0.6"
   end
 end

data/spec/db_spec.rb

@@ -4,7 +4,7 @@ require "RocksDB"
 
 describe RocksDB do
   before do
-    @rocksdb = RocksDB::DB.new "/tmp/file3"
+    @rocksdb = RocksDB::DB.new "/tmp/file"
   end
 
   it 'should get data' do
@@ -65,19 +65,23 @@ describe RocksDB do
     expect(@rocksdb.get("test:japanese")).to eq "あいうえお"
   end
 
-  it 'should use iterator' do
-    iterator = @rocksdb.new_iterator
+  it 'should use each' do
+    array = []
+    @rocksdb.each do |value|
+      expect(value).not_to be_empty
+      array << value
+    end
 
-    iterator.seek_to_first
-    
-    while(iterator.valid)
-      expect(iterator.value).not_to be_empty
-      expect(iterator.key).not_to be_empty
-      iterator.next
+    rev_array = []
+    @rocksdb.reverse_each do |value|
+      expect(value).not_to be_empty
+      rev_array << value
     end
-    iterator.close
-  end
 
+    expect(array).to eq rev_array.reverse
+    
+  end
+  
   after do
     @rocksdb.close
   end

data/spec/iterator_spec.rb

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+require "RocksDB"
+
+describe RocksDB do
+  before do
+    @rocksdb = RocksDB::DB.new "/tmp/file"
+  end
+
+  it 'should use iterator' do
+    iterator = @rocksdb.new_iterator
+
+    iterator.seek_to_first
+    
+    expect(iterator.valid).to be_true
+    while(iterator.valid)
+      expect(iterator.value).not_to be_empty
+      expect(iterator.key).not_to be_empty
+      iterator.next
+    end
+    iterator.close
+  end
+
+  it 'should seek iterator' do
+    iterator = @rocksdb.new_iterator
+
+    iterator.seek("test:put")
+    
+    iterator.valid
+    expect(iterator.value).to eq "2"
+    expect(iterator.key).to eq "test:put"
+
+    iterator.close
+  end
+
+  after do
+    @rocksdb.close
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rocksdb-ruby
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Isamu Arimoto
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-11-29 00:00:00.000000000 Z
+date: 2013-11-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -80,6 +80,7 @@ files:
 - lib/rocksdb/ruby/version.rb
 - rocksdb-ruby.gemspec
 - spec/db_spec.rb
+- spec/iterator_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/isamu/rocksdb-ruby
 licenses:
@@ -108,4 +109,5 @@ specification_version: 4
 summary: A simple RocksDB library for Ruby
 test_files:
 - spec/db_spec.rb
+- spec/iterator_spec.rb
 - spec/spec_helper.rb