Catch not authorized exception in appropriate places

RADAR-base · Sep 8, 2021 · 99bfb07 · 99bfb07
1 parent dc7f9b0
commit 99bfb07
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/kafka-connect-rest-source/src/main/java/org/radarbase/connect/rest/RestSourceTask.java b/kafka-connect-rest-source/src/main/java/org/radarbase/connect/rest/RestSourceTask.java
@@ -26,6 +26,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
+import javax.ws.rs.NotAuthorizedException;
 import org.apache.kafka.connect.errors.ConnectException;
 import org.apache.kafka.connect.source.SourceRecord;
 import org.apache.kafka.connect.source.SourceTask;
@@ -86,7 +87,7 @@ public List<SourceRecord> poll() throws InterruptedException {
         try {
           requests = request.handleRequest()
               .collect(Collectors.toList());
-        } catch (IOException ex) {
+        } catch (IOException | NotAuthorizedException ex) {
           logger.warn("Failed to make request: {}", ex.toString());
         }
       }

diff --git a/kafka-connect-rest-source/src/main/java/org/radarbase/connect/rest/request/RestRequest.java b/kafka-connect-rest-source/src/main/java/org/radarbase/connect/rest/request/RestRequest.java
@@ -22,6 +22,7 @@
 import java.util.Map;
 import java.util.function.Predicate;
 import java.util.stream.Stream;
+import javax.ws.rs.NotAuthorizedException;
 import okhttp3.Headers;
 import okhttp3.OkHttpClient;
 import okhttp3.Request;
@@ -95,7 +96,7 @@ public Stream<SourceRecord> handleRequest() throws IOException {
       headers = response.headers();
       ResponseBody body = response.body();
       data = body != null ? body.bytes() : null;
-    } catch (IOException ex) {
+    } catch (IOException | NotAuthorizedException ex) {
       route.requestFailed(this, null);
       throw ex;
     }