org.apache.giraph.utils

Class TestGraph<I extends org.apache.hadoop.io.WritableComparable,V extends org.apache.hadoop.io.Writable,E extends org.apache.hadoop.io.Writable>

java.lang.Object

org.apache.giraph.utils.TestGraph<I,V,E>

Type Parameters:

I - Vertex index type

V - Vertex type

E - Edge type

All Implemented Interfaces:

Iterable<Vertex<I,V,E>>

public class TestGraph<I extends org.apache.hadoop.io.WritableComparable,V extends org.apache.hadoop.io.Writable,E extends org.apache.hadoop.io.Writable>
extends Object
implements Iterable<Vertex<I,V,E>>

TestGraph class for in-memory testing.

Field Summary

Fields

Modifier and Type	Field and Description
protected ImmutableClassesGiraphConfiguration<I,V,E>	conf The configuration
protected VertexValueCombiner<V>	vertexValueCombiner Vertex value combiner
protected Basic2ObjectMap<I,Vertex<I,V,E>>	vertices The vertex values

Constructor Summary

Constructors

Constructor and Description
TestGraph(GiraphConfiguration conf) Constructor requiring classes

Method Summary

Modifier and Type	Method and Description
TestGraph<I,V,E>	addEdge(I vertexId, I toVertex, E edgeValue) Add an edge to an existing vertex
TestGraph<I,V,E>	addEdge(I vertexId, Map.Entry<I,E> edgePair) Add an edge to an existing vertex `
TestGraph<I,V,E>	addVertex(I id, V value, Map.Entry<I,E>... edges) Add vertex with given ID
TestGraph<I,V,E>	addVertex(Vertex<I,V,E> vertex) Add vertex
void	clear() Clear all data
protected Iterable<Edge<I,E>>	createEdges(Map.Entry<I,E>... destEdgess) Create edges for given ids
ImmutableClassesGiraphConfiguration<I,V,E>	getConf()
Vertex<I,V,E>	getVertex(I id) Return a given vertex
int	getVertexCount()
Collection<Vertex<I,V,E>>	getVertices()
Iterator<Vertex<I,V,E>>	iterator() An iterator over the vertices
protected Vertex<I,V,E>	makeVertex(I id, V value, Map.Entry<I,E>... edges) Create a vertex
TestGraph<I,V,E>	setVertex(I id, V value, Map.Entry<I,E>... edges) Set vertex, replace if there was already a vertex with same id added
TestGraph<I,V,E>	setVertex(Vertex<I,V,E> vertex) Set vertex, replace if there was already a vertex with same id added
String	toString()

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Field Detail

vertexValueCombiner

protected final VertexValueCombiner<V extends org.apache.hadoop.io.Writable> vertexValueCombiner

Vertex value combiner

vertices

protected Basic2ObjectMap<I extends org.apache.hadoop.io.WritableComparable,Vertex<I extends org.apache.hadoop.io.WritableComparable,V extends org.apache.hadoop.io.Writable,E extends org.apache.hadoop.io.Writable>> vertices

The vertex values

conf

protected ImmutableClassesGiraphConfiguration<I extends org.apache.hadoop.io.WritableComparable,V extends org.apache.hadoop.io.Writable,E extends org.apache.hadoop.io.Writable> conf

The configuration

Constructor Detail

TestGraph

public TestGraph(GiraphConfiguration conf)

Constructor requiring classes

Parameters:

conf - Should have vertex and edge classes set.

Method Detail

getVertices

public Collection<Vertex<I,V,E>> getVertices()

getVertexCount

public int getVertexCount()

getConf

public ImmutableClassesGiraphConfiguration<I,V,E> getConf()

clear

public void clear()

Clear all data

addVertex

public TestGraph<I,V,E> addVertex(Vertex<I,V,E> vertex)

Add vertex

Parameters:

vertex - Vertex

Returns:

this

addVertex

public TestGraph<I,V,E> addVertex(I id,
                                  V value,
                                  Map.Entry<I,E>... edges)

Add vertex with given ID

Parameters:

id - the index

value - the value

edges - all edges

Returns:

this

setVertex

public TestGraph<I,V,E> setVertex(Vertex<I,V,E> vertex)

Set vertex, replace if there was already a vertex with same id added

Parameters:

vertex - Vertex

Returns:

this

setVertex

public TestGraph<I,V,E> setVertex(I id,
                                  V value,
                                  Map.Entry<I,E>... edges)

Set vertex, replace if there was already a vertex with same id added

Parameters:

id - the index

value - the value

edges - all edges

Returns:

this

addEdge

public TestGraph<I,V,E> addEdge(I vertexId,
                                Map.Entry<I,E> edgePair)

Add an edge to an existing vertex `

Parameters:

vertexId - Edge origin

edgePair - The edge

Returns:

this

addEdge

public TestGraph<I,V,E> addEdge(I vertexId,
                                I toVertex,
                                E edgeValue)

Add an edge to an existing vertex

Parameters:

vertexId - Edge origin

toVertex - Edge destination

edgeValue - Edge value

Returns:

this

iterator

public Iterator<Vertex<I,V,E>> iterator()

An iterator over the vertices

Specified by:

iterator in interface Iterable<Vertex<I extends org.apache.hadoop.io.WritableComparable,V extends org.apache.hadoop.io.Writable,E extends org.apache.hadoop.io.Writable>>

Returns:

the iterator

getVertex

public Vertex<I,V,E> getVertex(I id)

Return a given vertex

Parameters:

id - the id

Returns:

the value

createEdges

protected Iterable<Edge<I,E>> createEdges(Map.Entry<I,E>... destEdgess)

Create edges for given ids

Parameters:

destEdgess - ids to which the edges link

Returns:

an iterable containing the edges

makeVertex

protected Vertex<I,V,E> makeVertex(I id,
                                   V value,
                                   Map.Entry<I,E>... edges)

Create a vertex

Parameters:

id - the id of the vertex

value - the vertex value

edges - edges to other vertices

Returns:

a new vertex

toString

public String toString()

Overrides:

toString in class Object