Replace notion-sdk-jvm with Spring Framework's HTTP Service Client

Signed-off-by: Stefano Cordio <[email protected]>

Author: scordio

spring-batch-notion/pom.xml

@@ -61,12 +61,16 @@
     </dependency>
     <dependency>
       <groupId>org.springframework</groupId>
-      <artifactId>spring-beans</artifactId>
+      <artifactId>spring-web</artifactId>
     </dependency>
     <dependency>
       <groupId>org.springframework.batch</groupId>
       <artifactId>spring-batch-infrastructure</artifactId>
     </dependency>
+    <dependency>
+      <groupId>tools.jackson.core</groupId>
+      <artifactId>jackson-databind</artifactId>
+    </dependency>
     <!-- Test -->
     <dependency>
       <groupId>com.h2database</groupId>

spring-batch-notion/src/main/java/org/springframework/batch/extensions/notion/NotionDatabaseItemReader.java

@@ -15,22 +15,19 @@
  */
 package org.springframework.batch.extensions.notion;
 
-import notion.api.v1.NotionClient;
-import notion.api.v1.http.JavaNetHttpClient;
-import notion.api.v1.logging.Slf4jLogger;
-import notion.api.v1.model.databases.QueryResults;
-import notion.api.v1.model.databases.query.filter.QueryTopLevelFilter;
-import notion.api.v1.model.databases.query.sort.QuerySort;
-import notion.api.v1.model.pages.Page;
-import notion.api.v1.model.pages.PageProperty;
-import notion.api.v1.model.pages.PageProperty.RichText;
-import notion.api.v1.request.databases.QueryDatabaseRequest;
 import org.jspecify.annotations.Nullable;
+import org.springframework.batch.extensions.notion.PageProperty.RichTextProperty;
+import org.springframework.batch.extensions.notion.PageProperty.TitleProperty;
 import org.springframework.batch.extensions.notion.mapping.PropertyMapper;
 import org.springframework.batch.infrastructure.item.ExecutionContext;
 import org.springframework.batch.infrastructure.item.ItemReader;
 import org.springframework.batch.infrastructure.item.data.AbstractPaginatedDataItemReader;
+import org.springframework.http.HttpHeaders;
 import org.springframework.util.Assert;
+import org.springframework.web.client.ApiVersionInserter;
+import org.springframework.web.client.RestClient;
+import org.springframework.web.client.support.RestClientAdapter;
+import org.springframework.web.service.invoker.HttpServiceProxyFactory;
 
 import java.util.Collections;
 import java.util.Iterator;
@@ -39,7 +36,6 @@
 import java.util.Map.Entry;
 import java.util.Objects;
 import java.util.stream.Collectors;
-import java.util.stream.Stream;
 
 /**
  * Restartable {@link ItemReader} that reads entries from a Notion database via a paging
@@ -71,11 +67,11 @@ public class NotionDatabaseItemReader<T> extends AbstractPaginatedDataItemReader
 
 	private String baseUrl = DEFAULT_BASE_URL;
 
-	private @Nullable QueryTopLevelFilter filter;
+	private @Nullable Filter filter;
 
-	private @Nullable List<QuerySort> sorts;
+	private Sort[] sorts = new Sort[0];
 
-	private @Nullable NotionClient client;
+	private @Nullable NotionDatabaseService service;
 
 	private boolean hasMore;
 
@@ -117,7 +113,7 @@ public void setBaseUrl(String baseUrl) {
 	 * @see Filter#where(Filter)
 	 */
 	public void setFilter(Filter filter) {
-		this.filter = filter.toQueryTopLevelFilter();
+		this.filter = filter;
 	}
 
 	/**
@@ -130,7 +126,7 @@ public void setFilter(Filter filter) {
 	 * @see Sort#by(Sort.Timestamp)
 	 */
 	public void setSorts(Sort... sorts) {
-		this.sorts = Stream.of(sorts).map(Sort::toQuerySort).toList();
+		this.sorts = sorts;
 	}
 
 	/**
@@ -151,10 +147,15 @@ public void setPageSize(int pageSize) {
 	 */
 	@Override
 	protected void doOpen() {
-		client = new NotionClient(token);
-		client.setHttpClient(new JavaNetHttpClient());
-		client.setLogger(new Slf4jLogger());
-		client.setBaseUrl(baseUrl);
+		RestClient restClient = RestClient.builder()
+			.baseUrl(baseUrl)
+			.apiVersionInserter(ApiVersionInserter.useHeader("Notion-Version"))
+			.defaultHeader(HttpHeaders.AUTHORIZATION, "Bearer " + token)
+			.build();
+
+		RestClientAdapter adapter = RestClientAdapter.create(restClient);
+		HttpServiceProxyFactory factory = HttpServiceProxyFactory.builderFor(adapter).build();
+		service = factory.createClient(NotionDatabaseService.class);
 
 		hasMore = true;
 	}
@@ -168,53 +169,47 @@ protected Iterator<T> doPageRead() {
 			return Collections.emptyIterator();
 		}
 
-		QueryDatabaseRequest request = new QueryDatabaseRequest(databaseId);
-		request.setFilter(filter);
-		request.setSorts(sorts);
-		request.setStartCursor(nextCursor);
-		request.setPageSize(pageSize);
+		QueryRequest request = new QueryRequest(pageSize, nextCursor, filter, sorts);
 
 		@SuppressWarnings("DataFlowIssue")
-		QueryResults queryResults = client.queryDatabase(request);
+		QueryResult result = service.query(databaseId, request);
 
-		hasMore = queryResults.getHasMore();
-		nextCursor = queryResults.getNextCursor();
+		hasMore = result.hasMore();
+		nextCursor = result.nextCursor();
 
-		return queryResults.getResults()
+		return result.results()
 			.stream()
 			.map(NotionDatabaseItemReader::getProperties)
 			.map(propertyMapper::map)
 			.iterator();
 	}
 
-	private static Map<String, String> getProperties(Page element) {
-		return element.getProperties()
+	private static Map<String, String> getProperties(Page page) {
+		return page.properties()
 			.entrySet()
 			.stream()
 			.collect(Collectors.toUnmodifiableMap(Entry::getKey, entry -> getPropertyValue(entry.getValue())));
 	}
 
 	private static String getPropertyValue(PageProperty property) {
-		return switch (property.getType()) {
-			case RichText -> getPlainText(property.getRichText());
-			case Title -> getPlainText(property.getTitle());
-			default -> throw new IllegalArgumentException("Unsupported type: " + property.getType());
-		};
+		if (property instanceof RichTextProperty p) {
+			return getPlainText(p.richText());
+		}
+		if (property instanceof TitleProperty p) {
+			return getPlainText(p.title());
+		}
+		throw new IllegalArgumentException("Unsupported type: " + property.getClass());
 	}
 
 	private static String getPlainText(List<RichText> texts) {
-		return texts.isEmpty() ? "" : texts.get(0).getPlainText();
+		return texts.isEmpty() ? "" : texts.get(0).plainText();
 	}
 
 	/**
 	 * {@inheritDoc}
 	 */
-	@SuppressWarnings("DataFlowIssue")
 	@Override
 	protected void doClose() {
-		client.close();
-		client = null;
-
 		hasMore = false;
 	}
 

spring-batch-notion/src/main/java/org/springframework/batch/extensions/notion/NotionDatabaseService.java

@@ -0,0 +1,30 @@
+/*
+ * Copyright 2024-2025 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.batch.extensions.notion;
+
+import org.springframework.http.MediaType;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.service.annotation.HttpExchange;
+import org.springframework.web.service.annotation.PostExchange;
+
+@HttpExchange(url = "/databases", version = "2022-06-28", accept = MediaType.APPLICATION_JSON_VALUE)
+interface NotionDatabaseService {
+
+	@PostExchange("/{databaseId}/query")
+	QueryResult query(@PathVariable String databaseId, @RequestBody QueryRequest request);
+
+}

spring-batch-notion/src/main/java/org/springframework/batch/extensions/notion/Page.java

@@ -0,0 +1,6 @@
+package org.springframework.batch.extensions.notion;
+
+import java.util.Map;
+
+record Page(Map<String, PageProperty> properties) {
+}

spring-batch-notion/src/main/java/org/springframework/batch/extensions/notion/PageProperty.java

@@ -0,0 +1,24 @@
+package org.springframework.batch.extensions.notion;
+
+import com.fasterxml.jackson.annotation.JsonSubTypes;
+import com.fasterxml.jackson.annotation.JsonTypeInfo;
+import tools.jackson.databind.PropertyNamingStrategies;
+import tools.jackson.databind.annotation.JsonNaming;
+
+import java.util.List;
+
+@JsonTypeInfo(use = JsonTypeInfo.Id.NAME, property = "type")
+@JsonSubTypes({ //
+		@JsonSubTypes.Type(name = "rich_text", value = PageProperty.RichTextProperty.class),
+		@JsonSubTypes.Type(name = "title", value = PageProperty.TitleProperty.class) //
+})
+interface PageProperty {
+
+	@JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
+	record RichTextProperty(List<RichText> richText) implements PageProperty {
+	}
+
+	record TitleProperty(List<RichText> title) implements PageProperty {
+	}
+
+}

spring-batch-notion/src/main/java/org/springframework/batch/extensions/notion/QueryRequest.java

@@ -0,0 +1,14 @@
+package org.springframework.batch.extensions.notion;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.annotation.JsonInclude.Include;
+import org.jspecify.annotations.Nullable;
+import tools.jackson.databind.PropertyNamingStrategies.SnakeCaseStrategy;
+import tools.jackson.databind.annotation.JsonNaming;
+
+import java.util.List;
+
+@JsonNaming(SnakeCaseStrategy.class)
+@JsonInclude(Include.NON_EMPTY)
+record QueryRequest(int pageSize, @Nullable String startCursor, @Nullable Filter filter, Sort... sorts) {
+}

spring-batch-notion/src/main/java/org/springframework/batch/extensions/notion/QueryResult.java

@@ -0,0 +1,11 @@
+package org.springframework.batch.extensions.notion;
+
+import tools.jackson.databind.PropertyNamingStrategies.SnakeCaseStrategy;
+import tools.jackson.databind.annotation.JsonNaming;
+
+import java.util.List;
+
+@JsonNaming(SnakeCaseStrategy.class)
+record QueryResult(List<Page> results, String nextCursor, boolean hasMore) {
+
+}

spring-batch-notion/src/main/java/org/springframework/batch/extensions/notion/RichText.java

@@ -0,0 +1,8 @@
+package org.springframework.batch.extensions.notion;
+
+import tools.jackson.databind.PropertyNamingStrategies;
+import tools.jackson.databind.annotation.JsonNaming;
+
+@JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
+record RichText(String plainText) {
+}