Apollo Link overview

Customize Apollo Client's data flow


The Apollo Link library helps you customize the flow of data between Apollo Client and your GraphQL server. You can define your client's network behavior as a chain of link objects that execute in a sequence:

Each link should represent either a self-contained modification to a GraphQL operation or a side effect (such as logging).

In the above diagram:

  1. The first link might log the details of the operation for debugging purposes.

  2. The second link might add an HTTP header to the outgoing operation request for authentication purposes.

  3. The final link (called the terminating link) sends the operation to its destination (usually a GraphQL server over HTTP).

  4. The server's response is passed back up each link in reverse order, enabling links to modify the response or take other actions before the data is cached.

A link chain consists of one or more link objects. Each link represents either a self-contained modification to a GraphQL operation or a side effect (such as logging). By composing these links into a chain, you can create an arbitrarily complex model for your client's data flow.

When you compose multiple links together, the result is a single link object that represents the entire chain. You provide the composed link to your Apollo Client instance using the link option:

TypeScript
1import { ApolloClient, InMemoryCache } from "@apollo/client";
2
3const client = new ApolloClient({
4  link: /* your composed link */,
5  cache: new InMemoryCache(),
6});

The last link in the link chain is called the terminating link. Instead of calling the forward function, it is responsible for sending the GraphQL operation to the destination that executes it (typically a GraphQL server) and returning a result. If a link chain consists of a single link, the single link is the terminating link.

HttpLink and BatchHttpLink are examples of terminating links.

You compose multiple links together to form a link chain using two forms of link composition: additive and directional.

note
No matter how your chain branches, each branch always ends in a terminating link.

Additive composition

Additive composition composes a link chain by executing links in serial order. You use the from and concat helpers to create a link chain using additive composition.

from

The most common way to compose multiple links together is to use the from helper. Pass an array of link objects to from to create a composed link that executes each link in serial order:

TypeScript
1import { from, HttpLink, ApolloLink } from "@apollo/client";
2import { RetryLink } from "@apollo/client/link/retry";
3import MyAuthLink from "../auth";
4
5const link = from([
6  new RetryLink(),
7  new MyAuthLink(),
8  new HttpLink({ uri: "http://localhost:4000/graphql" }),
9]);
10
11// this also works
12const link = ApolloLink.from([
13  new RetryLink(),
14  new MyAuthLink(),
15  new HttpLink({ uri: "http://localhost:4000/graphql" }),
16]);
concat

Combine two links together into a single composed link using the concat method.

TypeScript
1import { concat, ApolloLink, HttpLink } from "@apollo/client";
2import MyAuthLink from "../auth";
3
4const link = concat(
5  new MyAuthLink(),
6  new HttpLink({ uri: "http://localhost:4000/graphql" })
7);
8
9// this also works
10const link = ApolloLink.concat(
11  new MyAuthLink(),
12  new HttpLink({ uri: "http://localhost:4000/graphql" })
13);

Each link object includes a concat instance method to combine two links into a single composed link. This is useful to combine multiple links together using a chain of function calls:

TypeScript
1import { ApolloLink, HttpLink } from "@apollo/client";
2import { RetryLink } from "@apollo/client/link/retry";
3import MyAuthLink from "../auth";
4
5const link = new RetryLink()
6  .concat(new MyAuthLink())
7  .concat(new HttpLink({ uri: "http://localhost:4000/graphql" }));
note
The link chain created by the from and concat examples are functionally equivalent and differ only in style. Choose the style that best fits your application. You may choose to mix and match these styles as needed.

Directional composition

You might want your link chain's execution to branch, depending on the details of the operation being performed. You use the split function to create a link that conditionally routes to different sub-chains.

The split function takes three arguments:

NameDescription
testA function that takes in the current operation and returns true or false. Returning true executes the left link. Returning false executes the right link.
leftThe link passed as the second argument to split. This link executes when the test function returns true.
rightAn optional link passed as the third argument to split. This link executes when the test function returns false. If this is not provided, the request handler's forward parameter is used.

The following example uses split to create a link that routes to different HttpLink instances depending on the associated context's version:

TypeScript
1import { ApolloLink, HttpLink, split } from "@apollo/client";
2
3const link = split(
4  (operation) => operation.getContext().version === 1,
5  new HttpLink({ uri: "http://localhost:4000/v1/graphql" }),
6  new HttpLink({ uri: "http://localhost:4000/v2/graphql" })
7);
8
9// this also works
10const link = ApolloLink.split(
11  (operation) => operation.getContext().version === 1,
12  new HttpLink({ uri: "http://localhost:4000/v1/graphql" }),
13  new HttpLink({ uri: "http://localhost:4000/v2/graphql" })
14);
note
Each link instance also provides a split function. This is used to apply directional composition to an existing link.
Example
The following example adds directional composition to a previousLink to determine which HttpLink should be used as the terminating link.
TypeScript
1const link = previousLink.split(
2  (operation) => operation.getContext().version === 1,
3  new HttpLink({ uri: "http://localhost:4000/v1/graphql" }),
4  new HttpLink({ uri: "http://localhost:4000/v2/graphql" })
5);

Other uses for the split method include:

  • Using different transport methods depending on the operation type (such as HTTP for queries and WebSocket for subscriptions)

  • Customizing the number of allowed retry attempts depending on the operation type

  • Customizing logic depending on whether a user is logged in

In the following example, all subscription operations are sent to GraphQLWsLink, with all other operations sent to HttpLink:

TypeScript
1import { split, HttpLink } from "@apollo/client";
2import { GraphQLWsLink } from "@apollo/client/link/subscriptions";
3import { OperationTypeNode } from "graphql";
4import { createClient } from "graphql-ws";
5
6const link = split(
7  ({ operationType }) => {
8    return operationType === OperationTypeNode.SUBSCRIPTION;
9  },
10  new GraphQLWsLink(createClient({ url: "ws://localhost:3000/subscriptions" })),
11  new HttpLink({ uri: "http://localhost:4000/graphql" })
12);

A link object is an instance of the ApolloLink class. Each link must define a request handler to handle the request.

The request handler

Every link must define a request handler. The request handler is a method responsible for performing some logic and executing the request, either by forwarding the operation to the next link in the chain, or sending the operation to the destination that executes it, such as a GraphQL server.

This request handler receives the following arguments:

  • operation: The operation object that provides information about the currently executed GraphQL request passed through the link.

  • forward: A function called to execute the next link in the chain.

Apollo Client executes a GraphQL operation by calling the request handler of the first link in the composed link chain. Each non-terminating link is then responsible for executing its logic and delegating execution down to the next link by calling the forward function. After the operation reaches the terminating link, the request handler sends the operation to the destination and reports the result back up through the chain.

Define a request handler using a callback function

The most common way to provide a request handler is by passing a callback function to the ApolloLink constructor:

TypeScript
1import { ApolloLink } from "@apollo/client";
2
3const link = new ApolloLink((operation, forward) => {
4  // Handle the request
5});

The callback function is called for each GraphQL operation sent through the link chain.

You define a request handler for a subclass of ApolloLink by overriding the request method. Subclassing is typically used to create stateful links.

TypeScript
1class MyCustomLink extends ApolloLink {
2  constructor() {
3    super();
4    // ...other setup logic
5  }
6
7  request(operation, forward) {
8    // Handle the request
9  }
10}

The request method is called for each GraphQL operation sent through the link chain.

The Operation object

The Operation object includes the following fields:

ApolloClient

The Apollo Client instance executing the request.

Record<string, any>

A map that stores extensions data to be sent to the server.

() => Readonly<ApolloLink.OperationContext>

A function that gets the current context of the request. This can be used by links to determine which actions to perform. See managing context

string | undefined

The string name of the GraphQL operation. If it is anonymous, operationName will be undefined.

OperationTypeNode

The type of the GraphQL operation, such as query or mutation.

DocumentNode

A DocumentNode that describes the operation taking place.

{ (context: Partial<ApolloLink.OperationContext>): void; (updateContext: (previousContext: Readonly<ApolloLink.OperationContext>) => Partial<ApolloLink.OperationContext>): void; }

A function that takes either a new context object, or a function which takes in the previous context and returns a new one. See managing context.

OperationVariables

A map of GraphQL variables being sent with the operation.

The forward function

After your custom link's request handler is finished performing its logic, return a call to the forward function and provide the operation as an argument. Calling the forward function executes the next link in the chain.

TypeScript
1const link = new ApolloLink((operation, forward) => {
2  // ...Handle the request
3
4  // Execute the next link in the chain
5  return forward(operation);
6});

The forward function returns an Observable provided by the RxJS library. Learn more about Observables in the RxJS documentation.

caution
A non-terminating custom link's request handler must call forward(operation). If it doesn't, the link is treated as a terminating link. This might result in the associated GraphQL operation not executed and the request remains pending indefinitely.

Handling a response

When your GraphQL server responds with an operation result, that result is passed back up through each link in your chain:

Each link can perform custom logic on the result using operators. These are special functions passed to an Observable's pipe function.

The following example uses the RxJS map operator to modify the result before returning it back up the chain:

TypeScript
1import { map } from "rxjs";
2
3const link = new ApolloLink((operation, forward) => {
4  return forward(operation).pipe(
5    map((result) => {
6      // ...modify result as desired here...
7      return result;
8    })
9  );
10});

This modification is seen by any previous link in the chain because the result travels back up the chain in reverse order.

note
The map function is the most common operator for performing modifications to the result. See the RxJS operators documentation for references to other operators that provide other functionality that you can use when building custom links.

Executing side-effects

Links commonly perform side-effects when receiving responses instead of modifying the result.

The following example uses the tap operator to estimate the round-trip time of an operation by defining a start time on the request context then logging the total time when the response is received.

TypeScript
1import { ApolloLink } from "@apollo/client";
2import { tap } from "rxjs";
3
4const roundTripLink = new ApolloLink((operation, forward) => {
5  // Called before operation is sent to server
6  operation.setContext({ start: new Date() });
7
8  return forward(operation).pipe(
9    tap((result) => {
10      // Called after server responds
11      const time = new Date() - operation.getContext().start;
12      console.log(
13        `Operation ${operation.operationName} took ${time} to complete`
14      );
15    })
16  );
17});
note
The tap function is the most common operator for performing side-effects on the response. See the RxJS operators documentation for references to other operators that provide other functionality that you might need when building custom links.

Most links perform the same logic for every operation they process and they don't need to know anything about operations that have been executed previously. These links are stateless.

Stateless links typically define request handlers by providing a callback function to the constructor of an ApolloLink object:

TypeScript
1import { ApolloLink } from "@apollo/client";
2import { tap } from "rxjs";
3
4const consoleLink = new ApolloLink((operation, forward) => {
5  console.log(`starting request for ${operation.operationName}`);
6  return forward(operation).pipe(
7    tap(() => {
8      console.log(`ending request for ${operation.operationName}`);
9    })
10  );
11});

Stateless links are great for building middleware. The following link adds an Authorization header to every outgoing request:

TypeScript
1import { ApolloLink } from "@apollo/client";
2
3const authLink = new ApolloLink((operation, forward) => {
4  operation.setContext(({ headers }) => ({
5    headers: {
6      authorization: Auth.userId(), // however you get your token
7      ...headers,
8    },
9  }));
10
11  return forward(operation);
12});

You can customize stateless links by wrapping them in a function:

TypeScript
1import { ApolloLink } from "@apollo/client";
2
3function reportErrors(errorCallback) {
4  return new ApolloLink((operation, forward) => {
5    return new Observable((observer) => {
6      const observable = forward(operation);
7      const subscription = observable.subscribe({
8        next(value) {
9          observer.next(value);
10        },
11        error(networkError) {
12          errorCallback({ networkError, operation });
13          observer.error(networkError);
14        },
15        complete() {
16          observer.complete();
17        },
18      });
19
20      return () => subscription.unsubscribe();
21    });
22  });
23}
24
25const link = reportErrors(console.error);

You can also create stateless links by subclassing the ApolloLink class and overriding the constructor and request handler.

Here's the same reportErrors link written as a subclass of ApolloLink:

TypeScript
1import { ApolloLink } from "@apollo/client";
2import { tap } from "rxjs";
3
4class ReportErrorLink extends ApolloLink {
5  constructor(errorCallback) {
6    super();
7    this.errorCallback = errorCallback;
8  }
9  request(operation, forward) {
10    return forward(operation).pipe(
11      tap({
12        // errors will be sent to the errorCallback
13        error: this.errorCallback,
14      })
15    );
16  }
17}
18
19const link = new ReportErrorLink(console.error);
note
The ReportErrorLink link is still considered a stateless link despite being built as a subclass because it does not maintain state between requests.

Links that maintain state between operations are called stateful links. You create stateful links by defining subclasses of ApolloLink and overriding the constructor and request methods.

The following example maintains an operation counter using an instance variable. The counter is incremented for every request that passes through the link.

TypeScript
1import { ApolloLink } from "@apollo/client";
2
3class OperationCountLink extends ApolloLink {
4  constructor() {
5    super();
6    this.operationCount = 0;
7  }
8  request(operation, forward) {
9    this.operationCount += 1;
10    return forward(operation);
11  }
12}
13
14const link = new OperationCountLink();

Managing context

As an operation moves along the link chain, it maintains a context that each link can read and modify. This enables both links and request-based APIs (such as useQuery) to pass metadata along the chain that other links use in their execution logic. Context is not included in the terminating link's request to the GraphQL server or other destination.

note
If your application is built using TypeScript, we recommend reading the Defining context types guide to learn how to provide types for the context object.

Reading context

You get the current context object by calling operation.getContext() in a request handler.

TypeScript
1const link = new ApolloLink((operation, forward) => {
2  const context = operation.getContext();
3
4  // handle the request
5});
note
Changes to the context object returned from operation.getContext() are not persisted, and downstream links won't see them. Use operation.setContext() to make modifications to context.

Modifying context

Modify context by calling operation.setContext(). You provide the updated context as an object directly to setContext.

TypeScript
1const timeStartLink = new ApolloLink((operation, forward) => {
2  operation.setContext({ start: new Date() });
3
4  return forward(operation);
5});

It is common to read existing context to make modifications to it before writing it back. As a shortcut, you can provide a callback function to operation.setContext(). The callback function provides the previous context as an argument and expects the function to return the new context.

TypeScript
1const counterLink = new ApolloLink((operation, forward) => {
2  operation.setContext((prevContext) => ({
3    count: prevContext.count + 1,
4  }));
5
6  return forward(operation);
7});

Providing context for an operation

Context is typically used to communicate between a request API (such as useQuery) and the links in the link chain. You provide context for a particular operation using the context option.

The following example provides the initial context with useQuery to define the initial count from the previous example.

TypeScript
1function MyComponent() {
2  const { data } = useQuery(MY_QUERY, { context: { count: 10 } });
3
4  // render data
5}
Feedback

Edit on GitHub

Ask Community