Class RocksDatabase

ZezeJava/ZezeJava/src/main/java/Zeze/Util/RocksDatabase.java:25–896

import org.rocksdb.*;

public class RocksDatabase extends ReentrantLock implements Closeable {
	static {
		RocksDB.loadLibrary();
	}

	private static final @NotNull Logger logger = LogManager.getLogger(RocksDatabase.class);
	private static final LRUCache dbCache = new LRUCache(Str.parseLongSize(System.getProperty("rocksdbCache"), 64 << 20));
	private static final TableFormatConfig tableCfg = new BlockBasedTableConfig().setBlockCache(dbCache);
	private static final long dbBuffer = Str.parseLongSize(System.getProperty("rocksdbBuffer"), 64 << 20);
	private static final Options commonOptions = new Options()
			.setCreateIfMissing(true)
			.setTableFormatConfig(tableCfg)
			.setDbWriteBufferSize(dbBuffer) // total write buffer bytes, include all the columns
			.setKeepLogFileNum(5); // reserve "LOG.old.*" file count
	private static final DBOptions commonDbOptions = new DBOptions()
			.setCreateIfMissing(true)
			.setDbWriteBufferSize(dbBuffer) // total write buffer bytes, include all the columns
			.setKeepLogFileNum(5) // reserve "LOG.old.*" file count
			// .setAtomicFlush(true); // atomic batch 独立于这个选项？
			.setMaxWriteBatchGroupSizeBytes(100 * 1024 * 1024);
	private static final ColumnFamilyOptions commonCfOptions = new ColumnFamilyOptions()
			.setTableFormatConfig(tableCfg);
	private static final ReadOptions defaultReadOptions = new ReadOptions();
	private static final WriteOptions defaultWriteOptions = new WriteOptions();
	private static final WriteOptions syncWriteOptions = new WriteOptions().setSync(true);
	private static final TransactionDBOptions transactionDbOptions = new TransactionDBOptions();
	private static final @NotNull MethodHandle mhWriteBatchPutCf;
	private static final @NotNull MethodHandle mhWriteBatchDeleteCf;
	private static final @NotNull MethodHandle mhWriteBatchNativeNew;
	private static final @NotNull MethodHandle mhWriteBatchNew;
	private static final @Nullable ZezeCounter.LongObserver rocksDbGetObserver;
	private static final @Nullable ZezeCounter.LongObserver rocksDbPutObserver;
	private static final @Nullable ZezeCounter.LongObserver rocksDbDeleteObserver;
	private static final @Nullable ZezeCounter.LongObserver rocksDbDeleteRangeObserver;
	private static final @Nullable ZezeCounter.LongObserver rocksDbWriteObserver;
	private static final @Nullable ZezeCounter.LongObserver rocksDbTxnPutObserver;
	private static final @Nullable ZezeCounter.LongObserver rocksDbTxnDeleteObserver;
	private static final @Nullable ZezeCounter.LongObserver rocksDbCompactObserver;

	static {
		try {
			var lookup = MethodHandles.lookup();
			var clsWriteBatch = WriteBatch.class;
			// native void put(long handle, byte[] key, int keyLen, byte[] value, int valueLen, long cfHandle);
			var m = clsWriteBatch.getDeclaredMethod("put",
					long.class, byte[].class, int.class, byte[].class, int.class, long.class);
			m.setAccessible(true);
			mhWriteBatchPutCf = lookup.unreflect(m);
			// native void delete(long handle, byte[] key, int keyLen, long cfHandle);
			m = clsWriteBatch.getDeclaredMethod("delete", long.class, byte[].class, int.class, long.class);
			m.setAccessible(true);
			mhWriteBatchDeleteCf = lookup.unreflect(m);
			// native static long newWriteBatch(byte[] serialized, int serializedLength)
			m = clsWriteBatch.getDeclaredMethod("newWriteBatch", byte[].class, int.class);
			m.setAccessible(true);
			mhWriteBatchNativeNew = lookup.unreflect(m);
			// WriteBatch(long nativeHandle, boolean owningNativeHandle)
			var c = clsWriteBatch.getDeclaredConstructor(long.class, boolean.class);