001/*
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *     http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.apache.hadoop.hbase.codec;
019
020import java.io.IOException;
021import java.io.InputStream;
022import java.io.OutputStream;
023import org.apache.hadoop.hbase.CellBuilderType;
024import org.apache.hadoop.hbase.ExtendedCell;
025import org.apache.hadoop.hbase.ExtendedCellBuilder;
026import org.apache.hadoop.hbase.ExtendedCellBuilderFactory;
027import org.apache.hadoop.hbase.HBaseInterfaceAudience;
028import org.apache.hadoop.hbase.io.ByteBuffInputStream;
029import org.apache.hadoop.hbase.nio.ByteBuff;
030import org.apache.yetus.audience.InterfaceAudience;
031
032import org.apache.hbase.thirdparty.com.google.protobuf.UnsafeByteOperations;
033
034import org.apache.hadoop.hbase.shaded.protobuf.ProtobufUtil;
035import org.apache.hadoop.hbase.shaded.protobuf.generated.CellProtos;
036
037/**
038 * Codec that just writes out Cell as a protobuf Cell Message. Does not write the mvcc stamp. Use a
039 * different codec if you want that in the stream.
040 */
041@InterfaceAudience.LimitedPrivate(HBaseInterfaceAudience.CONFIG)
042public class MessageCodec implements Codec {
043  static class MessageEncoder extends BaseEncoder {
044    MessageEncoder(final OutputStream out) {
045      super(out);
046    }
047
048    @Override
049    public void write(ExtendedCell cell) throws IOException {
050      checkFlushed();
051      CellProtos.Cell.Builder builder = CellProtos.Cell.newBuilder();
052      // This copies bytes from Cell to ByteString. I don't see anyway around the copy.
053      // ByteString is final.
054      builder.setRow(UnsafeByteOperations.unsafeWrap(cell.getRowArray(), cell.getRowOffset(),
055        cell.getRowLength()));
056      builder.setFamily(UnsafeByteOperations.unsafeWrap(cell.getFamilyArray(),
057        cell.getFamilyOffset(), cell.getFamilyLength()));
058      builder.setQualifier(UnsafeByteOperations.unsafeWrap(cell.getQualifierArray(),
059        cell.getQualifierOffset(), cell.getQualifierLength()));
060      builder.setTimestamp(cell.getTimestamp());
061      builder.setCellType(CellProtos.CellType.valueOf(cell.getTypeByte()));
062      builder.setValue(UnsafeByteOperations.unsafeWrap(cell.getValueArray(), cell.getValueOffset(),
063        cell.getValueLength()));
064      CellProtos.Cell pbcell = builder.build();
065      pbcell.writeDelimitedTo(this.out);
066    }
067  }
068
069  static class MessageDecoder extends BaseDecoder {
070    private final ExtendedCellBuilder cellBuilder =
071      ExtendedCellBuilderFactory.create(CellBuilderType.SHALLOW_COPY);
072
073    MessageDecoder(final InputStream in) {
074      super(in);
075    }
076
077    @Override
078    protected ExtendedCell parseCell() throws IOException {
079      return ProtobufUtil.toCell(cellBuilder, CellProtos.Cell.parseDelimitedFrom(this.in), false);
080    }
081  }
082
083  @Override
084  public Decoder getDecoder(InputStream is) {
085    return new MessageDecoder(is);
086  }
087
088  @Override
089  public Decoder getDecoder(ByteBuff buf) {
090    return getDecoder(new ByteBuffInputStream(buf));
091  }
092
093  @Override
094  public Encoder getEncoder(OutputStream os) {
095    return new MessageEncoder(os);
096  }
097}