Changed to throw HiveException instead of RuntimeException

myui · Jul 29, 2015 · 8520825 · 8520825
1 parent 44d6f83
commit 8520825
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/main/java/hivemall/tools/mapred/JobConfGetsUDF.java b/src/main/java/hivemall/tools/mapred/JobConfGetsUDF.java
@@ -27,6 +27,7 @@
 import org.apache.hadoop.hive.ql.exec.MapredContext;
 import org.apache.hadoop.hive.ql.exec.MapredContextAccessor;
 import org.apache.hadoop.hive.ql.exec.UDF;
+import org.apache.hadoop.hive.ql.metadata.HiveException;
 import org.apache.hadoop.hive.ql.udf.UDFType;
 import org.apache.hadoop.io.Text;
 import org.apache.hadoop.mapred.JobConf;
@@ -38,18 +39,18 @@
 @UDFType(deterministic = false, stateful = true)
 public class JobConfGetsUDF extends UDF {
 
-    public Text evaluate() {
+    public Text evaluate() throws HiveException {
         return evaluate(null);
     }
 
-    public Text evaluate(@Nullable final String regexKey) {
+    public Text evaluate(@Nullable final String regexKey) throws HiveException {
         MapredContext ctx = MapredContextAccessor.get();
         if(ctx == null) {
-            throw new IllegalStateException("MapredContext is not set");
+            throw new HiveException("MapredContext is not set");
         }
         JobConf jobconf = ctx.getJobConf();
         if(jobconf == null) {
-            throw new IllegalStateException("JobConf is not set");
+            throw new HiveException("JobConf is not set");
         }
 
         String dumped = HadoopUtils.toString(jobconf, regexKey);