open-consul/ui/packages/consul-ui/app/services/client/connections.js

98 lines
2.6 KiB
JavaScript
Raw Normal View History

import Service, { inject as service } from '@ember/service';
export default class ConnectionsService extends Service {
@service('dom')
dom;
@service('env')
env;
@service('data-source/service')
data;
init() {
super.init(...arguments);
this._listeners = this.dom.listeners();
this.connections = new Set();
this.addVisibilityChange();
}
willDestroy() {
this._listeners.remove();
this.purge();
super.willDestroy(...arguments);
}
addVisibilityChange() {
// when the user hides the tab, abort all connections
this._listeners.add(this.dom.document(), {
visibilitychange: e => {
if (e.target.hidden) {
this.purge(-1);
}
},
});
}
whenAvailable(e) {
// if the user has hidden the tab (hidden browser/tab switch)
// any aborted errors should restart
const doc = this.dom.document();
if (doc.hidden) {
return new Promise(resolve => {
const remove = this._listeners.add(doc, {
visibilitychange: function(event) {
remove();
// we resolve with the event that comes from
// whenAvailable not visibilitychange
resolve(e);
},
});
});
}
return Promise.resolve(e);
}
purge(statusCode = 0) {
[...this.connections].forEach(function(connection) {
// Cancelled
connection.abort(statusCode);
});
this.connections = new Set();
}
acquire(request) {
if (this.connections.size >= this.env.var('CONSUL_HTTP_MAX_CONNECTIONS')) {
const closed = this.data.closed();
let connection = [...this.connections].find(item => {
const id = item.headers()['x-request-id'];
if (id) {
return closed.includes(item.headers()['x-request-id']);
}
return false;
});
if (typeof connection === 'undefined') {
// all connections are being used on the page
// if the new one is a blocking query then cancel the oldest connection
if (request.headers()['content-type'] === 'text/event-stream') {
connection = this.connections.values().next().value;
}
// otherwise wait for a connection to become available
}
// cancel the connection
if (typeof connection !== 'undefined') {
// if its a shared blocking query cancel everything
// listening to it
this.release(connection);
// Too Many Requests
connection.abort(429);
}
}
this.connections.add(request);
}
release(request) {
this.connections.delete(request);
}
}