Class RocksDBStore<K,V>
- java.lang.Object
-
- org.infinispan.persistence.rocksdb.RocksDBStore<K,V>
-
- All Implemented Interfaces:
Lifecycle
,AdvancedCacheLoader<K,V>
,AdvancedCacheWriter<K,V>
,AdvancedLoadWriteStore<K,V>
,CacheLoader<K,V>
,CacheWriter<K,V>
,ExternalStore<K,V>
,SegmentedAdvancedLoadWriteStore<K,V>
public class RocksDBStore<K,V> extends Object implements SegmentedAdvancedLoadWriteStore<K,V>
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from interface org.infinispan.persistence.spi.AdvancedCacheLoader
AdvancedCacheLoader.CacheLoaderTask<K,V>, AdvancedCacheLoader.TaskContext
-
Nested classes/interfaces inherited from interface org.infinispan.persistence.spi.AdvancedCacheWriter
AdvancedCacheWriter.PurgeListener<K>
-
-
Constructor Summary
Constructors Constructor Description RocksDBStore()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
addSegments(IntSet segments)
Invoked when a node becomes an owner of the given segments.void
clear()
Removes all the data from the storage.void
clear(IntSet segments)
Removes all the data that maps to the given segments from the storage.boolean
contains(int segment, Object key)
Returns true if the storage contains an entry associated with the given key in the given segmentboolean
contains(Object key)
Returns true if the storage contains an entry associated with the given key.boolean
delete(int segment, Object key)
Removes the entry for the provided key which is in the given segment.boolean
delete(Object key)
void
deleteBatch(Iterable<Object> keys)
Remove all provided keys from the store in a single batch operation.void
destroy()
Method to be used to destroy and clean up any resources associated with this store.void
init(InitializationContext ctx)
Used to initialize a cache loader.boolean
isAvailable()
MarshalledEntry<K,V>
load(int segment, Object key)
Fetches an entry from the storage given a segment to optimize this lookup based on.MarshalledEntry
load(Object key)
Fetches an entry from the storage.protected org.rocksdb.RocksDB
openDatabase(String location, org.rocksdb.Options options)
Creates database if it doesn't exist.org.reactivestreams.Publisher<MarshalledEntry<K,V>>
publishEntries(Predicate<? super K> filter, boolean fetchValue, boolean fetchMetadata)
Publishes all entries from this store.org.reactivestreams.Publisher<MarshalledEntry<K,V>>
publishEntries(IntSet segments, Predicate<? super K> filter, boolean fetchValue, boolean fetchMetadata)
Publishes all entries from this store.org.reactivestreams.Publisher<K>
publishKeys(Predicate<? super K> filter)
Publishes all the keys from this store.org.reactivestreams.Publisher<K>
publishKeys(IntSet segments, Predicate<? super K> filter)
Publishes all the keys that map to the given segments from this store.void
purge(Executor executor, AdvancedCacheWriter.PurgeListener purgeListener)
Using the thread in the pool, removed all the expired data from the persistence storage.void
removeSegments(IntSet segments)
Invoked when a node loses ownership of a segment.int
size()
Returns the number of elements in the store.int
size(IntSet segments)
Returns the number of elements in the store that map to the given segments that aren't expired.void
start()
Invoked on component startvoid
stop()
Invoked on component stopvoid
write(int segment, MarshalledEntry<? extends K,? extends V> entry)
Persists the entry to the storage with the given segment to optimize further lookups based onvoid
write(MarshalledEntry entry)
Persists the entry to the storage.void
writeBatch(Iterable<MarshalledEntry<? extends K,? extends V>> marshalledEntries)
Persist all provided entries to the store in a single batch update.-
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
Methods inherited from interface org.infinispan.persistence.spi.AdvancedCacheLoader
process
-
-
-
-
Method Detail
-
init
public void init(InitializationContext ctx)
Description copied from interface:CacheLoader
Used to initialize a cache loader. Typically invoked by thePersistenceManager
when setting up cache loaders.- Specified by:
init
in interfaceCacheLoader<K,V>
- Specified by:
init
in interfaceCacheWriter<K,V>
-
start
public void start()
Description copied from interface:Lifecycle
Invoked on component start
-
openDatabase
protected org.rocksdb.RocksDB openDatabase(String location, org.rocksdb.Options options) throws IOException, org.rocksdb.RocksDBException
Creates database if it doesn't exist.- Throws:
IOException
org.rocksdb.RocksDBException
-
stop
public void stop()
Description copied from interface:Lifecycle
Invoked on component stop
-
destroy
public void destroy()
Description copied from interface:ExternalStore
Method to be used to destroy and clean up any resources associated with this store. This is normally only useful for non shared stores.This method will ensure the store is stopped and properly cleans up all resources for it.
- Specified by:
destroy
in interfaceExternalStore<K,V>
-
isAvailable
public boolean isAvailable()
- Specified by:
isAvailable
in interfaceCacheLoader<K,V>
- Specified by:
isAvailable
in interfaceCacheWriter<K,V>
- Specified by:
isAvailable
in interfaceExternalStore<K,V>
- Returns:
- true if the writer can be connected to, otherwise false
-
clear
public void clear()
Description copied from interface:AdvancedCacheWriter
Removes all the data from the storage.- Specified by:
clear
in interfaceAdvancedCacheWriter<K,V>
-
clear
public void clear(IntSet segments)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Removes all the data that maps to the given segments from the storage.This method must only remove entries that map to the provided segments.
This method may be invoked irrespective if the configuration is
StoreConfiguration.segmented()
or not.- Specified by:
clear
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segments
- data mapping to these segments are removed. Always non null.
-
size
public int size()
Description copied from interface:AdvancedCacheLoader
Returns the number of elements in the store.- Specified by:
size
in interfaceAdvancedCacheLoader<K,V>
-
size
public int size(IntSet segments)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Returns the number of elements in the store that map to the given segments that aren't expired.The segments here must be adhered to and the size must not count any entries that don't belong to the provided segments.
This method is not invoked invoked when the store is not configured to be
StoreConfiguration.segmented()
.- Specified by:
size
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segments
- the segments which should have their entries counted. Always non null.- Returns:
- the count of entries in the given segments
-
contains
public boolean contains(Object key)
Description copied from interface:CacheLoader
Returns true if the storage contains an entry associated with the given key.- Specified by:
contains
in interfaceCacheLoader<K,V>
-
contains
public boolean contains(int segment, Object key)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Returns true if the storage contains an entry associated with the given key in the given segmentThe provided segment may be used for performance purposes, however it it is acceptable to ignore this argument.
This method may be invoked invoked irrespective if the store is
StoreConfiguration.segmented()
.- Specified by:
contains
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segment
- the segment that the key maps tokey
- the key to see if exists- Returns:
- true if the key is present in this loader with a given segment
-
publishKeys
public org.reactivestreams.Publisher<K> publishKeys(Predicate<? super K> filter)
Description copied from interface:AdvancedCacheLoader
Publishes all the keys from this store. The given publisher can be used by as manySubscriber
s as desired. Keys are not retrieved until a given Subscriber requests them from theSubscription
.Stores will return only non expired keys
- Specified by:
publishKeys
in interfaceAdvancedCacheLoader<K,V>
- Parameters:
filter
- a filter - null is treated as allowing all entries- Returns:
- a publisher that will provide the keys from the store
-
publishKeys
public org.reactivestreams.Publisher<K> publishKeys(IntSet segments, Predicate<? super K> filter)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Publishes all the keys that map to the given segments from this store. The given publisher can be used by as manySubscriber
s as desired. Keys are not retrieved until a given Subscriber requests them from theSubscription
.Stores will return only non expired keys
The segments here must be adhered to and the keys published must not include any that don't belong to the provided segments.
This method is not invoked invoked when the store is not configured to be
StoreConfiguration.segmented()
.- Specified by:
publishKeys
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segments
- the segments that the keys must map to. Always non null.filter
- a filter- Returns:
- a publisher that will provide the keys from the store
-
publishEntries
public org.reactivestreams.Publisher<MarshalledEntry<K,V>> publishEntries(Predicate<? super K> filter, boolean fetchValue, boolean fetchMetadata)
Description copied from interface:AdvancedCacheLoader
Publishes all entries from this store. The given publisher can be used by as manySubscriber
s as desired. Entries are not retrieved until a given Subscriber requests them from theSubscription
.If fetchMetadata is true this store must guarantee to not return any expired entries.
- Specified by:
publishEntries
in interfaceAdvancedCacheLoader<K,V>
- Parameters:
filter
- a filter - null is treated as allowing all entriesfetchValue
- whether or not to fetch the value from the persistent store. E.g. if the iteration is intended only over the key set, no point fetching the values from the persistent store as wellfetchMetadata
- whether or not to fetch the metadata from the persistent store. E.g. if the iteration is intended only ove the key set, then no point fetching the metadata from the persistent store as well- Returns:
- a publisher that will provide the entries from the store
-
publishEntries
public org.reactivestreams.Publisher<MarshalledEntry<K,V>> publishEntries(IntSet segments, Predicate<? super K> filter, boolean fetchValue, boolean fetchMetadata)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Publishes all entries from this store. The given publisher can be used by as manySubscriber
s as desired. Entries are not retrieved until a given Subscriber requests them from theSubscription
.If fetchMetadata is true this store must guarantee to not return any expired entries.
The segments here must be adhered to and the entries published must not include any that don't belong to the provided segments.
This method is not invoked invoked when the store is not configured to be
StoreConfiguration.segmented()
.StoreConfiguration.segmented()
.- Specified by:
publishEntries
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segments
- the segments that the keys of the entries must map to. Always non null.filter
- a filter on the keys of the entries that if passed will allow the given entry to be returned from the publisherfetchValue
- whether the value should be included in the marshalled entryfetchMetadata
- whether the metadata should be included in the marshalled entry- Returns:
- a publisher that will provide the entries from the store that map to the given segments
-
delete
public boolean delete(Object key)
- Specified by:
delete
in interfaceCacheWriter<K,V>
- Returns:
- true if the entry existed in the persistent store and it was deleted.
-
delete
public boolean delete(int segment, Object key)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Removes the entry for the provided key which is in the given segment. This method then returns whether the entry was removed or not.The provided segment may be used for performance purposes, however it it is acceptable to ignore this argument.
This method may be invoked invoked irrespective if the store is
StoreConfiguration.segmented()
.- Specified by:
delete
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segment
- the segment that this key maps tokey
- the key of the entry to remove- Returns:
- true if the entry existed in the persistent store and it was deleted.
-
write
public void write(MarshalledEntry entry)
Description copied from interface:CacheWriter
Persists the entry to the storage.- Specified by:
write
in interfaceCacheWriter<K,V>
- See Also:
MarshalledEntry
-
write
public void write(int segment, MarshalledEntry<? extends K,? extends V> entry)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Persists the entry to the storage with the given segment to optimize further lookups based onThe provided segment may be used for performance purposes, however it it is acceptable to ignore this argument.
This method may be invoked invoked irrespective if the store is
StoreConfiguration.segmented()
.- Specified by:
write
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segment
- the segment to persist this entry toentry
- the entry to write to the store- See Also:
MarshalledEntry
-
load
public MarshalledEntry load(Object key)
Description copied from interface:CacheLoader
Fetches an entry from the storage. If aMarshalledEntry
needs to be created here,InitializationContext.getMarshalledEntryFactory()
andInitializationContext.getByteBufferFactory()
should be used.- Specified by:
load
in interfaceCacheLoader<K,V>
- Returns:
- the entry, or null if the entry does not exist
-
load
public MarshalledEntry<K,V> load(int segment, Object key)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Fetches an entry from the storage given a segment to optimize this lookup based on. If aMarshalledEntry
needs to be created here,InitializationContext.getMarshalledEntryFactory()
andInitializationContext.getByteBufferFactory()
should be used.The provided segment may be used for performance purposes, however it it is acceptable to ignore this argument.
This method may be invoked invoked irrespective if the store is
StoreConfiguration.segmented()
.- Specified by:
load
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segment
- the segment that the key maps tokey
- the key of the entry to fetch- Returns:
- the entry, or null if the entry does not exist
-
writeBatch
public void writeBatch(Iterable<MarshalledEntry<? extends K,? extends V>> marshalledEntries)
Description copied from interface:CacheWriter
Persist all provided entries to the store in a single batch update. If this is not supported by the underlying store, then entries are written to the store individually viaCacheWriter.write(MarshalledEntry)
.- Specified by:
writeBatch
in interfaceCacheWriter<K,V>
- Parameters:
marshalledEntries
- an Iterable of MarshalledEntry to be written to the store.
-
deleteBatch
public void deleteBatch(Iterable<Object> keys)
Description copied from interface:CacheWriter
Remove all provided keys from the store in a single batch operation. If this is not supported by the underlying store, then keys are removed from the store individually viaCacheWriter.delete(Object)
.- Specified by:
deleteBatch
in interfaceCacheWriter<K,V>
- Parameters:
keys
- an Iterable of entry Keys to be removed from the store.
-
purge
public void purge(Executor executor, AdvancedCacheWriter.PurgeListener purgeListener)
Description copied from interface:AdvancedCacheWriter
Using the thread in the pool, removed all the expired data from the persistence storage. For each removed entry, the supplied listener is invoked.When this method returns all entries will be purged and no tasks will be running due to this loader in the provided executor. If however an exception is thrown there could be tasks still pending or running in the executor.
- Specified by:
purge
in interfaceAdvancedCacheWriter<K,V>
-
addSegments
public void addSegments(IntSet segments)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Invoked when a node becomes an owner of the given segments. Note this method is only invoked for non shared store implementations.This method may be invoked irrespective if the configuration is
StoreConfiguration.segmented()
or not.- Specified by:
addSegments
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segments
- segments to associate with this store
-
removeSegments
public void removeSegments(IntSet segments)
Description copied from interface:SegmentedAdvancedLoadWriteStore
Invoked when a node loses ownership of a segment. The provided segments are the ones this node no longer owns. Note this method is only invoked for non shared store implementations.This method may be invoked irrespective if the configuration is
StoreConfiguration.segmented()
or not.StoreConfiguration.segmented()
.- Specified by:
removeSegments
in interfaceSegmentedAdvancedLoadWriteStore<K,V>
- Parameters:
segments
- segments that should no longer be associated with this store
-
-