ai.grakn.engine.tasks.manager.redisqueue

Class RedisTaskManager

java.lang.Object

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

All Implemented Interfaces:

TaskManager, TaskSubmitter, 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, GraknEngineConfig config, redis.clients.util.Pool<redis.clients.jedis.Jedis> jedisPool, EngineGraknTxFactory factory, LockProvider distributedLockClient, com.codahale.metrics.MetricRegistry metricRegistry)
RedisTaskManager(EngineID engineId, GraknEngineConfig config, redis.clients.util.Pool<redis.clients.jedis.Jedis> jedisPool, int threads, EngineGraknTxFactory factory, LockProvider distributedLockClient, com.codahale.metrics.MetricRegistry metricRegistry)

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
void	stopTask(ai.grakn.engine.TaskId id) Stop a Scheduled, Paused or Running task.
RedisTaskStorage	storage() Return the StateStorage instance that is used by this class.
Future<Void>	subscribeToTask(ai.grakn.engine.TaskId taskId)
void	waitForTask(ai.grakn.engine.TaskId taskId, long timeout, TimeUnit timeUnit)

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.grakn.engine.tasks.manager.TaskSubmitter

getNoopTaskSubmitter

Constructor Detail

RedisTaskManager

public RedisTaskManager(EngineID engineId,
                        GraknEngineConfig config,
                        redis.clients.util.Pool<redis.clients.jedis.Jedis> jedisPool,
                        EngineGraknTxFactory factory,
                        LockProvider distributedLockClient,
                        com.codahale.metrics.MetricRegistry metricRegistry)

RedisTaskManager

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

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

stopTask

public void stopTask(ai.grakn.engine.TaskId id)

Description copied from interface: TaskManager

Stop a Scheduled, Paused or Running task. Task's .stop() method will be called to perform any cleanup and the task is killed afterwards.

Specified by:

stopTask in interface TaskManager

Parameters:

id - ID of task to stop.

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: TaskSubmitter

Schedule a BackgroundTask for execution.

Specified by:

addTask in interface TaskSubmitter

Parameters:

taskState - Task to execute

runTask

public void runTask(TaskState taskState,
                    TaskConfiguration configuration)

Specified by:

runTask in interface TaskSubmitter

subscribeToTask

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

Throws:

ai.grakn.redisq.exceptions.StateFutureInitializationException

ExecutionException

InterruptedException

waitForTask

public void waitForTask(ai.grakn.engine.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()