Proper way to set stale time using data response?

[Falci]

I have a function that returns the JWT from localStorage. I wrapped this function with useQuery. I'd like to set the stale time based on the JWT's expiration property.

This is the code I have:

export const useSession = () => {
  const client = useQueryClient();

  return useQuery({
    queryKey: ['session'],
    queryFn: () => getSessions(),
    select: (token) => {
      if (token) {
        client.setQueryDefaults(['session'], {
          staleTime: token.exp.getTime() - Date.now(),
        });
      }
      return data;
    },
  });
};

Is there a better solution?

[TkDodo]

This won’t work as setQueryDefaults can’t impact an active query. It’s only for queries that mount afterwards. You want to set staleTime to a function:

useQuery({
  queryKey: ['session'],
  queryFn: () => getSessions(),
  staleTime: (query) => {
    // compute based on `query.state.data` (can be undefined)
  }
});