From 904e6872a2ee7e67e10d3db3598e47f5d4fc41cd Mon Sep 17 00:00:00 2001
From: Amogh Margoor <amoghm@qubole.com>
Date: Mon, 18 Jan 2016 03:45:25 -0500
Subject: [PATCH] chg:dev:NEZ-100 Enable setting of Materialized Service
 instance as thread local

---
 .../org/apache/calcite/materialize/MaterializationService.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/core/src/main/java/org/apache/calcite/materialize/MaterializationService.java b/core/src/main/java/org/apache/calcite/materialize/MaterializationService.java
index fc9ac2b28ffd..bae0d51164db 100644
--- a/core/src/main/java/org/apache/calcite/materialize/MaterializationService.java
+++ b/core/src/main/java/org/apache/calcite/materialize/MaterializationService.java
@@ -334,6 +334,9 @@ public static void setThreadLocal() {
     THREAD_INSTANCE.set(new MaterializationService());
   }
 
+  public static void setThreadLocal(MaterializationService service) {
+    THREAD_INSTANCE.set(service);
+  }
   /** Returns the instance of the materialization service. Usually the global
    * one, but returns a thread-local one during testing (when
    * {@link #setThreadLocal()} has been called by the current thread). */