dbeaver/dbeaver#23361 Proper user agent

dbeaver · Nov 4, 2024 · 801817d · 801817d
1 parent 9da3438
commit 801817d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/...ver.jdbc.driver.libsql/src/main/java/com/dbeaver/jdbc/driver/libsql/LibSqlConnection.java b/...ver.jdbc.driver.libsql/src/main/java/com/dbeaver/jdbc/driver/libsql/LibSqlConnection.java
@@ -61,6 +61,10 @@ public LibSqlConnection(
     }
 
     public LibSqlClient getClient() {
+        String applicationName = getClientApplicationName();
+        if (!CommonUtils.isEmpty(applicationName)) {
+            client.setUserAgent(applicationName);
+        }
         return client;
     }
 

diff --git a/....jdbc.driver.libsql/src/main/java/com/dbeaver/jdbc/driver/libsql/client/LibSqlClient.java b/....jdbc.driver.libsql/src/main/java/com/dbeaver/jdbc/driver/libsql/client/LibSqlClient.java
@@ -91,7 +91,7 @@ public LibSqlExecutionResult[] executeBatch(
             final HttpRequest.Builder builder = HttpRequest.newBuilder()
                 .uri(url.toURI())
                 .version(HttpClient.Version.HTTP_1_1)
-                //.header("Content-Type", "application/json")
+                .header("Content-Type", "application/json")
                 .header("User-Agent", userAgent)
                 .POST(HttpRequest.BodyPublishers.ofString(requestBuffer.toString()));
             if (authToken != null) {
@@ -149,7 +149,9 @@ public HttpURLConnection openSimpleConnection(String endpoint) throws IOExceptio
             baseURL += "/";
         }
         baseURL += endpoint;
-        return (HttpURLConnection) new URL(baseURL).openConnection();
+        HttpURLConnection connection = (HttpURLConnection) new URL(baseURL).openConnection();
+        connection.setRequestProperty("User-Agent", userAgent);
+        return connection;
     }
 
     private void executeQuery(