ai.grakn.engine.task.postprocessing.redisstorage

Class RedisCountStorage

java.lang.Object

ai.grakn.engine.task.postprocessing.redisstorage.RedisCountStorage

All Implemented Interfaces:

CountStorage

public class RedisCountStorage
extends Object
implements CountStorage

Stores a cache of counts so that we can know which Types to shard when they have too many instances.

Author:

fppt

Method Summary

Modifier and Type	Method and Description
static RedisCountStorage	create(redis.clients.util.Pool<redis.clients.jedis.Jedis> jedisPool, com.codahale.metrics.MetricRegistry metricRegistry)
long	getCount(String key) Gets the count for the specified key.
long	getInstanceCount(ai.grakn.Keyspace keyspace, ai.grakn.concept.ConceptId conceptId) Get the instance count for a specific concept.
static String	getKeyNumInstances(ai.grakn.Keyspace keyspace, ai.grakn.concept.ConceptId conceptId) All the valid keys which map to values in the redis cache
static String	getKeyNumShards(ai.grakn.Keyspace keyspace, ai.grakn.concept.ConceptId conceptId)
long	getShardCount(ai.grakn.Keyspace keyspace, ai.grakn.concept.ConceptId conceptId) Get the shard count for a specific concept.
long	incrementCount(String key, long incrementBy) Adjusts the count for a specific key.
long	incrementInstanceCount(ai.grakn.Keyspace keyspace, ai.grakn.concept.ConceptId conceptId, long incrementBy) Adjusts the count for a specific concept.
long	incrementShardCount(ai.grakn.Keyspace keyspace, ai.grakn.concept.ConceptId conceptId, long incrementBy) Adjusts the shard count for a specific concept.

Methods inherited from class java.lang.Object

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

Method Detail

create

public static RedisCountStorage create(redis.clients.util.Pool<redis.clients.jedis.Jedis> jedisPool,
                                       com.codahale.metrics.MetricRegistry metricRegistry)

incrementInstanceCount

public long incrementInstanceCount(ai.grakn.Keyspace keyspace,
                                   ai.grakn.concept.ConceptId conceptId,
                                   long incrementBy)

Description copied from interface: CountStorage

Adjusts the count for a specific concept.

Specified by:

incrementInstanceCount in interface CountStorage

incrementBy - the number to adjust the key by

Returns:

the final value after being adjusted

incrementShardCount

public long incrementShardCount(ai.grakn.Keyspace keyspace,
                                ai.grakn.concept.ConceptId conceptId,
                                long incrementBy)

Description copied from interface: CountStorage

Adjusts the shard count for a specific concept.

Specified by:

incrementShardCount in interface CountStorage

incrementBy - the number to adjust the key by

Returns:

the final value after being adjusted

getInstanceCount

public long getInstanceCount(ai.grakn.Keyspace keyspace,
                             ai.grakn.concept.ConceptId conceptId)

Description copied from interface: CountStorage

Get the instance count for a specific concept.

Specified by:

getInstanceCount in interface CountStorage

Returns:

the instance count

getShardCount

public long getShardCount(ai.grakn.Keyspace keyspace,
                          ai.grakn.concept.ConceptId conceptId)

Description copied from interface: CountStorage

Get the shard count for a specific concept.

Specified by:

getShardCount in interface CountStorage

Returns:

the shard count

incrementCount

public long incrementCount(String key,
                           long incrementBy)

Adjusts the count for a specific key.

Parameters:

key - the key of the value to adjust

incrementBy - the number to adjust the key by

Returns:

true

getCount

public long getCount(String key)

Gets the count for the specified key. A count of 0 is returned if the key is not in redis

Parameters:

key - the key stored in redis

Returns:

the current count.

getKeyNumInstances

public static String getKeyNumInstances(ai.grakn.Keyspace keyspace,
                                        ai.grakn.concept.ConceptId conceptId)

All the valid keys which map to values in the redis cache

getKeyNumShards

public static String getKeyNumShards(ai.grakn.Keyspace keyspace,
                                     ai.grakn.concept.ConceptId conceptId)