/tags/release-0.0.0-rc0/hive/external/ql/src/java/org/apache/hadoop/hive/ql/exec/persistence/MapJoinObjectKey.java
Java | 160 lines | 102 code | 24 blank | 34 comment | 26 complexity | 6bd62ebd3d202d85a8ef733eb44f7fce MD5 | raw file
Possible License(s): Apache-2.0, BSD-3-Clause, JSON, CPL-1.0
- /**
- * 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.ql.exec.persistence;
- import java.io.IOException;
- import java.io.ObjectInput;
- import java.io.ObjectOutput;
- import java.util.ArrayList;
- import org.apache.hadoop.hive.ql.exec.MapJoinMetaData;
- import org.apache.hadoop.hive.ql.exec.HashTableSinkOperator.HashTableSinkObjectCtx;
- import org.apache.hadoop.hive.serde2.SerDeException;
- import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspectorUtils;
- import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspectorUtils.ObjectInspectorCopyOption;
- import org.apache.hadoop.io.Writable;
- /**
- * Map Join Object used for both key.
- */
- public class MapJoinObjectKey extends AbstractMapJoinKey {
- protected transient Object[] obj;
- public MapJoinObjectKey() {
- }
- /**
- * @param metadataTag
- * @param obj
- */
- public MapJoinObjectKey(Object[] obj) {
- this.obj = obj;
- }
- @Override
- public boolean equals(Object o) {
- if (o instanceof MapJoinObjectKey) {
- MapJoinObjectKey mObj = (MapJoinObjectKey) o;
- Object[] mObjArray = mObj.getObj();
- if ((obj == null) && (mObjArray == null)) {
- return true;
- }
- if ((obj != null) && (mObjArray != null)) {
- if (obj.length == mObjArray.length) {
- for (int i = 0; i < obj.length; i++) {
- if (!obj[i].equals(mObjArray[i])) {
- return false;
- }
- }
- return true;
- }
- }
- }
- return false;
- }
- @Override
- public int hashCode() {
- int hashCode;
- if (obj == null) {
- hashCode = metadataTag;
- } else {
- hashCode = 1;
- for (int i = 0; i < obj.length; i++) {
- Object o = obj[i];
- hashCode = 31 * hashCode + (o == null ? 0 : o.hashCode());
- }
- }
- return hashCode;
- }
- @Override
- public void readExternal(ObjectInput in) throws IOException,
- ClassNotFoundException {
- try {
- // get the tableDesc from the map stored in the mapjoin operator
- HashTableSinkObjectCtx ctx = MapJoinMetaData.get(
- Integer.valueOf(metadataTag));
- Writable val = ctx.getSerDe().getSerializedClass().newInstance();
- val.readFields(in);
- ArrayList<Object> list = (ArrayList<Object>) ObjectInspectorUtils.copyToStandardObject(ctx
- .getSerDe().deserialize(val), ctx.getSerDe().getObjectInspector(),
- ObjectInspectorCopyOption.WRITABLE);
- if(list == null){
- obj = new ArrayList(0).toArray();
- }else{
- obj = list.toArray();
- }
- } catch (Exception e) {
- throw new IOException(e);
- }
- }
- @Override
- public void writeExternal(ObjectOutput out) throws IOException {
- try {
- // get the tableDesc from the map stored in the mapjoin operator
- HashTableSinkObjectCtx ctx = MapJoinMetaData.get(
- Integer.valueOf(metadataTag));
- // Different processing for key and value
- Writable outVal = ctx.getSerDe().serialize(obj, ctx.getStandardOI());
- outVal.write(out);
- } catch (SerDeException e) {
- throw new IOException(e);
- }
- }
- /**
- * @return the obj
- */
- public Object[] getObj() {
- return obj;
- }
- /**
- * @param obj
- * the obj to set
- */
- public void setObj(Object[] obj) {
- this.obj = obj;
- }
- @Override
- public boolean hasAnyNulls(){
- if (obj != null && obj.length> 0) {
- for (Object k : obj) {
- if (k == null) {
- return true;
- }
- }
- }
- return false;
- }
- }