SNOW-931105: Add createSessionlessTelemetry(CloseableHttpClient, Stri…

…ng) method (#1527)

* add createSessionlessTelemetry(CloseableHttpClient, String) method

* fix style

* add comment to javadoc

src/main/java/net/snowflake/client/jdbc/telemetry/TelemetryClient.java

@@ -184,11 +184,26 @@ public static Telemetry createTelemetry(SFSession session, int flushSize) {
     return new TelemetryClient(session, flushSize);
   }
 
+  /**
+   * Initialize the sessionless telemetry connector using KEYPAIR_JWT as the default auth type
+   *
+   * @param httpClient client object used to communicate with other machine
+   * @param serverUrl server url
+   * @return a telemetry connector
+   */
+  public static Telemetry createSessionlessTelemetry(
+      CloseableHttpClient httpClient, String serverUrl) {
+    // By default, use KEYPAIR_JWT as the auth type
+    return createSessionlessTelemetry(
+        httpClient, serverUrl, "KEYPAIR_JWT", DEFAULT_FORCE_FLUSH_SIZE);
+  }
+
   /**
    * Initialize the sessionless telemetry connector
    *
    * @param httpClient client object used to communicate with other machine
    * @param serverUrl server url
+   * @param authType authorization type for sessionless telemetry
    * @return a telemetry connector
    */
   public static Telemetry createSessionlessTelemetry(

src/test/java/net/snowflake/client/jdbc/telemetry/TelemetryIT.java

@@ -47,6 +47,13 @@ public void testTelemetry() throws Exception {
     testTelemetryInternal(telemetry);
   }
 
+  @Ignore
+  @Test
+  @ConditionalIgnoreRule.ConditionalIgnore(condition = RunningOnGithubAction.class)
+  public void testSessionlessTelemetry() throws Exception, SFException {
+    testTelemetryInternal(createSessionlessTelemetry());
+  }
+
   @Ignore
   @Test
   @ConditionalIgnoreRule.ConditionalIgnore(condition = RunningOnGithubAction.class)
@@ -197,6 +204,26 @@ public void testClosedOAuthSessionlessTelemetry() throws Exception, SFException
     Assert.assertFalse(telemetry.sendBatchAsync().get());
   }
 
+  // Helper function to create a sessionless telemetry
+  // using createSessionlessTelemetry(CloseableHttpClient httpClient, String serverUrl)
+  private TelemetryClient createSessionlessTelemetry()
+      throws SFException, SQLException, IOException {
+    setUpPublicKey();
+    String privateKeyLocation = getFullPathFileInResource("rsa_key.p8");
+    Map<String, String> parameters = getConnectionParameters();
+    String jwtToken =
+        SessionUtil.generateJWTToken(
+            null, privateKeyLocation, null, parameters.get("account"), parameters.get("user"));
+
+    CloseableHttpClient httpClient = HttpUtil.buildHttpClient(null, null, false);
+    TelemetryClient telemetry =
+        (TelemetryClient)
+            TelemetryClient.createSessionlessTelemetry(
+                httpClient, String.format("%s:%s", parameters.get("host"), parameters.get("port")));
+    telemetry.refreshToken(jwtToken);
+    return telemetry;
+  }
+
   // Helper function to create a sessionless telemetry using keypair JWT
   private TelemetryClient createJWTSessionlessTelemetry()
       throws SFException, SQLException, IOException {