appsync-apollo-subscription-node.js

import { Credentials } from "@aws-amplify/core";
import { AuthOptions, createAuthLink } from "aws-appsync-auth-link";
import { createSubscriptionHandshakeLink } from "aws-appsync-subscription-link";
import {
  ApolloClient, ApolloLink, HttpLink, InMemoryCache
} from "@apollo/client/core"
import gql from 'graphql-tag';

global.WebSocket = require('ws');

// Amplify config setup... 

export async function subscribe() {
  const url = config.graphqlEndpoint;
  const region = config.region;
  const auth: AuthOptions = {
    type: "AWS_IAM",
    credentials: async () => {
      const credentials = await Credentials.get();
      return Credentials.shear(credentials);
    },
  };

  const httpLink = new HttpLink({ uri: url, })

  const links = ApolloLink.from([
    createAuthLink({ url, region, auth }),
    createSubscriptionHandshakeLink({ url, region, auth }, httpLink),
  ]);

  const apolloClient = new ApolloClient({
    link: links,
    cache: new InMemoryCache(),
  })

  apolloClient.subscribe({ query: gql`subscription onLogs {
    logs {
      message
      timestamp
    }
  }`}).subscribe({
    next: data => {
      console.log(JSON.stringify(data, null, 2));
    },
    error: error => {
      console.log(JSON.stringify(error, null, 2));
    }
  });

}