pass a function to a custom react hook

Question

Previously, I was using a graphql query like this. The data is returned from the query and I use setShowFlatList with the data returned:

  const [loadUsers, { data }] = useUsersLazyQuery({
    onCompleted: () => {
      setShowFlatList(data);
    },
});

Now I am creating a custom react hook where I use this graphql query. It looks like this:

export const useLoadUsers = (onCompleted: any) => {
  const [usersQuery, { data }] = useUsersLazyQuery({
    onCompleted: () => {
      if(onCompleted){
        onCompleted();
      }
    },
    onError: onLoadUserError,
    fetchPolicy: 'network-only',
  });

const loadUsers = async (
phoneNumber: number,
) => {
  const data = await usersQuery({
    variables: {
      where: {
        OR: [
          { phoneNumber: newPhoneNumber }
        ],
      },
    },
    });
  return data;
};
return loadUsers;
};

But I cannot figure out how to pass the setShowFlatList function to the onCompleted of the hook, such that I can still use data as its parameter from within the hook.

Answer 1

EDIT: The onCompleted should also have an parameter data, so you can use that one.

export const useLoadUsers = (onCompleted: any) => {
  const [usersQuery, { data }] = useUsersLazyQuery({
    onCompleted: (data) => {
      if(onCompleted){
        onCompleted(data);
      }
    },
    onError: onLoadUserError,
    fetchPolicy: 'network-only',
  });

const loadUsers = async (
phoneNumber: number,
) => {
  const data = await usersQuery({
    variables: {
      where: {
        OR: [
          { phoneNumber: newPhoneNumber }
        ],
      },
    },
    });
  return data;
};
return loadUsers;
};

Answer 2

export const useEditLocationName = (callback) => {
  return (params) => async(favouritePlaceId, customisedName) => {
    const [updateFavouritePlace] = useUpdateFavouritePlaceMutation({
      onCompleted: () => {
        if(callback){
          callback(params)
        }
        Alert.alert('Name aktualisiert');
      },
      onError: () => {
        Alert.alert('Oops, etwas ist schiefgelaufen');
      },
    });
    updateFavouritePlace({
      variables: {
        favouritePlaceId: favouritePlaceId,
        input: { customisedName: customisedName },
      },
    });
    return null;
  }

const editLocationName = useEditLocationName(setShowFlatList);
...
  const handleSubmitForm = (
    values: FormValues,
  ) => {
    editLocationName(params)(route.params.favouritePlaceId, values.locationName)
  };


/*
export const useEditLocationName = (callback) => {
  const [updateFavouritePlace] = useUpdateFavouritePlaceMutation({
    onCompleted: () => {
      if(callback){
        callback()
      }
      Alert.alert('Name aktualisiert');
    },
    onError: () => {
      Alert.alert('Oops, etwas ist schiefgelaufen');
    },
  });

  const editLocationName = async (
    favouritePlaceId: number,
    customisedName: string,
  ) => {
    updateFavouritePlace({
      variables: {
        favouritePlaceId: favouritePlaceId,
        input: { customisedName: customisedName },
      },
    });
    return null;
  };

  return editLocationName;
};
*/
....