tajo-storage/tajo-storage-hbase/src/main/java/org/apache/tajo/storage/hbase/HBaseTextSerializerDeserializer.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.tajo.storage.hbase;

import org.apache.tajo.catalog.Column;
import org.apache.tajo.datum.Datum;
import org.apache.tajo.datum.DatumFactory;
import org.apache.tajo.datum.NullDatum;
import org.apache.tajo.storage.Tuple;
import org.apache.tajo.util.NumberUtil;

import java.io.IOException;

public class HBaseTextSerializerDeserializer {
  public static Datum deserialize(Column col, byte[] bytes) throws IOException {
    Datum datum;
    switch (col.getDataType().getType()) {
      case INT1:
      case INT2:
        datum = bytes == null  || bytes.length == 0 ? NullDatum.get() :
            DatumFactory.createInt2((short)NumberUtil.parseInt(bytes, 0, bytes.length));
        break;
      case INT4:
        datum = bytes == null || bytes.length == 0 ? NullDatum.get() :
            DatumFactory.createInt4(NumberUtil.parseInt(bytes, 0, bytes.length));
        break;
      case INT8:
        datum = bytes == null || bytes.length == 0 ? NullDatum.get() :
            DatumFactory.createInt8(new String(bytes, 0, bytes.length));
        break;
      case FLOAT4:
        datum = bytes == null || bytes.length == 0 ? NullDatum.get() :
            DatumFactory.createFloat4(new String(bytes, 0, bytes.length));
        break;
      case FLOAT8:
        datum = bytes == null || bytes.length == 0 ? NullDatum.get() :
            DatumFactory.createFloat8(NumberUtil.parseDouble(bytes, 0, bytes.length));
        break;
      case TEXT:
        datum = bytes == null ? NullDatum.get() : DatumFactory.createText(bytes);
        break;
      default:
        datum = NullDatum.get();
        break;
    }
    return datum;
  }

  public static byte[] serialize(Column col, Datum datum) {
    if (datum == null || datum instanceof NullDatum) {
      return null;
    }

    return datum.asChars().getBytes();
  }

  public static byte[] serialize(Column col, Tuple tuple, int index) throws IOException {
    if (tuple.isBlankOrNull(index)) {
      return null;
    }

    return tuple.getBytes(index);
  }
}