Merge pull request vespa-engine#30411 from vespa-engine/freva/json

Move Json to vespajlib

container-core/src/main/java/com/yahoo/restapi/RestApiMappers.java

@@ -1,6 +1,7 @@
 // Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.restapi;
 
+import ai.vespa.json.Json;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.yahoo.container.jdisc.HttpResponse;

http-client/src/main/java/ai/vespa/hosted/client/AbstractHttpClient.java

@@ -4,6 +4,8 @@
 import ai.vespa.http.HttpURL;
 import ai.vespa.http.HttpURL.Path;
 import ai.vespa.http.HttpURL.Query;
+import ai.vespa.json.Json;
+import com.yahoo.slime.SlimeUtils;
 import com.yahoo.time.TimeBudget;
 import org.apache.hc.client5.http.config.RequestConfig;
 import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;
@@ -281,6 +283,11 @@ public <T> T read(Function<byte[], T> mapper) {
             });
         }
 
+        @Override
+        public Json readAsJson() {
+            return read(bytes -> Json.of(SlimeUtils.jsonToSlime(bytes)));
+        }
+
         @Override
         public void discard() throws UncheckedIOException, ResponseException {
             handle((response, __) -> {

http-client/src/main/java/ai/vespa/hosted/client/HttpClient.java

@@ -4,6 +4,7 @@
 import ai.vespa.http.HttpURL;
 import ai.vespa.http.HttpURL.Path;
 import ai.vespa.http.HttpURL.Query;
+import ai.vespa.json.Json;
 import com.yahoo.time.TimeBudget;
 import org.apache.hc.client5.http.config.RequestConfig;
 import org.apache.hc.core5.http.ClassicHttpRequest;
@@ -140,6 +141,9 @@ default RequestBuilder parameters(String... pairs) {
         /** Reads and maps the response, or throws if unsuccessful. */
         <T> T read(Function<byte[], T> mapper);
 
+        /** Reads the response as a {@link Json}, or throws if unsuccessful. */
+        Json readAsJson();
+
         /** Discards the response, but throws if unsuccessful. */
         void discard();
 

vespajlib/src/main/java/ai/vespa/json/InvalidJsonException.java

@@ -0,0 +1,11 @@
+// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package ai.vespa.json;
+
+/**
+ * @author freva
+ */
+public class InvalidJsonException extends IllegalArgumentException {
+    public InvalidJsonException(String message) {
+        super(message);
+    }
+}

container-core/src/main/java/com/yahoo/restapi/Json.java → vespajlib/src/main/java/ai/vespa/json/Json.java

@@ -1,4 +1,4 @@
-package com.yahoo.restapi;
+package ai.vespa.json;
 
 import com.yahoo.slime.Cursor;
 import com.yahoo.slime.Inspector;
@@ -26,7 +26,7 @@
 import static com.yahoo.slime.Type.STRING;
 
 /**
- * A {@link Slime} wrapper that throws {@link RestApiException.BadRequest} on missing members or invalid types.
+ * A {@link Slime} wrapper that throws {@link InvalidJsonException} on missing members or invalid types.
  *
  * @author bjorncs
  */
@@ -146,16 +146,16 @@ private void requireType(Type... types) {
 
     private void requirePresent() { if (isMissing()) throw createMissingMemberException(); }
 
-    private RestApiException.BadRequest createInvalidTypeException(Type... expected) {
+    private InvalidJsonException createInvalidTypeException(Type... expected) {
         var expectedTypesString = Arrays.stream(expected).map(this::toString).collect(Collectors.joining("' or '", "'", "'"));
         var pathString = path.isEmpty() ? "JSON" : "JSON member '%s'".formatted(path);
-        return new RestApiException.BadRequest(
+        return new InvalidJsonException(
                 "Expected %s to be a %s but got '%s'"
                         .formatted(pathString, expectedTypesString, toString(inspector.type())));
     }
 
-    private RestApiException.BadRequest createMissingMemberException() {
-        return new RestApiException.BadRequest(path.isEmpty() ? "Missing JSON" : "Missing JSON member '%s'".formatted(path));
+    private InvalidJsonException createMissingMemberException() {
+        return new InvalidJsonException(path.isEmpty() ? "Missing JSON" : "Missing JSON member '%s'".formatted(path));
     }
 
     private String toString(Type type) {

vespajlib/src/main/java/ai/vespa/json/package-info.java

@@ -0,0 +1,5 @@
+// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+@ExportPackage
+package ai.vespa.json;
+
+import com.yahoo.osgi.annotation.ExportPackage;

container-core/src/test/java/com/yahoo/restapi/JsonTest.java → vespajlib/src/test/java/ai/vespa/json/JsonTest.java

@@ -1,8 +1,10 @@
-package com.yahoo.restapi;
+package ai.vespa.json;
 
 import org.junit.jupiter.api.Test;
 
-import static org.junit.jupiter.api.Assertions.*;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 /**
  * @author bjorncs
@@ -65,16 +67,16 @@ void throws_on_missing_and_invalid_members() {
                 """;
         var json = Json.of(text);
 
-        var exception = assertThrows(RestApiException.BadRequest.class, () -> json.f("unknown").asString());
+        var exception = assertThrows(InvalidJsonException.class, () -> json.f("unknown").asString());
         assertEquals("Missing JSON member 'unknown'", exception.getMessage());
 
-        exception = assertThrows(RestApiException.BadRequest.class, () -> json.a(0));
+        exception = assertThrows(InvalidJsonException.class, () -> json.a(0));
         assertEquals("Expected JSON to be a 'array' but got 'object'", exception.getMessage());
 
-        exception = assertThrows(RestApiException.BadRequest.class, () -> json.f("string").f("unknown"));
+        exception = assertThrows(InvalidJsonException.class, () -> json.f("string").f("unknown"));
         assertEquals("Expected JSON member 'string' to be a 'object' but got 'string'", exception.getMessage());
 
-        exception = assertThrows(RestApiException.BadRequest.class, () -> json.f("string").asLong());
+        exception = assertThrows(InvalidJsonException.class, () -> json.f("string").asLong());
         assertEquals("Expected JSON member 'string' to be a 'integer' or 'float' but got 'string'", exception.getMessage());
     }