1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.giraph.comm.requests;
20
21 import org.apache.giraph.utils.UnsafeByteArrayInputStream;
22
23 import java.io.ByteArrayInputStream;
24 import java.io.DataInput;
25 import java.io.DataInputStream;
26 import java.io.DataOutput;
27 import java.io.IOException;
28
29
30
31
32 public abstract class ByteArrayRequest extends WritableRequest {
33
34 private byte[] data;
35
36
37
38
39
40
41 ByteArrayRequest(byte[] data) {
42 this.data = data;
43 }
44
45
46
47
48 ByteArrayRequest() {
49 }
50
51
52
53
54
55
56 public byte[] getData() {
57 return data;
58 }
59
60
61
62
63
64
65 public DataInput getDataInput() {
66 return new DataInputStream(new ByteArrayInputStream(data));
67 }
68
69
70
71
72
73 public UnsafeByteArrayInputStream getUnsafeByteArrayInput() {
74 return new UnsafeByteArrayInputStream(data);
75 }
76
77 @Override
78 void readFieldsRequest(DataInput input) throws IOException {
79 int dataLength = input.readInt();
80 data = new byte[dataLength];
81 input.readFully(data);
82 }
83
84 @Override
85 void writeRequest(DataOutput output) throws IOException {
86 output.writeInt(data.length);
87 output.write(data);
88 }
89
90 @Override
91 public int getSerializedSize() {
92
93 return super.getSerializedSize() + 4 + data.length;
94 }
95 }