issue-32: adding new signature (cerner#33)

joelplourde3 · Jul 17, 2018 · 2d30289 · 2d30289
1 parent f808237
commit 2d30289
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/bunsen-core/src/main/java/com/cerner/bunsen/ValueSetUdfs.java b/bunsen-core/src/main/java/com/cerner/bunsen/ValueSetUdfs.java
@@ -94,6 +94,19 @@ public static synchronized void pushUdf(SparkSession spark, BroadcastableValueSe
 
     Broadcast<BroadcastableValueSets> broadcast = ctx.broadcast(valueSets);
 
+    pushUdf(spark, broadcast);
+  }
+
+  /**
+   * Pushes an "in_valueset" UDF that uses an already broadcast instance of
+   * {@link BroadcastableValueSets} for its content.
+   *
+   * @param spark the spark session
+   * @param broadcast the broadcast valuesets to use in the UDF
+   */
+  public static synchronized void pushUdf(SparkSession spark,
+      Broadcast<BroadcastableValueSets> broadcast) {
+
     spark.udf()
         .register("in_valueset",
             new InValuesetUdf(broadcast),
@@ -103,7 +116,6 @@ public static synchronized void pushUdf(SparkSession spark, BroadcastableValueSe
     valueSetStack.push(broadcast);
   }
 
-
   /**
    * Pops a BroadcastableValueSets from the user-defined function stack.
    *