Right now, we're getting all of the habits and entries in one query. This is fine for such a small data set, but what if we had hundreds of habits with hundreds of entries? It might be nice to only get the entries when clicking a button instead of when the page renders.

We can do this using the useLazyQuery hook that Apollo provides. This hook lets us run a query in response to an event like a button click instead of just when a component renders.

First, let's remove the entries from the habits query in App.js:

// src/App.js
export const HABITS_QUERY = gql`
  query HABITS_QUERY {
    habits {

    totalPoints {

Then, we'll write our query for getting the entries separately based on the habit ID:

// src/Habit.js
export const ENTRIES_QUERY = gql`
  query ENTRIES_QUERY($id: ID!) {
    getEntriesByHabitId(id: $id) {

Now we're ready for the useLazyQuery hook. Let's import it at the top of Habit.js:

// src/Habit.js
import { useMutation, gql, useLazyQuery } from "@apollo/client";

Then, add it underneath the rest of our hooks:

// src/Habit.js
const [loadEntries, { data: entriesData }] = useLazyQuery(ENTRIES_QUERY);

Notice that, just like with useMutation, we're getting an array that contains a function and an object with data, error, and loading. We're renaming data to entriesData to keep things clear and we're skipping error and loading since we've covered that in previous lessons.

Let's also create a variable to help us avoid null reference errors:

// src/Habit.js
const entries =
    entriesData && entriesData.getEntriesByHabitId
      ? entriesData.getEntriesByHabitId
      : [];

Be sure to swap out any references of habit.entries with our new entries variable like we do in the video.

Finally, we can create our button to load the entries:

// src/Habit.js
    onClick={() => loadEntries({ variables: { id: habit.id } })}

If you check out the browser, you should be able to click the button and load the habit's entries on demand!