1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.giraph.utils.io;
20
21 import org.apache.giraph.conf.ImmutableClassesGiraphConfiguration;
22 import org.apache.giraph.utils.ExtendedDataInput;
23 import org.apache.giraph.utils.ExtendedDataOutput;
24 import org.apache.giraph.utils.WritableUtils;
25
26 import java.io.DataInput;
27 import java.io.DataOutput;
28 import java.io.IOException;
29
30
31
32
33
34 public class ExtendedDataInputOutput extends DataInputOutput {
35
36 private final ImmutableClassesGiraphConfiguration conf;
37
38 private ExtendedDataOutput dataOutput;
39
40
41
42
43
44
45 public ExtendedDataInputOutput(ImmutableClassesGiraphConfiguration conf) {
46 this.conf = conf;
47 dataOutput = conf.createExtendedDataOutput();
48 }
49
50 @Override
51 public DataOutput getDataOutput() {
52 return dataOutput;
53 }
54
55 @Override
56 public ExtendedDataInput createDataInput() {
57 return conf.createExtendedDataInput(dataOutput.getByteArray(), 0,
58 dataOutput.getPos());
59 }
60
61 @Override
62 public void write(DataOutput out) throws IOException {
63 WritableUtils.writeExtendedDataOutput(dataOutput, out);
64 }
65
66 @Override
67 public void readFields(DataInput in) throws IOException {
68 dataOutput = WritableUtils.readExtendedDataOutput(in, conf);
69 }
70 }