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.hadoop.hbase.codec;
19  
20  import java.io.IOException;
21  import java.io.InputStream;
22  import java.io.OutputStream;
23  
24  import org.apache.commons.io.IOUtils;
25  import org.apache.hadoop.hbase.Cell;
26  import org.apache.hadoop.hbase.CellUtil;
27  import org.apache.hadoop.hbase.util.Bytes;
28  
29  /**
30   * Basic Cell codec that just writes out all the individual elements of a Cell.  Uses ints
31   * delimiting all lengths. Profligate. Needs tune up.  Does not write the mvcc stamp.
32   * Use a different codec if you want that in the stream.
33   */
34  public class CellCodec implements Codec {
35    static class CellEncoder extends BaseEncoder {
36      CellEncoder(final OutputStream out) {
37        super(out);
38      }
39  
40      @Override
41      public void write(Cell cell) throws IOException {
42        checkFlushed();
43        try {
44          // Row
45          write(cell.getRowArray(), cell.getRowOffset(), cell.getRowLength());
46          // Column family
47          write(cell.getFamilyArray(), cell.getFamilyOffset(), cell.getFamilyLength());
48          // Qualifier
49          write(cell.getQualifierArray(), cell.getQualifierOffset(), cell.getQualifierLength());
50          // Version
51          this.out.write(Bytes.toBytes(cell.getTimestamp()));
52          // Type
53          this.out.write(cell.getTypeByte());
54          // Value
55          write(cell.getValueArray(), cell.getValueOffset(), cell.getValueLength());
56        } catch (IOException e) {
57          throw new CodecException(e);
58        }
59      }
60  
61      /**
62       * Write int length followed by array bytes.
63       * @param bytes
64       * @param offset
65       * @param length
66       * @throws IOException
67       */
68      private void write(final byte [] bytes, final int offset, final int length)
69      throws IOException {
70        this.out.write(Bytes.toBytes(length));
71        this.out.write(bytes, offset, length);
72      }
73    }
74  
75    static class CellDecoder extends BaseDecoder {
76      public CellDecoder(final InputStream in) {
77        super(in);
78      }
79  
80      protected Cell parseCell() throws IOException {
81        byte [] row = readByteArray(this.in);
82        byte [] family = readByteArray(in);
83        byte [] qualifier = readByteArray(in);
84        byte [] longArray = new byte[Bytes.SIZEOF_LONG];
85        IOUtils.readFully(this.in, longArray);
86        long timestamp = Bytes.toLong(longArray);
87        byte type = (byte) this.in.read();
88        byte [] value = readByteArray(in);
89        return CellUtil.createCell(row, family, qualifier, timestamp, type, value);
90      }
91  
92      /**
93       * @return Byte array read from the stream.
94       * @throws IOException
95       */
96      private byte [] readByteArray(final InputStream in) throws IOException {
97        byte [] intArray = new byte[Bytes.SIZEOF_INT];
98        IOUtils.readFully(in, intArray);
99        int length = Bytes.toInt(intArray);
100       byte [] bytes = new byte [length];
101       IOUtils.readFully(in, bytes);
102       return bytes;
103     }
104   }
105 
106   @Override
107   public Decoder getDecoder(InputStream is) {
108     return new CellDecoder(is);
109   }
110 
111   @Override
112   public Encoder getEncoder(OutputStream os) {
113     return new CellEncoder(os);
114   }
115 }