Java Code Examples for org.apache.giraph.graph.Vertex#initialize()

The following examples show how to use org.apache.giraph.graph.Vertex#initialize() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: PageRankAlgorithm.java    From rheem with Apache License 2.0 6 votes vote down vote up
@Override
public Vertex<LongWritable, DoubleWritable, FloatWritable>
getCurrentVertex() throws IOException {
    Vertex<LongWritable, DoubleWritable, FloatWritable> vertex =
            getConf().createVertex();
    LongWritable vertexId = new LongWritable(
            (inputSplit.getSplitIndex() * totalRecords) + recordsRead);
    DoubleWritable vertexValue = new DoubleWritable(vertexId.get() * 10d);
    long targetVertexId =
            (vertexId.get() + 1) %
                    (inputSplit.getNumSplits() * totalRecords);
    float edgeValue = vertexId.get() * 100f;
    List<Edge<LongWritable, FloatWritable>> edges = Lists.newLinkedList();
    edges.add(EdgeFactory.create(new LongWritable(targetVertexId),
            new FloatWritable(edgeValue)));
    vertex.initialize(vertexId, vertexValue, edges);
    ++recordsRead;
    if (LOG.isInfoEnabled()) {
        LOG.info("next: Return vertexId=" + vertex.getId().get() +
                ", vertexValue=" + vertex.getValue() +
                ", targetVertexId=" + targetVertexId + ", edgeValue=" + edgeValue);
    }
    return vertex;
}
 
Example 2
Source File: ZombieTextVertexInputFormat.java    From hadoop-arch-book with Apache License 2.0 6 votes vote down vote up
@Override
public Vertex<LongWritable, Text, LongWritable> getCurrentVertex()
        throws IOException, InterruptedException {
    Text line = getRecordReader().getCurrentValue();
    String[] majorParts = line.toString().split("\\|");
    LongWritable id = new LongWritable(Long.parseLong(majorParts[0]));
    Text value = new Text(majorParts[1]);

    ArrayList<Edge<LongWritable, LongWritable>> edgeIdList = new ArrayList<Edge<LongWritable, LongWritable>>();

    if (majorParts.length > 2) {
        String[] edgeIds = majorParts[2].split(",");
        for (String edgeId:  edgeIds) {
            DefaultEdge<LongWritable, LongWritable> edge = new DefaultEdge<LongWritable, LongWritable>();
            LongWritable longEdgeId = new LongWritable(Long.parseLong(edgeId));
            edge.setTargetVertexId(longEdgeId);
            edge.setValue(longEdgeId); // dummy value
            edgeIdList.add(edge);
        }
    }

    Vertex<LongWritable, Text, LongWritable> vertex = getConf().createVertex();

    vertex.initialize(id, value, edgeIdList);
    return vertex;
}
 
Example 3
Source File: RyaVertexReader.java    From rya with Apache License 2.0 5 votes vote down vote up
@Override
public Vertex<Text, RyaTypeWritable, RyaStatementWritable> getCurrentVertex() throws IOException, InterruptedException {
    RyaStatementWritable currentStatement = reader.getCurrentValue();
    RyaStatement ryaStatement = currentStatement.getRyaStatement();
    RyaTypeWritable vertexWritable = new RyaTypeWritable();
    vertexWritable.setRyaType(ryaStatement.getSubject());
   Text vertexId = new Text(ryaStatement.getSubject().getData()); 
   Vertex<Text, RyaTypeWritable, RyaStatementWritable> vertex = classesConfiguration.createVertex();
   Edge<Text, RyaStatementWritable> edge = EdgeFactory.create(new Text(ryaStatement.toString()),
           currentStatement);
   List<Edge<Text, RyaStatementWritable>> edges = new ArrayList<Edge<Text, RyaStatementWritable>>();
   edges.add(edge);
    vertex.initialize(vertexId, vertexWritable, edges);
    return vertex;
}
 
Example 4
Source File: ArabesqueInputFormat.java    From Arabesque with Apache License 2.0 5 votes vote down vote up
@Override
public Vertex<IntWritable, NullWritable, NullWritable> getCurrentVertex() throws IOException, InterruptedException {
    Vertex<IntWritable, NullWritable, NullWritable> vertex =
            getConf().createVertex();
    IntWritable id = new IntWritable(inputSplit.getSplitIndex());

    List<Edge<IntWritable, NullWritable>> edges = Collections.emptyList();
    vertex.initialize(id, NullWritable.get(), edges);

    generated = true;
    return vertex;
}