trying to fix issue #19 (#20)

jckleiner · Sep 12, 2024 · 47baf4d · 47baf4d
1 parent 1c1febf
commit 47baf4d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/com/greydev/notionbackup/NotionClient.java b/src/main/java/com/greydev/notionbackup/NotionClient.java
@@ -201,7 +201,7 @@ private Optional<String> getDownloadLink(String taskId) throws IOException, Inte
 				.uri(URI.create(GET_TASKS_ENDPOINT))
 				.header("Cookie", TOKEN_V2 + "=" + notionTokenV2)
 				.header("Content-Type", "application/json")
-				.timeout(Duration.ofSeconds(10))
+				.timeout(Duration.ofSeconds(20))
 				.POST(HttpRequest.BodyPublishers.ofString(postBody))
 				.build();
 
@@ -210,12 +210,12 @@ private Optional<String> getDownloadLink(String taskId) throws IOException, Inte
 
 			Results results = objectMapper.readValue(response.body(), Results.class);
 
-			if (results.getResults().isEmpty()) {
+			if (results == null || results.getResults() == null || results.getResults().isEmpty()) {
 				sleep(6000);
 				continue;
 			}
 
-			Result result = results.getResults().stream().findFirst().get();
+			Result result = results.getResults().get(0);
 
 			if (result.isFailure()) {
 				log.info("Notion API workspace export returned a 'failure' state. Reason: {}", result.getError());