1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
|
12 | |
|
13 | |
|
14 | |
|
15 | |
|
16 | |
|
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 | |
|
35 | |
|
36 | |
public class GoraGVertexVertexOutputFormat |
37 | |
extends GoraVertexOutputFormat<LongWritable, DoubleWritable, |
38 | |
FloatWritable> { |
39 | |
|
40 | |
|
41 | |
|
42 | |
|
43 | 0 | public GoraGVertexVertexOutputFormat() { |
44 | 0 | } |
45 | |
|
46 | |
@Override |
47 | |
public VertexWriter<LongWritable, DoubleWritable, FloatWritable> |
48 | |
createVertexWriter(TaskAttemptContext context) |
49 | |
throws IOException, InterruptedException { |
50 | 0 | return new GoraGVertexVertexWriter(); |
51 | |
} |
52 | |
|
53 | |
|
54 | |
|
55 | |
|
56 | 0 | protected class GoraGVertexVertexWriter extends GoraVertexWriter { |
57 | |
|
58 | |
@Override |
59 | |
protected Persistent getGoraVertex( |
60 | |
Vertex<LongWritable, DoubleWritable, FloatWritable> vertex) { |
61 | 0 | GVertexResult tmpGVertex = new GVertexResult(); |
62 | 0 | tmpGVertex.setVertexId(vertex.getId().toString()); |
63 | 0 | tmpGVertex.setVertexValue(Float.parseFloat(vertex.getValue().toString())); |
64 | 0 | Iterator<Edge<LongWritable, FloatWritable>> it = |
65 | 0 | vertex.getEdges().iterator(); |
66 | 0 | while (it.hasNext()) { |
67 | 0 | Edge<LongWritable, FloatWritable> edge = it.next(); |
68 | 0 | tmpGVertex.getEdges().put( |
69 | 0 | edge.getTargetVertexId().toString(), |
70 | 0 | edge.getValue().toString()); |
71 | 0 | } |
72 | 0 | return tmpGVertex; |
73 | |
} |
74 | |
|
75 | |
@Override |
76 | |
protected Object getGoraKey( |
77 | |
Vertex<LongWritable, DoubleWritable, FloatWritable> vertex) { |
78 | 0 | String goraKey = String.valueOf(vertex.getId()); |
79 | 0 | return goraKey; |
80 | |
} |
81 | |
} |
82 | |
} |