shared-dataloader
A class for sharing dataloaders across GraphQL subscriptions
Installation
yarn add shared-dataloader
What's it do
Allows you to use the same dataloader for multiple GraphQL subscription clients. If you think of a mutation as a single operation, this allows you to share data across that operation.
Usage
1. Create it on your GraphQL server
;const sharedDataLoader = PROD onShare: '_share' ttl: 5000;
2. Add it to your query/mutation context. Call dispose after it's completed.
;; const allMyDataLoaders = todos: todoBatchFn;const dataLoader = sharedDataLoader;const result = await ;dataLoader;
3. Add it to your subscription context. Call dispose after the subscription ends.
; // Important! Note {cache: false}. You should already have been doing this since subs are long lived.const allMyDataLoaders = todos: todoBatchFn cache: false;const dataLoader = sharedDataLoader;const asyncIterator = await ;await ;dataLoader;
4. Share the ID when you push to the pubsub
// UpdateTodo.js { const updatedTodo = db; const operationId = dataLoader; pubsub}
5. Use the shared ID in your subscription iterator and unsub when the sub closes
async { const asyncIterator = pubsub; const getNextPromise = async { const nextRes = await asyncIteratornext; const value done = nextRes; if done return asyncIterator; if valueoperationId dataLoader; return nextRes; }; return { return ; } return dataLoader; return asyncIterator; }
getter
method to get individual loaders:
6. Use the dataloader todos: resolve: source args dataLoader return dataLoader
API
The SharedDataLoader takes the following args
PROD
: true if running in production (don't show warnings). Defaults to false.ttl
: time to live (ms). Smaller number means less memory usage. 100-5000 is reasonable.onShare
: The name of the method in your dataloader object to call when you callshare()
. Use this to sanitize your dataloader of any sensitive info that might have been provided to it (such as an auth token) This is not required, but provides peace of mind if you're unsure about your schema authorization.
The SharedDataLoader has a single public method:
add(allMyLoaders)
: Call this with an object containing all your loaders. It returns a ShareableDataLoader.
The ShareableDataLoader (the result of SharedDataLoader#add) has the following API
dispose(options)
: dispose of the data loader if it is not being shared. It has the following option:force
: boolean, defaults to false. If true, calling dispose will dispose of the dataloader even if it is being shared.
share(ttl)
: Returns a unique ID to be fed touseShared
. Also begins the TTL. Although strongly discouraged, you may provide a TTL here to override the one defined by theSharedDataLoader
. This is useful if you need to extend the time because you are making an external API call, or usingsetTimeout
.useShared(operationId)
: Replaces the current dataloader with the dataloader belonging to theoperationId
. You'll want to call this on your subscription with theoperationId
that comes from the mutationgetID
: returns the ID of the current dataloader. Useful for testing.isShared
: returns true if the dataloader is currently being shared. Useful for testing.
License
MIT