From 7c40bde18945848aedfc94c42bfa5e84b7b30a7b Mon Sep 17 00:00:00 2001
From: Ben Sidhom <sidhom@google.com>
Date: Thu, 22 Mar 2018 16:27:21 -0700
Subject: [PATCH] Don't close data channels from environment sessions

---
 .../flink/execution/JobResourceEnvironmentSession.java        | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/runners/flink/src/main/java/org/apache/beam/runners/flink/execution/JobResourceEnvironmentSession.java b/runners/flink/src/main/java/org/apache/beam/runners/flink/execution/JobResourceEnvironmentSession.java
index 14065ba5dd4e3..538fa8eae1462 100644
--- a/runners/flink/src/main/java/org/apache/beam/runners/flink/execution/JobResourceEnvironmentSession.java
+++ b/runners/flink/src/main/java/org/apache/beam/runners/flink/execution/JobResourceEnvironmentSession.java
@@ -107,6 +107,8 @@ public void close() throws Exception {
     // TODO: eventually use this for reference counting open sessions.
     this.isClosed = true;
     client.close();
-    dataServer.close();
+    // TODO: Close data server when we solve bundle completion/channel close races between runners
+    // and SDK harnesses.
+    // dataServer.close();
   }
 }