From ae979912df9ddf06778d5df529510be620dd2123 Mon Sep 17 00:00:00 2001
From: "Michael B. Klein" <mbklein@gmail.com>
Date: Tue, 28 Jan 2025 23:51:22 +0000
Subject: [PATCH] Add retries to HTTP timeouts in Meadow.Search.HTTP and
 Meadow.Utils.AWS

---
 app/lib/meadow/search/http.ex | 16 +++++++++++-----
 app/lib/meadow/utils/aws.ex   | 28 ++++++++++++++++++++++++++--
 2 files changed, 37 insertions(+), 7 deletions(-)

diff --git a/app/lib/meadow/search/http.ex b/app/lib/meadow/search/http.ex
index b0ea9dd6e..1a89991b6 100644
--- a/app/lib/meadow/search/http.ex
+++ b/app/lib/meadow/search/http.ex
@@ -58,14 +58,14 @@ defmodule Meadow.Search.HTTP do
             atoms: [:retry],
             rescue_only: [] do
         case ElastixHTTP.request(method, url, body, headers, options) do
-          {:ok, %{status_code: status} = response} when status in 200..399 ->
-            {:ok, response}
+          {:ok, %{status_code: status} = response} when status in [429, 503, 504] ->
+            {:retry, response}
 
-          {:ok, %{status_code: 404} = response} ->
+          {:ok, response} ->
             {:ok, response}
 
-          {:ok, %{status_code: 429} = response} ->
-            {:retry, response}
+          {:error, %HTTPoison.Error{reason: :timeout} = error} ->
+            {:retry, error}
 
           {:error, error} ->
             {:error, error}
@@ -80,6 +80,12 @@ defmodule Meadow.Search.HTTP do
         result ->
           result |> maybe_report(%{method: method, url: url, body: body})
       else
+        {:retry, %HTTPoison.Response{} = response} ->
+          response |> maybe_report(%{method: method, url: url, body: body})
+
+        {:retry, reason} ->
+          {:error, reason} |> maybe_report(%{method: method, url: url, body: body})
+
         error ->
           error |> maybe_report(%{method: method, url: url, body: body})
       end
diff --git a/app/lib/meadow/utils/aws.ex b/app/lib/meadow/utils/aws.ex
index 0ca267589..6d3252a3d 100644
--- a/app/lib/meadow/utils/aws.ex
+++ b/app/lib/meadow/utils/aws.ex
@@ -10,6 +10,8 @@ defmodule Meadow.Utils.AWS do
   alias Meadow.Utils.AWS.MultipartCopy
   alias Meadow.Utils.Pairtree
 
+  use Retry
+
   import SweetXml, only: [sigil_x: 2]
 
   require Logger
@@ -18,14 +20,36 @@ defmodule Meadow.Utils.AWS do
   Drop-in replacement for ExAws.request/2 that reports errors to Honeybadger
   """
   def request(op, config_overrides \\ []) do
-    ExAws.request(op, config_overrides) |> handle_response()
+    retry with: exponential_backoff() |> randomize() |> cap(1_000) |> Stream.take(10),
+          atoms: [:retry],
+          rescue_only: [] do
+      case ExAws.request(op, config_overrides) |> handle_response() do
+        :timeout ->
+          {:retry, :timeout}
+
+        {:error, {:http_error, status, _}} = response when status in [429, 503, 504] ->
+          {:retry, response}
+
+        {:ok, result} ->
+          {:ok, result}
+
+        error ->
+          error
+      end
+    after
+      {:ok, result} -> {:ok, result}
+      {:error, error} -> {:error, error}
+    else
+      {:retry, error} -> error
+      error -> error
+    end
   end
 
   @doc """
   Drop-in replacement for ExAws.request!/2 that reports errors to Honeybadger
   """
   def request!(op, config_overrides \\ []) do
-    case ExAws.request(op, config_overrides) |> handle_response() do
+    case request(op, config_overrides) do
       {:ok, result} ->
         result