ai.grakn.engine.loader

Class LoaderImpl

java.lang.Object

ai.grakn.engine.loader.LoaderImpl

All Implemented Interfaces:

Loader

public class LoaderImpl
extends Object
implements Loader

Manage loading tasks in the Task Manager

Constructor Summary

Constructors

Constructor and Description
LoaderImpl(String keyspace)

Method Summary

Modifier and Type	Method and Description
void	add(InsertQuery query) Add an insert query to the queue
void	flush() Load any remaining batches in the queue.
int	getBatchSize()
void	printLoaderState() Method that logs the current state of loading tasks
void	sendQueriesToLoader(Collection<InsertQuery> batch) Method to load data into the graph.
Loader	setBatchSize(int size) Set the size of the each transaction in terms of number of vars.
Loader	setQueueSize(int size) Set the size of the queue- this is equivalent to the size of the semaphore.
void	waitToFinish() Wait for all tasks to finish for one minute.
void	waitToFinish(int timeout) Wait for all tasks to finish.

Methods inherited from class java.lang.Object

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

Constructor Detail

LoaderImpl

public LoaderImpl(String keyspace)

Method Detail

getBatchSize

public int getBatchSize()

Specified by:

getBatchSize in interface Loader

Returns:

the current batch size - minimum number of vars to be loaded in a transaction

setBatchSize

public Loader setBatchSize(int size)

Set the size of the each transaction in terms of number of vars.

Specified by:

setBatchSize in interface Loader

Parameters:

size - number of vars in each transaction

setQueueSize

public Loader setQueueSize(int size)

Set the size of the queue- this is equivalent to the size of the semaphore.

Specified by:

setQueueSize in interface Loader

Parameters:

size - the size of the queue

flush

public void flush()

Load any remaining batches in the queue.

Specified by:

flush in interface Loader

add

public void add(InsertQuery query)

Add an insert query to the queue

Specified by:

add in interface Loader

Parameters:

query - insert query to be executed

sendQueriesToLoader

public void sendQueriesToLoader(Collection<InsertQuery> batch)

Method to load data into the graph. Implementation depends on the type of the loader.

waitToFinish

public void waitToFinish()

Wait for all tasks to finish for one minute.

Specified by:

waitToFinish in interface Loader

waitToFinish

public void waitToFinish(int timeout)

Wait for all tasks to finish.

Specified by:

waitToFinish in interface Loader

Parameters:

timeout - amount of time (in ms) to wait.

printLoaderState

public void printLoaderState()

Method that logs the current state of loading tasks

Specified by:

printLoaderState in interface Loader