ai.grakn.engine.tasks.manager.redisqueue

Class RedisTaskManager

java.lang.Object

ai.grakn.engine.tasks.manager.redisqueue.RedisTaskManager

All Implemented Interfaces:

TaskManager, Closeable, AutoCloseable

public class RedisTaskManager
extends Object
implements TaskManager

Handle the lifecycle of tasks in Redis. Given a jedis pool it starts a set of consumers that subscribe to the task queue. Tasks can be added using the addTask method.

Author:

pluraliseseverythings

Constructor Summary

Constructors

Constructor and Description
RedisTaskManager(EngineID engineId, GraknConfig config, redis.clients.util.Pool<redis.clients.jedis.Jedis> jedisPool, int threads, EngineGraknTxFactory factory, com.codahale.metrics.MetricRegistry metricRegistry, PostProcessor postProcessor)

Method Summary

Modifier and Type	Method and Description
void	addTask(TaskState taskState, TaskConfiguration configuration) Schedule a BackgroundTask for execution.
void	close()
ai.grakn.redisq.Redisq	getQueue()
void	runTask(TaskState taskState, TaskConfiguration configuration)
CompletableFuture<Void>	start() Make sure the manager is initialized and starts processing tasks
RedisTaskStorage	storage() Return the StateStorage instance that is used by this class.
Future<Void>	subscribeToTask(TaskId taskId)
void	waitForTask(TaskId taskId, long timeout, TimeUnit timeUnit)

Methods inherited from class java.lang.Object

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

Constructor Detail

RedisTaskManager

public RedisTaskManager(EngineID engineId,
                        GraknConfig config,
                        redis.clients.util.Pool<redis.clients.jedis.Jedis> jedisPool,
                        int threads,
                        EngineGraknTxFactory factory,
                        com.codahale.metrics.MetricRegistry metricRegistry,
                        PostProcessor postProcessor)

Method Detail

close

public void close()

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

start

public CompletableFuture<Void> start()

Description copied from interface: TaskManager

Make sure the manager is initialized and starts processing tasks

Specified by:

start in interface TaskManager

storage

public RedisTaskStorage storage()

Description copied from interface: TaskManager

Return the StateStorage instance that is used by this class.

Specified by:

storage in interface TaskManager

Returns:

A StateStorage instance.

addTask

public void addTask(TaskState taskState,
                    TaskConfiguration configuration)

Description copied from interface: TaskManager

Schedule a BackgroundTask for execution.

Specified by:

addTask in interface TaskManager

Parameters:

taskState - Task to execute

runTask

public void runTask(TaskState taskState,
                    TaskConfiguration configuration)

Specified by:

runTask in interface TaskManager

subscribeToTask

public Future<Void> subscribeToTask(TaskId taskId)
                             throws ai.grakn.redisq.exceptions.StateFutureInitializationException,
                                    ExecutionException,
                                    InterruptedException

Throws:

ai.grakn.redisq.exceptions.StateFutureInitializationException

ExecutionException

InterruptedException

waitForTask

public void waitForTask(TaskId taskId,
                        long timeout,
                        TimeUnit timeUnit)
                 throws ai.grakn.redisq.exceptions.StateFutureInitializationException,
                        ExecutionException,
                        InterruptedException,
                        TimeoutException

Throws:

ai.grakn.redisq.exceptions.StateFutureInitializationException

ExecutionException

InterruptedException

TimeoutException

getQueue

public ai.grakn.redisq.Redisq getQueue()