rxdb
Version:
A local-first realtime NoSQL Database for JavaScript applications - https://rxdb.info/
41 lines • 1.39 kB
JavaScript
/**
* Import from 'graphql-ws/client' instead of 'graphql-ws' so that
* only the zero-dependency client is bundled. The package root also
* contains the server code which pulls in the whole graphql package.
* @link https://github.com/enisdenjo/graphql-ws/blob/master/CHANGELOG.md#600
*/
import { createClient } from 'graphql-ws/client';
import { getFromMapOrCreate, getFromMapOrThrow } from "../../plugins/utils/index.js";
import { WebSocket } from 'ws';
export var GRAPHQL_WEBSOCKET_BY_URL = new Map();
export function getGraphQLWebSocket(url, headers, options = {}) {
var has = getFromMapOrCreate(GRAPHQL_WEBSOCKET_BY_URL, url, () => {
var connectionParamsHeaders = headers ? {
headers
} : undefined;
var wsClient = createClient({
...options,
url,
shouldRetry: () => true,
webSocketImpl: WebSocket,
connectionParams: options.connectionParams || connectionParamsHeaders
});
return {
url,
socket: wsClient,
refCount: 1
};
}, value => {
value.refCount = value.refCount + 1;
});
return has.socket;
}
export function removeGraphQLWebSocketRef(url) {
var obj = getFromMapOrThrow(GRAPHQL_WEBSOCKET_BY_URL, url);
obj.refCount = obj.refCount - 1;
if (obj.refCount === 0) {
GRAPHQL_WEBSOCKET_BY_URL.delete(url);
obj.socket.dispose();
}
}
//# sourceMappingURL=graphql-websocket.js.map