1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.hadoop.hbase.rest.model;
21
22 import java.io.IOException;
23 import java.io.Serializable;
24 import java.util.ArrayList;
25 import java.util.List;
26
27 import javax.xml.bind.annotation.XmlAccessType;
28 import javax.xml.bind.annotation.XmlAccessorType;
29 import javax.xml.bind.annotation.XmlElement;
30 import javax.xml.bind.annotation.XmlRootElement;
31
32 import org.apache.hadoop.hbase.util.ByteStringer;
33 import org.apache.hadoop.hbase.classification.InterfaceAudience;
34 import org.apache.hadoop.hbase.HConstants;
35 import org.apache.hadoop.hbase.protobuf.ProtobufUtil;
36 import org.apache.hadoop.hbase.rest.ProtobufMessageHandler;
37 import org.apache.hadoop.hbase.rest.protobuf.generated.CellMessage.Cell;
38 import org.apache.hadoop.hbase.rest.protobuf.generated.CellSetMessage.CellSet;
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73 @XmlRootElement(name="CellSet")
74 @XmlAccessorType(XmlAccessType.FIELD)
75 @InterfaceAudience.Private
76 public class CellSetModel implements Serializable, ProtobufMessageHandler {
77
78 private static final long serialVersionUID = 1L;
79
80 @XmlElement(name="Row")
81 private List<RowModel> rows;
82
83
84
85
86 public CellSetModel() {
87 this.rows = new ArrayList<RowModel>();
88 }
89
90
91
92
93 public CellSetModel(List<RowModel> rows) {
94 super();
95 this.rows = rows;
96 }
97
98
99
100
101
102 public void addRow(RowModel row) {
103 rows.add(row);
104 }
105
106
107
108
109 public List<RowModel> getRows() {
110 return rows;
111 }
112
113 @Override
114 public byte[] createProtobufOutput() {
115 CellSet.Builder builder = CellSet.newBuilder();
116 for (RowModel row: getRows()) {
117 CellSet.Row.Builder rowBuilder = CellSet.Row.newBuilder();
118 rowBuilder.setKey(ByteStringer.wrap(row.getKey()));
119 for (CellModel cell: row.getCells()) {
120 Cell.Builder cellBuilder = Cell.newBuilder();
121 cellBuilder.setColumn(ByteStringer.wrap(cell.getColumn()));
122 cellBuilder.setData(ByteStringer.wrap(cell.getValue()));
123 if (cell.hasUserTimestamp()) {
124 cellBuilder.setTimestamp(cell.getTimestamp());
125 }
126 rowBuilder.addValues(cellBuilder);
127 }
128 builder.addRows(rowBuilder);
129 }
130 return builder.build().toByteArray();
131 }
132
133 @Override
134 public ProtobufMessageHandler getObjectFromMessage(byte[] message)
135 throws IOException {
136 CellSet.Builder builder = CellSet.newBuilder();
137 ProtobufUtil.mergeFrom(builder, message);
138 for (CellSet.Row row: builder.getRowsList()) {
139 RowModel rowModel = new RowModel(row.getKey().toByteArray());
140 for (Cell cell: row.getValuesList()) {
141 long timestamp = HConstants.LATEST_TIMESTAMP;
142 if (cell.hasTimestamp()) {
143 timestamp = cell.getTimestamp();
144 }
145 rowModel.addCell(
146 new CellModel(cell.getColumn().toByteArray(), timestamp,
147 cell.getData().toByteArray()));
148 }
149 addRow(rowModel);
150 }
151 return this;
152 }
153 }