mirror of https://github.com/hashicorp/consul
35 lines
911 B
JavaScript
35 lines
911 B
JavaScript
|
/**
|
||
|
* Wraps an EventSource so that you can `close` and `reopen`
|
||
|
*
|
||
|
* @param {Class} eventSource - EventSource class to extend from
|
||
|
*/
|
||
|
export default function(eventSource = EventSource) {
|
||
|
const OpenableEventSource = function(source, configuration = {}) {
|
||
|
eventSource.apply(this, arguments);
|
||
|
this.configuration = configuration;
|
||
|
};
|
||
|
OpenableEventSource.prototype = Object.assign(
|
||
|
Object.create(eventSource.prototype, {
|
||
|
constructor: {
|
||
|
value: OpenableEventSource,
|
||
|
configurable: true,
|
||
|
writable: true,
|
||
|
},
|
||
|
}),
|
||
|
{
|
||
|
open: function() {
|
||
|
switch (this.readyState) {
|
||
|
case 3: // CLOSING
|
||
|
this.readyState = 1;
|
||
|
break;
|
||
|
case 2: // CLOSED
|
||
|
eventSource.apply(this, [this.source, this.configuration]);
|
||
|
break;
|
||
|
}
|
||
|
return this;
|
||
|
},
|
||
|
}
|
||
|
);
|
||
|
return OpenableEventSource;
|
||
|
}
|