@nxtedition/rocksdb 11.0.8 → 11.1.0

package/binding.cc

@@ -18,10 +18,12 @@
 #include <rocksdb/table.h>
 #include <rocksdb/write_batch.h>
 
+
 #include <iostream>
 #include <memory>
 #include <optional>
 #include <set>
+#include <regex>
 #include <string>
 #include <thread>
 #include <vector>
@@ -670,6 +672,7 @@ napi_status InitOptions(napi_env env, T& columnOptions, const U& options) {
       columnOptions.compaction_style = rocksdb::kCompactionStyleUniversal;
       columnOptions.compaction_options_universal.compression_size_percent = 80;
     } else if (*compactionOpt == "level") {
+      columnOptions.write_buffer_size = static_cast<size_t>(memtable_memory_budget / 4);
       // merge two memtables when flushing to L0
       columnOptions.min_write_buffer_number_to_merge = 2;
       // this means we'll use 50% extra memory in the worst case, but will reduce
@@ -1478,6 +1481,69 @@ NAPI_METHOD(batch_iterate) {
   return result;
 }
 
+NAPI_METHOD(regex_init) {
+  NAPI_ARGV(1);
+
+  std::string pattern;
+  NAPI_STATUS_THROWS(GetString(env, argv[0], pattern));
+
+  auto batch = std::make_unique<std::regex>(pattern);
+
+  napi_value result;
+  NAPI_STATUS_THROWS(napi_create_external(env, batch.get(), Finalize<std::regex>, batch.get(), &result));
+  batch.release();
+
+  return result;
+}
+
+NAPI_METHOD(regex_test) {
+  NAPI_ARGV(2);
+
+  std::regex* regex;
+  NAPI_STATUS_THROWS(napi_get_value_external(env, argv[0], reinterpret_cast<void**>(&regex)));
+
+  rocksdb::Slice value;
+  NAPI_STATUS_THROWS(GetValue(env, argv[1], value));
+
+  const bool match = std::regex_match(value.data(), value.data() + value.size(), *regex);
+
+  napi_value result;
+  NAPI_STATUS_THROWS(napi_get_boolean(env, match, &result));
+
+  return result;
+}
+
+NAPI_METHOD(regex_test_many) {
+  NAPI_ARGV(2);
+
+  std::regex* regex;
+  NAPI_STATUS_THROWS(napi_get_value_external(env, argv[0], reinterpret_cast<void**>(&regex)));
+
+  uint32_t count;
+  NAPI_STATUS_THROWS(napi_get_array_length(env, argv[1], &count));
+
+  napi_value result;
+  NAPI_STATUS_THROWS(napi_create_array_with_length(env, count, &result));
+
+  for (uint32_t n = 0; n < count; n++) {
+    napi_value valueElement;
+    NAPI_STATUS_THROWS(napi_get_element(env, argv[1], n, &valueElement));
+
+    rocksdb::Slice value;
+    NAPI_STATUS_THROWS(GetValue(env, valueElement, value));
+
+    const bool match = std::regex_match(value.data(), value.data() + value.size(), *regex);
+
+    napi_value matchElement;
+    NAPI_STATUS_THROWS(napi_get_boolean(env, match, &matchElement));
+
+    NAPI_STATUS_THROWS(napi_set_element(env, result, n, matchElement));
+  }
+
+  return result;
+}
+
+
 NAPI_INIT() {
   NAPI_EXPORT_FUNCTION(db_init);
   NAPI_EXPORT_FUNCTION(db_open);
@@ -1505,4 +1571,8 @@ NAPI_INIT() {
   NAPI_EXPORT_FUNCTION(batch_merge);
   NAPI_EXPORT_FUNCTION(batch_count);
   NAPI_EXPORT_FUNCTION(batch_iterate);
+
+  NAPI_EXPORT_FUNCTION(regex_init);
+  NAPI_EXPORT_FUNCTION(regex_test_many);
+  NAPI_EXPORT_FUNCTION(regex_test);
 }

package/index.js

@@ -5,6 +5,7 @@ const { AbstractLevel } = require('abstract-level')
 const ModuleError = require('module-error')
 const binding = require('./binding')
 const { ChainedBatch } = require('./chained-batch')
+const { Regex } = require('./regex')
 const { Iterator } = require('./iterator')
 const fs = require('node:fs')
 const assert = require('node:assert')
@@ -264,3 +265,4 @@ class RocksLevel extends AbstractLevel {
 }
 
 exports.RocksLevel = RocksLevel
+exports.Regex = Regex

package/iterator.js

@@ -107,16 +107,19 @@ class Iterator extends AbstractIterator {
   }
 
   _close (callback) {
-    this[kCache] = kEmpty
-
     try {
-      binding.iterator_close(this[kContext])
+      this._syncClose()
       process.nextTick(callback)
     } catch (err) {
       process.nextTick(callback, err)
     }
   }
 
+  _syncClose () {
+    this[kCache] = kEmpty
+    binding.iterator_close(this[kContext])
+  }
+
   _end (callback) {
     this._close(callback)
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nxtedition/rocksdb",
-  "version": "11.0.8",
+  "version": "11.1.0",
   "description": "A low-level Node.js RocksDB binding",
   "license": "MIT",
   "main": "index.js",

package/regex.js

@@ -0,0 +1,21 @@
+'use strict'
+
+const binding = require('./binding')
+
+class Regex {
+  #context
+
+  constructor (pattern) {
+    this.#context = binding.regex_init(pattern)
+  }
+
+  test (value) {
+    return binding.regex_test(this.#context, value)
+  }
+
+  testMany (values) {
+    return binding.regex_test_many(this.#context, values)
+  }
+}
+
+exports.Regex = Regex