feat: get server version

Author: NguyenHoangSon96

src/main/java/com/influxdb/v3/client/InfluxDBClient.java

@@ -450,12 +450,12 @@ Stream<VectorSchemaRoot> queryBatches(@Nonnull final String query,
                                           @Nonnull final QueryOptions options);
 
     /**
-     * Sends a ping and returns the InfluxDB version.
+     * Get InfluxDB server version.
      *
-     * @return a non-null string representing the response to the ping request
+     * @return a string representing the server version. Can be null.
      */
     @Nonnull
-    String ping();
+    String getServerVersion();
 
     /**
      * Creates a new instance of the {@link InfluxDBClient} for interacting with an InfluxDB server, simplifying

src/main/java/com/influxdb/v3/client/internal/InfluxDBClientImpl.java

@@ -292,8 +292,8 @@ public Stream<VectorSchemaRoot> queryBatches(@Nonnull final String query,
 
     @Nonnull
     @Override
-    public String ping() {
-        return this.restClient.ping();
+    public String getServerVersion() {
+        return this.restClient.getServerVersion();
     }
 
     @Override

src/main/java/com/influxdb/v3/client/internal/RestClient.java

@@ -141,7 +141,7 @@ public void checkServerTrusted(
     }
 
     @Nonnull
-    public String ping() throws InfluxDBApiException {
+    public String getServerVersion() throws InfluxDBApiException {
         HttpRequest request = HttpRequest.newBuilder()
                 .uri(URI.create(this.baseUrl + "ping"))
                 .GET()
@@ -151,7 +151,7 @@ public String ping() throws InfluxDBApiException {
         String influxdbVersion;
         try {
             response = client.send(request, HttpResponse.BodyHandlers.ofString());
-            influxdbVersion = response.headers().firstValue("x-influxdb-version").orElseThrow();
+            influxdbVersion = extractServerVersion(response);
         } catch (Exception e) {
             throw new InfluxDBApiException(e);
         }
@@ -160,6 +160,19 @@ public String ping() throws InfluxDBApiException {
         return influxdbVersion;
     }
 
+    private String extractServerVersion(HttpResponse<String> response) {
+        return response.headers()
+                .firstValue("x-influxdb-version")
+                .orElseGet(() -> {
+                    try {
+                        JsonNode jsonNode = objectMapper.readTree(response.body());
+                        return jsonNode.get("version").asText();
+                    } catch (JsonProcessingException e) {
+                        throw new InfluxDBApiException(e);
+                    }
+                });
+    }
+
     void request(@Nonnull final String path,
                  @Nonnull final HttpMethod method,
                  @Nullable final byte[] data,

src/test/java/com/influxdb/v3/client/integration/E2ETest.java

@@ -388,14 +388,14 @@ public void testQueryRowsExceptionCases() throws Exception {
     @EnabledIfEnvironmentVariable(named = "TESTING_INFLUXDB_TOKEN", matches = ".*")
     @EnabledIfEnvironmentVariable(named = "TESTING_INFLUXDB_DATABASE", matches = ".*")
     @Test
-    public void testPing() throws Exception {
+    public void testGetServerVersion() throws Exception {
         try (InfluxDBClient client = InfluxDBClient.getInstance(
                 System.getenv("TESTING_INFLUXDB_URL"),
                 System.getenv("TESTING_INFLUXDB_TOKEN").toCharArray(),
                 System.getenv("TESTING_INFLUXDB_DATABASE"),
                 null)) {
 
-            Assertions.assertThat(client.ping()).isNotEmpty();
+            Assertions.assertThat(client.getServerVersion()).isNotEmpty();
         }
     }
 

src/test/java/com/influxdb/v3/client/internal/RestClientTest.java

@@ -478,28 +478,40 @@ public void errorHttpExceptionThrown() {
     }
 
     @Test
-    public void pingReturnsVersionWhenSuccessful() throws Exception {
+    public void getServerVersionV2Successful() throws Exception {
         String influxDBVersion = "v2.1.0";
         mockServer.enqueue(createResponse(200).setHeader("x-influxdb-version", influxDBVersion));
 
         restClient = new RestClient(new ClientConfig.Builder()
                 .host(baseURL)
                 .build());
+        String version = restClient.getServerVersion();
 
-        String version = restClient.ping();
+        Assertions.assertThat(version).isEqualTo(influxDBVersion);
+    }
+
+    @Test
+    public void getServerVersionV3Successful() throws Exception {
+        String influxDBVersion = "3.0.0";
+        mockServer.enqueue(createResponse(200).setBody("{\"version\":\"" + influxDBVersion + "\"}"));
+
+        restClient = new RestClient(new ClientConfig.Builder()
+                .host(baseURL)
+                .build());
+        String version = restClient.getServerVersion();
 
         Assertions.assertThat(version).isEqualTo(influxDBVersion);
     }
 
     @Test
-    public void pingThrowsExceptionOnFailure() {
+    public void getServerVersionThrowsExceptionOnFailure() {
         mockServer.enqueue(createResponse(500).setBody("{\"message\":\"internal server error\"}"));
 
         restClient = new RestClient(new ClientConfig.Builder()
                 .host(baseURL)
                 .build());
 
-        Assertions.assertThatThrownBy(() -> restClient.ping())
+        Assertions.assertThatThrownBy(() -> restClient.getServerVersion())
                 .isInstanceOf(InfluxDBApiException.class);
     }
 }