From df0cf11a0f93da1197c9bd790fc8aaa00e1d1df8 Mon Sep 17 00:00:00 2001
From: John Cowen <johncowen@users.noreply.github.com>
Date: Tue, 4 Jun 2019 15:51:10 +0100
Subject: [PATCH] ui: Put some options in for event source runners (behind
 flags) (#5738)

---
 ui-v2/app/utils/dom/event-source/index.js | 55 ++++++++++++++++++++++-
 1 file changed, 54 insertions(+), 1 deletion(-)

diff --git a/ui-v2/app/utils/dom/event-source/index.js b/ui-v2/app/utils/dom/event-source/index.js
index d7168ad913..668ce29335 100644
--- a/ui-v2/app/utils/dom/event-source/index.js
+++ b/ui-v2/app/utils/dom/event-source/index.js
@@ -15,8 +15,61 @@ import ReopenableEventSourceFactory from 'consul-ui/utils/dom/event-source/reope
 import BlockingEventSourceFactory from 'consul-ui/utils/dom/event-source/blocking';
 import StorageEventSourceFactory from 'consul-ui/utils/dom/event-source/storage';
 
+import EmberObject from '@ember/object';
+import { task } from 'ember-concurrency';
+
+import env from 'consul-ui/env';
+
+let runner;
+switch (env('CONSUL_UI_REALTIME_RUNNER')) {
+  case 'ec':
+    runner = function(target, configuration, isClosed) {
+      return EmberObject.extend({
+        task: task(function* run() {
+          while (!isClosed(target)) {
+            yield target.source.bind(target)(configuration);
+          }
+        }),
+      })
+        .create()
+        .get('task')
+        .perform();
+    };
+    break;
+  case 'generator':
+    runner = async function(target, configuration, isClosed) {
+      const run = function*() {
+        while (!isClosed(target)) {
+          yield target.source.bind(target)(configuration);
+        }
+      };
+      let step = run().next();
+      let res;
+      while (!step.done) {
+        res = await step.value;
+        step = run().next();
+      }
+      return res;
+    };
+    break;
+  case 'async':
+    runner = async function(target, configuration, isClosed) {
+      const run = function() {
+        return target.source.bind(target)(configuration);
+      };
+      let res;
+      while (!isClosed(target)) {
+        res = await run();
+      }
+      return res;
+    };
+    break;
+  default:
+  // use the default runner
+}
+
 // All The EventSource-i
-export const CallableEventSource = CallableEventSourceFactory(EventTarget, Promise);
+export const CallableEventSource = CallableEventSourceFactory(EventTarget, Promise, runner);
 export const ReopenableEventSource = ReopenableEventSourceFactory(CallableEventSource);
 export const BlockingEventSource = BlockingEventSourceFactory(ReopenableEventSource);
 export const StorageEventSource = StorageEventSourceFactory(EventTarget, Promise);