0aab1005f8
Summary: Previous to this commit ColumnFamilyDescriptor took a String as name for the ColumnFamily name. String is however encoding dependent which is bad because listColumnFamilies returns byte arrays without any encoding information. All public API call were deprecated and flagged to be removed in 3.10.0 Test Plan: make rocksdbjava make test mvn -f rocksjni.pom package Reviewers: yhchiang, adamretter, ankgup87 Subscribers: dhruba Differential Revision: https://reviews.facebook.net/D30525
334 lines
9.9 KiB
Java
334 lines
9.9 KiB
Java
// Copyright (c) 2014, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
package org.rocksdb.test;
|
|
|
|
import org.junit.ClassRule;
|
|
import org.junit.Rule;
|
|
import org.junit.Test;
|
|
import org.junit.rules.TemporaryFolder;
|
|
import org.rocksdb.*;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import static org.assertj.core.api.Assertions.assertThat;
|
|
|
|
public class ReadOnlyTest {
|
|
|
|
@ClassRule
|
|
public static final RocksMemoryResource rocksMemoryResource =
|
|
new RocksMemoryResource();
|
|
|
|
@Rule
|
|
public TemporaryFolder dbFolder = new TemporaryFolder();
|
|
|
|
@Test
|
|
public void readOnlyOpen() throws RocksDBException {
|
|
RocksDB db = null;
|
|
RocksDB db2 = null;
|
|
RocksDB db3 = null;
|
|
Options options = null;
|
|
List<ColumnFamilyHandle> columnFamilyHandleList =
|
|
new ArrayList<>();
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList =
|
|
new ArrayList<>();
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList2 =
|
|
new ArrayList<>();
|
|
try {
|
|
options = new Options();
|
|
options.setCreateIfMissing(true);
|
|
|
|
db = RocksDB.open(options,
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
db.put("key".getBytes(), "value".getBytes());
|
|
db2 = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
assertThat("value").
|
|
isEqualTo(new String(db2.get("key".getBytes())));
|
|
db.close();
|
|
db2.close();
|
|
|
|
List<ColumnFamilyDescriptor> cfDescriptors = new ArrayList<>();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
|
|
db = RocksDB.open(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors, columnFamilyHandleList);
|
|
columnFamilyHandleList.add(db.createColumnFamily(
|
|
new ColumnFamilyDescriptor("new_cf".getBytes(), new ColumnFamilyOptions())));
|
|
columnFamilyHandleList.add(db.createColumnFamily(
|
|
new ColumnFamilyDescriptor("new_cf2".getBytes(), new ColumnFamilyOptions())));
|
|
db.put(columnFamilyHandleList.get(2), "key2".getBytes(),
|
|
"value2".getBytes());
|
|
|
|
db2 = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors,
|
|
readOnlyColumnFamilyHandleList);
|
|
assertThat(db2.get("key2".getBytes())).isNull();
|
|
assertThat(db2.get(readOnlyColumnFamilyHandleList.get(0), "key2".getBytes())).
|
|
isNull();
|
|
cfDescriptors.clear();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor("new_cf2".getBytes(), new ColumnFamilyOptions()));
|
|
db3 = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors, readOnlyColumnFamilyHandleList2);
|
|
assertThat(new String(db3.get(readOnlyColumnFamilyHandleList2.get(1),
|
|
"key2".getBytes()))).isEqualTo("value2");
|
|
} finally {
|
|
if (db != null) {
|
|
db.close();
|
|
}
|
|
if (db2 != null) {
|
|
db2.close();
|
|
}
|
|
if (db3 != null) {
|
|
db3.close();
|
|
}
|
|
if (options != null) {
|
|
options.dispose();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(expected = RocksDBException.class)
|
|
public void failToWriteInReadOnly() throws RocksDBException {
|
|
RocksDB db = null;
|
|
RocksDB rDb = null;
|
|
Options options = null;
|
|
try {
|
|
List<ColumnFamilyDescriptor> cfDescriptors = new ArrayList<>();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList =
|
|
new ArrayList<>();
|
|
options = new Options();
|
|
options.setCreateIfMissing(true);
|
|
|
|
db = RocksDB.open(options,
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
db.close();
|
|
rDb = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors,
|
|
readOnlyColumnFamilyHandleList);
|
|
|
|
// test that put fails in readonly mode
|
|
rDb.put("key".getBytes(), "value".getBytes());
|
|
} finally {
|
|
if (db != null) {
|
|
db.close();
|
|
}
|
|
if (rDb != null) {
|
|
rDb.close();
|
|
}
|
|
if (options != null) {
|
|
options.dispose();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(expected = RocksDBException.class)
|
|
public void failToCFWriteInReadOnly() throws RocksDBException {
|
|
RocksDB db = null;
|
|
RocksDB rDb = null;
|
|
Options options = null;
|
|
try {
|
|
List<ColumnFamilyDescriptor> cfDescriptors = new ArrayList<>();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList =
|
|
new ArrayList<>();
|
|
options = new Options();
|
|
options.setCreateIfMissing(true);
|
|
|
|
db = RocksDB.open(options,
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
db.close();
|
|
rDb = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors,
|
|
readOnlyColumnFamilyHandleList);
|
|
|
|
rDb.put(readOnlyColumnFamilyHandleList.get(0),
|
|
"key".getBytes(), "value".getBytes());
|
|
} finally {
|
|
if (db != null) {
|
|
db.close();
|
|
}
|
|
if (rDb != null) {
|
|
rDb.close();
|
|
}
|
|
if (options != null) {
|
|
options.dispose();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(expected = RocksDBException.class)
|
|
public void failToRemoveInReadOnly() throws RocksDBException {
|
|
RocksDB db = null;
|
|
RocksDB rDb = null;
|
|
Options options = null;
|
|
try {
|
|
List<ColumnFamilyDescriptor> cfDescriptors = new ArrayList<>();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList =
|
|
new ArrayList<>();
|
|
|
|
options = new Options();
|
|
options.setCreateIfMissing(true);
|
|
|
|
db = RocksDB.open(options,
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
db.close();
|
|
rDb = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors,
|
|
readOnlyColumnFamilyHandleList);
|
|
|
|
rDb.remove("key".getBytes());
|
|
} finally {
|
|
if (db != null) {
|
|
db.close();
|
|
}
|
|
if (rDb != null) {
|
|
rDb.close();
|
|
}
|
|
if (options != null) {
|
|
options.dispose();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(expected = RocksDBException.class)
|
|
public void failToCFRemoveInReadOnly() throws RocksDBException {
|
|
RocksDB db = null;
|
|
RocksDB rDb = null;
|
|
Options options = null;
|
|
try {
|
|
List<ColumnFamilyDescriptor> cfDescriptors = new ArrayList<>();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList =
|
|
new ArrayList<>();
|
|
|
|
options = new Options();
|
|
options.setCreateIfMissing(true);
|
|
|
|
db = RocksDB.open(options,
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
db.close();
|
|
|
|
rDb = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors,
|
|
readOnlyColumnFamilyHandleList);
|
|
|
|
rDb.remove(readOnlyColumnFamilyHandleList.get(0),
|
|
"key".getBytes());
|
|
} finally {
|
|
if (db != null) {
|
|
db.close();
|
|
}
|
|
if (rDb != null) {
|
|
rDb.close();
|
|
}
|
|
if (options != null) {
|
|
options.dispose();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(expected = RocksDBException.class)
|
|
public void failToWriteBatchReadOnly() throws RocksDBException {
|
|
RocksDB db = null;
|
|
RocksDB rDb = null;
|
|
Options options = null;
|
|
try {
|
|
List<ColumnFamilyDescriptor> cfDescriptors = new ArrayList<>();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList =
|
|
new ArrayList<>();
|
|
|
|
options = new Options();
|
|
options.setCreateIfMissing(true);
|
|
|
|
db = RocksDB.open(options,
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
db.close();
|
|
|
|
rDb = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors,
|
|
readOnlyColumnFamilyHandleList);
|
|
|
|
WriteBatch wb = new WriteBatch();
|
|
wb.put("key".getBytes(), "value".getBytes());
|
|
rDb.write(new WriteOptions(), wb);
|
|
} finally {
|
|
if (db != null) {
|
|
db.close();
|
|
}
|
|
if (rDb != null) {
|
|
rDb.close();
|
|
}
|
|
if (options != null) {
|
|
options.dispose();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(expected = RocksDBException.class)
|
|
public void failToCFWriteBatchReadOnly() throws RocksDBException {
|
|
RocksDB db = null;
|
|
RocksDB rDb = null;
|
|
Options options = null;
|
|
WriteBatch wb = null;
|
|
try {
|
|
List<ColumnFamilyDescriptor> cfDescriptors = new ArrayList<>();
|
|
cfDescriptors.add(
|
|
new ColumnFamilyDescriptor(RocksDB.DEFAULT_COLUMN_FAMILY,
|
|
new ColumnFamilyOptions()));
|
|
List<ColumnFamilyHandle> readOnlyColumnFamilyHandleList =
|
|
new ArrayList<>();
|
|
|
|
options = new Options();
|
|
options.setCreateIfMissing(true);
|
|
|
|
db = RocksDB.open(options,
|
|
dbFolder.getRoot().getAbsolutePath());
|
|
db.close();
|
|
|
|
rDb = RocksDB.openReadOnly(
|
|
dbFolder.getRoot().getAbsolutePath(), cfDescriptors,
|
|
readOnlyColumnFamilyHandleList);
|
|
|
|
wb = new WriteBatch();
|
|
wb.put(readOnlyColumnFamilyHandleList.get(0),
|
|
"key".getBytes(), "value".getBytes());
|
|
rDb.write(new WriteOptions(), wb);
|
|
} finally {
|
|
if (db != null) {
|
|
db.close();
|
|
}
|
|
if (rDb != null) {
|
|
rDb.close();
|
|
}
|
|
if (options != null) {
|
|
options.dispose();
|
|
}
|
|
if (wb != null) {
|
|
wb.dispose();
|
|
}
|
|
}
|
|
}
|
|
}
|