Adjusting how user agent is calculated

Author: Paultagoras

clickhouse-client/src/main/java/com/clickhouse/client/config/ClickHouseClientOption.java

@@ -480,9 +480,11 @@ public enum ClickHouseClientOption implements ClickHouseOption {
             ver = parts[3];
             PRODUCT_REVISION = ver.substring(0, ver.length() - 1);
         } else { // perhaps try harder by checking version from pom.xml?
-            PRODUCT_VERSION = UNKNOWN;
+            PRODUCT_VERSION = "";
             PRODUCT_REVISION = UNKNOWN;
         }
+
+
         CLIENT_OS_INFO = new StringBuilder().append(getSystemConfig("os.name", "O/S")).append('/')
                 .append(getSystemConfig("os.version", UNKNOWN)).toString();
         String javaVersion = System.getProperty("java.vendor.version");
@@ -510,15 +512,18 @@ public enum ClickHouseClientOption implements ClickHouseOption {
      * @param additionalProperty additional property if any
      * @return non-empty user-agent
      */
-    public static final String buildUserAgent(String productName, String additionalProperty) {
-        productName = productName == null || productName.isEmpty() ? (String) PRODUCT_NAME.getEffectiveDefaultValue()
-                : productName.trim();
-        StringBuilder builder = new StringBuilder(productName).append('/').append(PRODUCT_VERSION).append(" (")
-                .append(CLIENT_OS_INFO).append("; ").append(CLIENT_JVM_INFO);
+    public static String buildUserAgent(String productName, String additionalProperty) {
+        productName = productName == null || productName.isEmpty() ? (String) PRODUCT_NAME.getEffectiveDefaultValue() : productName.trim();
+        StringBuilder builder = new StringBuilder(productName).append(PRODUCT_VERSION.isEmpty() ? "" : "/" + PRODUCT_VERSION);
+
+        if (!String.valueOf(PRODUCT_NAME.getDefaultValue()).equals(productName)) {//Append if someone changed the original value
+            builder.append(" ").append(PRODUCT_NAME.getDefaultValue()).append(PRODUCT_VERSION.isEmpty() ? "" : "/" + PRODUCT_VERSION);
+        }
+        builder.append(" (").append(CLIENT_JVM_INFO);
         if (additionalProperty != null && !additionalProperty.isEmpty()) {
             builder.append("; ").append(additionalProperty.trim());
         }
-        return builder.append("; rv:").append(PRODUCT_REVISION).append(')').toString();
+        return builder.append(")").toString();
     }
 
     /**

clickhouse-http-client/src/main/java/com/clickhouse/client/http/ClickHouseHttpConnection.java

@@ -420,14 +420,14 @@ protected String getDefaultUserAgent() {
     protected final String getUserAgent() {
         final ClickHouseConfig c = config;
         String name = c.getClientName();
+        String userAgent = getDefaultUserAgent();
+
         if (!ClickHouseClientOption.CLIENT_NAME.getDefaultValue().equals(name)) {
-            return name;
+            return name + " " + userAgent;
         }
 
-        String userAgent = getDefaultUserAgent();
         name = c.getProductName();
-        return ClickHouseClientOption.PRODUCT_NAME.getDefaultValue().equals(name) ? userAgent
-                : new StringBuilder(name).append(userAgent.substring(userAgent.indexOf('/'))).toString();
+        return ClickHouseClientOption.PRODUCT_NAME.getDefaultValue().equals(name) ? userAgent : name + " " + userAgent;
     }
 
     /**

clickhouse-http-client/src/test/java/com/clickhouse/client/http/ClickHouseHttpClientTest.java

@@ -193,7 +193,8 @@ public void testUserAgent() throws Exception {
              ClickHouseResponse response = newRequest(client, server)
                      .query("select http_user_agent from system.query_log where query='select ''" + uuid + "'''")
                      .executeAndWait()) {
-            Assert.assertEquals(response.firstRecord().getValue(0).asString(), "MyCustomClient");
+            String result = response.firstRecord().getValue(0).asString();
+            Assert.assertTrue(result.startsWith("MyCustomClient"));
         }
     }