diff --git a/ql/src/java/org/apache/hadoop/hive/ql/exec/FunctionRegistry.java b/ql/src/java/org/apache/hadoop/hive/ql/exec/FunctionRegistry.java index 5353062..8cdec3d 100644 --- a/ql/src/java/org/apache/hadoop/hive/ql/exec/FunctionRegistry.java +++ b/ql/src/java/org/apache/hadoop/hive/ql/exec/FunctionRegistry.java @@ -101,6 +101,7 @@ import org.apache.hadoop.hive.ql.udf.UDFToShort; import org.apache.hadoop.hive.ql.udf.UDFToString; import org.apache.hadoop.hive.ql.udf.UDFType; +import org.apache.hadoop.hive.ql.udf.UDFUUID; import org.apache.hadoop.hive.ql.udf.UDFUnbase64; import org.apache.hadoop.hive.ql.udf.UDFUnhex; import org.apache.hadoop.hive.ql.udf.UDFWeekOfYear; @@ -235,6 +236,7 @@ system.registerUDF("sha", UDFSha1.class, false); system.registerGenericUDF("aes_encrypt", GenericUDFAesEncrypt.class); system.registerGenericUDF("aes_decrypt", GenericUDFAesDecrypt.class); + system.registerUDF("uuid", UDFUUID.class, false); system.registerGenericUDF("encode", GenericUDFEncode.class); system.registerGenericUDF("decode", GenericUDFDecode.class); diff --git a/ql/src/java/org/apache/hadoop/hive/ql/udf/UDFUUID.java b/ql/src/java/org/apache/hadoop/hive/ql/udf/UDFUUID.java new file mode 100644 index 0000000..1340ded --- /dev/null +++ b/ql/src/java/org/apache/hadoop/hive/ql/udf/UDFUUID.java @@ -0,0 +1,50 @@ +/** + * 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; + +import java.util.UUID; + +import org.apache.hadoop.hive.ql.exec.Description; +import org.apache.hadoop.hive.ql.exec.UDF; +import org.apache.hadoop.io.Text; + +/** + * UDFUUID. + * + */ +@Description(name = "uuid", +value = "_FUNC_() - Returns a universally unique identifier (UUID) string.", +extended = "The value is returned as a canonical UUID 36-character string.\n" ++ "Example:\n" ++ " > SELECT _FUNC_();\n" ++ " '0baf1f52-53df-487f-8292-99a03716b688'\n" ++ " > SELECT _FUNC_();\n" ++ " '36718a53-84f5-45d6-8796-4f79983ad49d'") +public class UDFUUID extends UDF { + private final Text result = new Text(); + /** + * Returns a universally unique identifier (UUID) string (36 characters). + * + * @return Text + */ + public Text evaluate() { + result.set(UUID.randomUUID().toString()); + return result; + } +} diff --git a/ql/src/test/org/apache/hadoop/hive/ql/udf/TestUDFUUID.java b/ql/src/test/org/apache/hadoop/hive/ql/udf/TestUDFUUID.java new file mode 100644 index 0000000..5f04547 --- /dev/null +++ b/ql/src/test/org/apache/hadoop/hive/ql/udf/TestUDFUUID.java @@ -0,0 +1,38 @@ +/** + * 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; + +import junit.framework.TestCase; + +import org.junit.Test; + +public class TestUDFUUID extends TestCase { + @Test + public void testUUID() throws Exception { + UDFUUID udf = new UDFUUID(); + + String id1 = udf.evaluate().toString(); + String id2 = udf.evaluate().toString(); + + assertFalse(id1.equals(id2)); + + assertEquals(id1.length(), 36); + assertEquals(id2.length(), 36); + } +}