org.apache.hadoop.hive.accumulo.AccumuloHiveRow Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hive-accumulo-handler Show documentation
The newest version!
/*
 * 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.hadoop.hive.accumulo;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import org.apache.commons.lang.builder.HashCodeBuilder;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.io.Writable;

import com.google.common.base.Preconditions;

/**
 * Holds column tuples for rowID. Each tuple contains column family label, qualifier label, and byte
 * array value.
 */
public class AccumuloHiveRow implements Writable {

  private String rowId;
  private List tuples = new ArrayList();

  public AccumuloHiveRow() {}

  public AccumuloHiveRow(String rowId) {
    this.rowId = rowId;
  }

  public void setRowId(String rowId) {
    this.rowId = rowId;
  }

  public List getTuples() {
    return Collections.unmodifiableList(tuples);
  }

  /**
   * @return true if this instance has a tuple containing fam and qual, false otherwise.
   */
  public boolean hasFamAndQual(Text fam, Text qual) {
    for (ColumnTuple tuple : tuples) {
      if (tuple.getCf().equals(fam) && tuple.getCq().equals(qual)) {
        return true;
      }
    }
    return false;
  }

  /**
   * @return byte [] value for first tuple containing fam and qual or null if no match.
   */
  public byte[] getValue(Text fam, Text qual) {
    for (ColumnTuple tuple : tuples) {
      if (tuple.getCf().equals(fam) && tuple.getCq().equals(qual)) {
        return tuple.getValue();
      }
    }
    return null;
  }

  public String getRowId() {
    return rowId;
  }

  public void clear() {
    this.rowId = null;
    this.tuples = new ArrayList();
  }

  @Override
  public String toString() {
    StringBuilder builder = new StringBuilder("AccumuloHiveRow{");
    builder.append("rowId='").append(rowId).append("', tuples: ");
    for (ColumnTuple tuple : tuples) {
      builder.append(tuple.toString());
      builder.append("\n");
    }
    return builder.toString();
  }

  @Override
  public boolean equals(Object o) {
    if (o instanceof AccumuloHiveRow) {
      AccumuloHiveRow other = (AccumuloHiveRow) o;
      if (null == rowId) {
        if (null != other.rowId) {
          return false;
        }
      } else if (!rowId.equals(other.rowId)) {
        return false;
      }

      return tuples.equals(other.tuples);
    }

    return false;
  }

  @Override
  public void write(DataOutput dataOutput) throws IOException {
    if (null != rowId) {
      dataOutput.writeBoolean(true);
      dataOutput.writeUTF(rowId);
    } else {
      dataOutput.writeBoolean(false);
    }
    int size = tuples.size();
    dataOutput.writeInt(size);
    for (ColumnTuple tuple : tuples) {
      Text cf = tuple.getCf(), cq = tuple.getCq();
      dataOutput.writeInt(cf.getLength());
      dataOutput.write(cf.getBytes(), 0, cf.getLength());
      dataOutput.writeInt(cq.getLength());
      dataOutput.write(cq.getBytes(), 0, cq.getLength());
      byte[] value = tuple.getValue();
      dataOutput.writeInt(value.length);
      dataOutput.write(value);
    }
  }

  @Override
  public void readFields(DataInput dataInput) throws IOException {
    if (dataInput.readBoolean()) {
      rowId = dataInput.readUTF();
    }
    int size = dataInput.readInt();
    for (int i = 0; i < size; i++) {
      int cfLength = dataInput.readInt();
      byte[] cfData = new byte[cfLength];
      dataInput.readFully(cfData, 0, cfLength);
      Text cf = new Text(cfData);
      int cqLength = dataInput.readInt();
      byte[] cqData = new byte[cqLength];
      dataInput.readFully(cqData, 0, cqLength);
      Text cq = new Text(cqData);
      int valSize = dataInput.readInt();
      byte[] val = new byte[valSize];
      for (int j = 0; j < valSize; j++) {
        val[j] = dataInput.readByte();
      }
      tuples.add(new ColumnTuple(cf, cq, val));
    }
  }

  public void add(String cf, String qual, byte[] val) {
    Preconditions.checkNotNull(cf);
    Preconditions.checkNotNull(qual);
    Preconditions.checkNotNull(val);

    add(new Text(cf), new Text(qual), val);
  }

  public void add(Text cf, Text qual, byte[] val) {
    Preconditions.checkNotNull(cf);
    Preconditions.checkNotNull(qual);
    Preconditions.checkNotNull(val);

    tuples.add(new ColumnTuple(cf, qual, val));
  }

  public static class ColumnTuple {
    private final Text cf;
    private final Text cq;
    private final byte[] value;

    public ColumnTuple(Text cf, Text cq, byte[] value) {
      this.value = value;
      this.cf = cf;
      this.cq = cq;
    }

    public byte[] getValue() {
      return value;
    }

    public Text getCf() {
      return cf;
    }

    public Text getCq() {
      return cq;
    }

    @Override
    public int hashCode() {
      HashCodeBuilder hcb = new HashCodeBuilder(9683, 68783);
      return hcb.append(cf).append(cq).append(value).toHashCode();
    }

    @Override
    public boolean equals(Object o) {
      if (o instanceof ColumnTuple) {
        ColumnTuple other = (ColumnTuple) o;
        if (null == cf) {
          if (null != other.cf) {
            return false;
          }
        } else if (!cf.equals(other.cf)) {
          return false;
        }

        if (null == cq) {
          if (null != other.cq) {
            return false;
          }
        } else if (!cq.equals(other.cq)) {
          return false;
        }

        if (null == value) {
          if (null != other.value) {
            return false;
          }
        }

        return Arrays.equals(value, other.value);
      }

      return false;
    }

    @Override
    public String toString() {
      return cf + " " + cq + " " + new String(value);
    }
  }
}