rocksdb-ffi 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 418f08a08ef63ebd020e564c356caffe4b1a0d5b7369671f593c688f3fe1a71a
-  data.tar.gz: 3eeafe7ffc7fc2ca09e97a21584e3d6a90dabf788373fc9205b99bd1cf724bd8
+  metadata.gz: 66601f422c933c7afb867e8f7ec679826bd95f3105a1958a722e7d58f40b9972
+  data.tar.gz: 2d1c8540cdfdcc9130199655c94d701b968e1bec8611d4445ec78201b447469c
 SHA512:
-  metadata.gz: 5473bb61e2338c41efedb1af337640d6ca070395c87f933bab5ce3117807594301eebf55cfbbb6a9eb8f12828de267e1ef300ead98797ba10f1a733210334f34
-  data.tar.gz: 0171a8b32370ff31edff39398e27d1af0a81d98e267cf40494268f08faefd54f010c217c9db7513864e33ffb2fc716aff2eac56b147a3225271aa560b51f38f4
+  metadata.gz: 4f65eee7416e962a23998d80da458b0aa787617eb20aa67d548ceccb02f10aa01dc3b825278bcbfa961339f7a41dbf3e2628a42ecd711fad25979730a3a8595d
+  data.tar.gz: 24b92957d2b0068f1ca2d690734e700e5c3168655ed17a938075f713b61b15f1606705b6df0a9560f31dbb9f3512bb0dfa13f0e1e933bf2770433bf75826b3d2

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 # CHANGELOG
 
+## v0.2.1
+
+* Fix memory leak due to missing `rocksdb_free` in `#get`
+
+## v0.2.0
+
+* Add `#each_prefix`
+
 ## v0.1.0
 
 * Initial release

data/Gemfile.lock

@@ -0,0 +1,59 @@
+PATH
+  remote: .
+  specs:
+    rocksdb-ffi (0.2.1)
+      ffi
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ast (2.4.2)
+    diff-lcs (1.5.0)
+    ffi (1.16.3)
+    json (2.6.3)
+    parallel (1.22.1)
+    parser (3.2.1.0)
+      ast (~> 2.4.1)
+    rainbow (3.1.1)
+    rake (13.0.6)
+    regexp_parser (2.7.0)
+    rexml (3.2.5)
+    rspec (3.12.0)
+      rspec-core (~> 3.12.0)
+      rspec-expectations (~> 3.12.0)
+      rspec-mocks (~> 3.12.0)
+    rspec-core (3.12.1)
+      rspec-support (~> 3.12.0)
+    rspec-expectations (3.12.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-mocks (3.12.3)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-support (3.12.0)
+    rubocop (1.46.0)
+      json (~> 2.3)
+      parallel (~> 1.10)
+      parser (>= 3.2.0.0)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8, < 3.0)
+      rexml (>= 3.2.5, < 4.0)
+      rubocop-ast (>= 1.26.0, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 3.0)
+    rubocop-ast (1.27.0)
+      parser (>= 3.2.1.0)
+    ruby-progressbar (1.11.0)
+    unicode-display_width (2.4.2)
+
+PLATFORMS
+  x86_64-linux
+
+DEPENDENCIES
+  rake (~> 13.0)
+  rocksdb-ffi!
+  rspec (~> 3.0)
+  rubocop
+
+BUNDLED WITH
+   2.3.26

data/README.md

@@ -36,12 +36,26 @@ 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
+* ruby 3.2 with rocksdb 7.9
 
 ## Thread-safety
 

data/lib/rocksdb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class RocksDB
-  VERSION = "0.1.0"
+  VERSION = "0.2.1"
 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
@@ -113,7 +114,11 @@ class RocksDB
 
     return if res.null?
 
-    read_string(res, length)
+    value = read_string(res, length)
+
+    Lib.rocksdb_free(res)
+
+    value
   end
 
   def delete(key)
@@ -164,6 +169,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 +200,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.1
 platform: ruby
 authors:
 - Benjamin Vetter
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-03-01 00:00:00.000000000 Z
+date: 2024-06-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -36,6 +36,7 @@ files:
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
 - Gemfile
+- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile