View Javadoc

1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *     http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing, software
13   * distributed under the License is distributed on an "AS IS" BASIS,
14   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15   * See the License for the specific language governing permissions and
16   * limitations under the License.
17   */
18  package org.apache.giraph.io.gora;
19  
20  import java.io.IOException;
21  import java.util.Iterator;
22  
23  import org.apache.giraph.edge.Edge;
24  import org.apache.giraph.graph.Vertex;
25  import org.apache.giraph.io.VertexWriter;
26  import org.apache.giraph.io.gora.generated.GVertexResult;
27  import org.apache.gora.persistency.Persistent;
28  import org.apache.hadoop.io.DoubleWritable;
29  import org.apache.hadoop.io.FloatWritable;
30  import org.apache.hadoop.io.LongWritable;
31  import org.apache.hadoop.mapreduce.TaskAttemptContext;
32  
33  /**
34   * Implementation of a specific reader for a generated data bean.
35   */
36  public class GoraGVertexVertexOutputFormat
37    extends GoraVertexOutputFormat<LongWritable, DoubleWritable,
38    FloatWritable> {
39  
40    /**
41     * DEfault constructor
42     */
43    public GoraGVertexVertexOutputFormat() {
44    }
45  
46    @Override
47    public VertexWriter<LongWritable, DoubleWritable, FloatWritable>
48    createVertexWriter(TaskAttemptContext context)
49      throws IOException, InterruptedException {
50      return new GoraGVertexVertexWriter();
51    }
52  
53    /**
54     * Gora vertex writer.
55     */
56    protected class GoraGVertexVertexWriter extends GoraVertexWriter {
57  
58      @Override
59      protected Persistent getGoraVertex(
60          Vertex<LongWritable, DoubleWritable, FloatWritable> vertex) {
61        GVertexResult tmpGVertex = new GVertexResult();
62        tmpGVertex.setVertexId(vertex.getId().toString());
63        tmpGVertex.setVertexValue(Float.parseFloat(vertex.getValue().toString()));
64        Iterator<Edge<LongWritable, FloatWritable>> it =
65            vertex.getEdges().iterator();
66        while (it.hasNext()) {
67          Edge<LongWritable, FloatWritable> edge = it.next();
68          tmpGVertex.getEdges().put(
69              edge.getTargetVertexId().toString(),
70              edge.getValue().toString());
71        }
72        return tmpGVertex;
73      }
74  
75      @Override
76      protected Object getGoraKey(
77          Vertex<LongWritable, DoubleWritable, FloatWritable> vertex) {
78        String goraKey = String.valueOf(vertex.getId());
79        return goraKey;
80      }
81    }
82  }