org.apache.flink.contrib.streaming.state

Class RocksDBMapState<K,N,UK,UV>

java.lang.Object

org.apache.flink.contrib.streaming.state.AbstractRocksDBState<K,N,org.apache.flink.api.common.state.MapState<UK,UV>,org.apache.flink.api.common.state.MapStateDescriptor<UK,UV>,Map<UK,UV>>

org.apache.flink.contrib.streaming.state.RocksDBMapState<K,N,UK,UV>

Type Parameters:

K - The type of the key.

N - The type of the namespace.

UK - The type of the keys in the map state.

UV - The type of the values in the map state.

All Implemented Interfaces:

org.apache.flink.api.common.state.MapState<UK,UV>, org.apache.flink.api.common.state.State, org.apache.flink.runtime.state.internal.InternalKvState<N>, org.apache.flink.runtime.state.internal.InternalMapState<N,UK,UV>

public class RocksDBMapState<K,N,UK,UV>
extends AbstractRocksDBState<K,N,org.apache.flink.api.common.state.MapState<UK,UV>,org.apache.flink.api.common.state.MapStateDescriptor<UK,UV>,Map<UK,UV>>
implements org.apache.flink.runtime.state.internal.InternalMapState<N,UK,UV>

MapState implementation that stores state in RocksDB.

RocksDBStateBackend must ensure that we set the StringAppendOperator on the column family that we use for our state since we use the merge() call.

Field Summary

Fields inherited from class org.apache.flink.contrib.streaming.state.AbstractRocksDBState

backend, columnFamily, keySerializationDataOutputView, keySerializationStream, stateDesc

Constructor Summary

Constructors

Constructor and Description
RocksDBMapState(org.rocksdb.ColumnFamilyHandle columnFamily, org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer, org.apache.flink.api.common.state.MapStateDescriptor<UK,UV> stateDesc, RocksDBKeyedStateBackend<K> backend) Creates a new RocksDBMapState.

Method Summary

Modifier and Type	Method and Description
void	clear()
boolean	contains(UK userKey)
Iterable<Map.Entry<UK,UV>>	entries()
UV	get(UK userKey)
byte[]	getSerializedValue(byte[] serializedKeyAndNamespace)
Iterator<Map.Entry<UK,UV>>	iterator()
Iterable<UK>	keys()
void	put(UK userKey, UV userValue)
void	putAll(Map<UK,UV> map)
void	remove(UK userKey)
Iterable<UV>	values()

Methods inherited from class org.apache.flink.contrib.streaming.state.AbstractRocksDBState

readKeyWithGroupAndNamespace, setCurrentNamespace, writeCurrentKeyWithGroupAndNamespace, writeKeyWithGroupAndNamespace

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.flink.runtime.state.internal.InternalKvState

setCurrentNamespace

Constructor Detail

RocksDBMapState

public RocksDBMapState(org.rocksdb.ColumnFamilyHandle columnFamily,
                       org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer,
                       org.apache.flink.api.common.state.MapStateDescriptor<UK,UV> stateDesc,
                       RocksDBKeyedStateBackend<K> backend)

Creates a new RocksDBMapState.

Parameters:

namespaceSerializer - The serializer for the namespace.

stateDesc - The state identifier for the state.

Method Detail

get

public UV get(UK userKey)
       throws IOException,
              org.rocksdb.RocksDBException

Specified by:

get in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

put

public void put(UK userKey,
                UV userValue)
         throws IOException,
                org.rocksdb.RocksDBException

Specified by:

put in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

putAll

public void putAll(Map<UK,UV> map)
            throws IOException,
                   org.rocksdb.RocksDBException

Specified by:

putAll in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

remove

public void remove(UK userKey)
            throws IOException,
                   org.rocksdb.RocksDBException

Specified by:

remove in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

contains

public boolean contains(UK userKey)
                 throws IOException,
                        org.rocksdb.RocksDBException

Specified by:

contains in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

entries

public Iterable<Map.Entry<UK,UV>> entries()
                                   throws IOException,
                                          org.rocksdb.RocksDBException

Specified by:

entries in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

keys

public Iterable<UK> keys()
                  throws IOException,
                         org.rocksdb.RocksDBException

Specified by:

keys in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

values

public Iterable<UV> values()
                    throws IOException,
                           org.rocksdb.RocksDBException

Specified by:

values in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

iterator

public Iterator<Map.Entry<UK,UV>> iterator()
                                    throws IOException,
                                           org.rocksdb.RocksDBException

Specified by:

iterator in interface org.apache.flink.api.common.state.MapState<UK,UV>

Throws:

IOException

org.rocksdb.RocksDBException

clear

public void clear()

Specified by:

clear in interface org.apache.flink.api.common.state.State

Overrides:

clear in class AbstractRocksDBState<K,N,org.apache.flink.api.common.state.MapState<UK,UV>,org.apache.flink.api.common.state.MapStateDescriptor<UK,UV>,Map<UK,UV>>

getSerializedValue

public byte[] getSerializedValue(byte[] serializedKeyAndNamespace)
                          throws Exception

Specified by:

getSerializedValue in interface org.apache.flink.runtime.state.internal.InternalKvState<N>

Overrides:

getSerializedValue in class AbstractRocksDBState<K,N,org.apache.flink.api.common.state.MapState<UK,UV>,org.apache.flink.api.common.state.MapStateDescriptor<UK,UV>,Map<UK,UV>>

Throws:

Exception