ai.grakn.kb.internal.computer

Class GraknSparkMemory

java.lang.Object

ai.grakn.kb.internal.computer.GraknSparkMemory

All Implemented Interfaces:

Serializable, org.apache.tinkerpop.gremlin.process.computer.Memory, org.apache.tinkerpop.gremlin.process.computer.Memory.Admin

public class GraknSparkMemory
extends Object
implements org.apache.tinkerpop.gremlin.process.computer.Memory.Admin, Serializable

This is a modified version of Spark Memory. We change its behaviour so it can work with our own graph computer.

Author:

Jason Liu, Marko A. Rodriguez

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.tinkerpop.gremlin.process.computer.Memory

org.apache.tinkerpop.gremlin.process.computer.Memory.Admin, org.apache.tinkerpop.gremlin.process.computer.Memory.Exceptions

Field Summary

Fields

Modifier and Type	Field and Description
Map<String,org.apache.tinkerpop.gremlin.process.computer.MemoryComputeKey>	memoryComputeKeys

Constructor Summary

Constructors

Constructor and Description
GraknSparkMemory(org.apache.tinkerpop.gremlin.process.computer.VertexProgram<?> vertexProgram, Set<org.apache.tinkerpop.gremlin.process.computer.MapReduce> mapReducers, org.apache.spark.api.java.JavaSparkContext sparkContext)

Method Summary

Modifier and Type	Method and Description
void	add(String key, Object value)
protected void	broadcastMemory(org.apache.spark.api.java.JavaSparkContext sparkContext)
protected void	complete()
<R> R	get(String key)
int	getIteration()
long	getRuntime()
void	incrIteration()
Set<String>	keys()
void	set(String key, Object value)
void	setInExecute(boolean inExecute)
void	setIteration(int iteration)
void	setRuntime(long runTime)
String	toString()

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.tinkerpop.gremlin.process.computer.Memory.Admin

asImmutable

Methods inherited from interface org.apache.tinkerpop.gremlin.process.computer.Memory

asMap, exists, isInitialIteration

Field Detail

memoryComputeKeys

public final Map<String,org.apache.tinkerpop.gremlin.process.computer.MemoryComputeKey> memoryComputeKeys

Constructor Detail

GraknSparkMemory

public GraknSparkMemory(org.apache.tinkerpop.gremlin.process.computer.VertexProgram<?> vertexProgram,
                        Set<org.apache.tinkerpop.gremlin.process.computer.MapReduce> mapReducers,
                        org.apache.spark.api.java.JavaSparkContext sparkContext)

Method Detail

keys

public Set<String> keys()

Specified by:

keys in interface org.apache.tinkerpop.gremlin.process.computer.Memory

incrIteration

public void incrIteration()

Specified by:

incrIteration in interface org.apache.tinkerpop.gremlin.process.computer.Memory.Admin

setIteration

public void setIteration(int iteration)

Specified by:

setIteration in interface org.apache.tinkerpop.gremlin.process.computer.Memory.Admin

getIteration

public int getIteration()

Specified by:

getIteration in interface org.apache.tinkerpop.gremlin.process.computer.Memory

setRuntime

public void setRuntime(long runTime)

Specified by:

setRuntime in interface org.apache.tinkerpop.gremlin.process.computer.Memory.Admin

getRuntime

public long getRuntime()

Specified by:

getRuntime in interface org.apache.tinkerpop.gremlin.process.computer.Memory

get

public <R> R get(String key)
          throws IllegalArgumentException

Specified by:

get in interface org.apache.tinkerpop.gremlin.process.computer.Memory

Throws:

IllegalArgumentException

add

public void add(String key,
                Object value)

Specified by:

add in interface org.apache.tinkerpop.gremlin.process.computer.Memory

set

public void set(String key,
                Object value)

Specified by:

set in interface org.apache.tinkerpop.gremlin.process.computer.Memory

toString

public String toString()

Overrides:

toString in class Object

complete

protected void complete()

setInExecute

public void setInExecute(boolean inExecute)

broadcastMemory

protected void broadcastMemory(org.apache.spark.api.java.JavaSparkContext sparkContext)