svn commit: r790770 - /hadoop/hive/trunk/ql/src/java/org/apache/hadoop/hive/ql/udf/generic/GenericUDFIndex.java

[pr...@apache.org]

Author: prasadc
Date: Fri Jul  3 00:37:13 2009
New Revision: 790770

URL: http://svn.apache.org/viewvc?rev=790770&view=rev
Log:
straggler from hive-593

Added:
    hadoop/hive/trunk/ql/src/java/org/apache/hadoop/hive/ql/udf/generic/GenericUDFIndex.java   (with props)

Added: hadoop/hive/trunk/ql/src/java/org/apache/hadoop/hive/ql/udf/generic/GenericUDFIndex.java
URL: http://svn.apache.org/viewvc/hadoop/hive/trunk/ql/src/java/org/apache/hadoop/hive/ql/udf/generic/GenericUDFIndex.java?rev=790770&view=auto
==============================================================================
--- hadoop/hive/trunk/ql/src/java/org/apache/hadoop/hive/ql/udf/generic/GenericUDFIndex.java (added)
+++ hadoop/hive/trunk/ql/src/java/org/apache/hadoop/hive/ql/udf/generic/GenericUDFIndex.java Fri Jul  3 00:37:13 2009
@@ -0,0 +1,126 @@
+/**
+ * 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.udf.generic;
+
+import org.apache.hadoop.hive.ql.exec.UDFArgumentException;
+import org.apache.hadoop.hive.ql.exec.UDFArgumentLengthException;
+import org.apache.hadoop.hive.ql.exec.UDFArgumentTypeException;
+import org.apache.hadoop.hive.ql.metadata.HiveException;
+import org.apache.hadoop.hive.serde.Constants;
+import org.apache.hadoop.hive.serde2.objectinspector.ListObjectInspector;
+import org.apache.hadoop.hive.serde2.objectinspector.MapObjectInspector;
+import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector;
+import org.apache.hadoop.hive.serde2.objectinspector.PrimitiveObjectInspector;
+import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector.Category;
+import org.apache.hadoop.hive.serde2.objectinspector.primitive.PrimitiveObjectInspectorFactory;
+import org.apache.hadoop.hive.serde2.objectinspector.primitive.PrimitiveObjectInspectorUtils;
+import org.apache.hadoop.io.IntWritable;
+
+public class GenericUDFIndex extends GenericUDF {
+  private MapObjectInspector mapOI;
+  private boolean mapKeyPreferWritable;
+  private ListObjectInspector listOI;
+  private PrimitiveObjectInspector indexOI;
+  private ObjectInspector returnOI;
+  private IntWritable result = new IntWritable(-1);
+
+  @Override
+  public ObjectInspector initialize(ObjectInspector[] arguments)
+      throws UDFArgumentException {
+    if (arguments.length != 2) {
+      throw new UDFArgumentLengthException(
+          "The function INDEX accepts exactly 2 arguments.");
+    }
+    
+    if (arguments[0] instanceof MapObjectInspector) {
+      // index into a map
+      mapOI = (MapObjectInspector)arguments[0];
+      listOI = null;
+    } else if (arguments[0] instanceof ListObjectInspector) {
+      // index into a list
+      listOI = (ListObjectInspector)arguments[0];
+      mapOI = null;
+    } else {
+      throw new UDFArgumentTypeException(0,
+          "\"" + Category.MAP.toString().toLowerCase() 
+          + "\" or \"" + Category.LIST.toString().toLowerCase() 
+          + "\" is expected at function INDEX, but \"" 
+          + arguments[0].getTypeName() + "\" is found");
+    }
+    
+    // index has to be a primitive
+    if (arguments[1] instanceof PrimitiveObjectInspector) {
+      indexOI = (PrimitiveObjectInspector) arguments[1];
+    } else {
+      throw new UDFArgumentTypeException(1,
+          "Primitive Type is expected but " + arguments[1].getTypeName()
+          + "\" is found");
+    }
+
+    if (mapOI != null) {
+      returnOI = mapOI.getMapValueObjectInspector();
+      ObjectInspector keyOI = mapOI.getMapKeyObjectInspector();
+      mapKeyPreferWritable = ((PrimitiveObjectInspector)keyOI).preferWritable();
+    } else {
+      returnOI = listOI.getListElementObjectInspector();
+    }
+    
+    return returnOI;
+  }
+
+  @Override
+  public Object evaluate(DeferredObject[] arguments) throws HiveException {
+    assert(arguments.length == 2);
+    Object main = arguments[0].get();
+    Object index = arguments[1].get();
+
+    if (mapOI != null) {
+      
+      Object indexObject;
+      if (mapKeyPreferWritable) {
+        indexObject = indexOI.getPrimitiveWritableObject(index);
+      } else {
+        indexObject = indexOI.getPrimitiveJavaObject(index);
+      }
+      return mapOI.getMapValueElement(main, indexObject);
+      
+    } else {
+      
+      assert(listOI != null);
+      int intIndex = 0;
+      try {
+        intIndex = PrimitiveObjectInspectorUtils.getInt(index, indexOI);
+      } catch (NullPointerException e) {
+        // If index is null, we should return null.
+        return null;
+      } catch (NumberFormatException e) {
+        // If index is not a number, we should return null.
+        return null;
+      }
+      return listOI.getListElement(main, intIndex);
+      
+    }
+  }
+
+  @Override
+  public String getDisplayString(String[] children) {
+    assert(children.length == 2);
+    return children[0] + "[" + children[1] + "]";
+  }
+}

Propchange: hadoop/hive/trunk/ql/src/java/org/apache/hadoop/hive/ql/udf/generic/GenericUDFIndex.java
------------------------------------------------------------------------------
    svn:eol-style = native