Added a case when the stream is stopped like with Enum.take

CoreCareinc · Apr 4, 2024 · c6746ef · c6746ef
1 parent 2da7a47
commit c6746ef
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/elasticsearch_ex/stream.ex b/lib/elasticsearch_ex/stream.ex
@@ -122,6 +122,9 @@ defmodule ElasticsearchEx.Stream do
     end
   end
 
+  @spec after_fun(acc()) :: :ok
+  defp after_fun({pit, _search_after}), do: after_fun(pit)
+
   @spec after_fun(pit()) :: :ok
   defp after_fun(%{id: pit_id}) do
     case SearchApi.close_pit(pit_id) do