rocksdb-ffi 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 418f08a08ef63ebd020e564c356caffe4b1a0d5b7369671f593c688f3fe1a71a
-  data.tar.gz: 3eeafe7ffc7fc2ca09e97a21584e3d6a90dabf788373fc9205b99bd1cf724bd8
+  metadata.gz: 691b849562ba32f10bb834756ddbc9a7bae89af2b06f51d272fcde575eeb23e4
+  data.tar.gz: b8b66067cd541f59d856aa915d7737f515164349e68ac03c694e5b3280fd0e19
 SHA512:
-  metadata.gz: 5473bb61e2338c41efedb1af337640d6ca070395c87f933bab5ce3117807594301eebf55cfbbb6a9eb8f12828de267e1ef300ead98797ba10f1a733210334f34
-  data.tar.gz: 0171a8b32370ff31edff39398e27d1af0a81d98e267cf40494268f08faefd54f010c217c9db7513864e33ffb2fc716aff2eac56b147a3225271aa560b51f38f4
+  metadata.gz: 48ebff505cd2f1d3a98fb7371d5b2b5d121d262dff1e4366a79d8e01198d42a85c841a35b746d3782294643b543905839016a655eb7bd565b9fdc9aee346d5f5
+  data.tar.gz: 6385d064e46e67afefd3497ab5d4b676edf5dedda9eb1b276e40cf1e00d668833871c15eb5c34d86f91139146b95e53e637865e5ec0f8616cbac403f5ef426cf

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## v0.2.0
+
+* Add `#each_prefix`
+
 ## v0.1.0
 
 * Initial release

data/README.md

@@ -36,12 +36,25 @@ rocksdb.each_key do |key|
   # ...
 end
 
+rocksdb.each_prefix("prefix") do |key, value|
+  # ...
+end
+
+rocksdb.flush
+
 rocksdb.close
 
 rocksdb.open("/path/to/other/db")
 ```
 
-The library is tested against ruby 3.0/3.1/3.2 and rocksdb 7.2. That's it for now!
+The library is tested against:
+
+* ruby 2.7 with rocksdb 7.2
+* ruby 3.0 with rocksdb 7.2
+* ruby 3.1 with rocksdb 7.2
+* ruby 3.1 with rocksdb 7.7
+* ruby 3.2 with rocksdb 7.2
+* ruby 3.2 with rocksdb 7.7
 
 ## Thread-safety
 

data/lib/rocksdb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class RocksDB
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/rocksdb.rb

@@ -24,6 +24,7 @@ class RocksDB
     attach_function :rocksdb_create_iterator, [:pointer, :pointer], :pointer
     attach_function :rocksdb_iter_valid, [:pointer], :uchar
     attach_function :rocksdb_iter_seek_to_first, [:pointer], :void
+    attach_function :rocksdb_iter_seek, [:pointer, :string, :size_t], :void
     attach_function :rocksdb_iter_next, [:pointer], :void
     attach_function :rocksdb_iter_key, [:pointer, :pointer], :pointer
     attach_function :rocksdb_iter_value, [:pointer, :pointer], :pointer
@@ -164,6 +165,19 @@ class RocksDB
     end
   end
 
+  def each_prefix(prefix)
+    raise(ClosedError, "Database is closed") if @closed
+
+    return enum_for(__method__, prefix) unless block_given?
+
+    iterate_prefix(prefix) do |key, iterator|
+      value_length = FFI::MemoryPointer.new(:size_t, 1)
+      value = read_string(Lib.rocksdb_iter_value(iterator, value_length), value_length)
+
+      yield(key, value)
+    end
+  end
+
   private
 
   def iterate
@@ -182,6 +196,27 @@ class RocksDB
     end
   end
 
+  def iterate_prefix(prefix)
+    iterator = Lib.rocksdb_create_iterator(@db, @read_options)
+
+    begin
+      Lib.rocksdb_iter_seek(iterator, prefix, prefix.bytesize)
+
+      while Lib.rocksdb_iter_valid(iterator) != 0
+        key_length = FFI::MemoryPointer.new(:size_t, 1)
+        key = read_string(Lib.rocksdb_iter_key(iterator, key_length), key_length)
+
+        return unless key.start_with?(prefix)
+
+        yield(key, iterator)
+
+        Lib.rocksdb_iter_next(iterator)
+      end
+    ensure
+      Lib.rocksdb_iter_destroy(iterator)
+    end
+  end
+
   def check_error(error)
     pointer = error.read_pointer
     return if pointer.null?

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rocksdb-ffi
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Benjamin Vetter
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-03-01 00:00:00.000000000 Z
+date: 2023-03-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi