Get Secure bundle binary content

Author: clun

astra-sdk-devops/src/main/java/com/dtsx/astra/sdk/db/DatabaseClient.java

@@ -111,6 +111,15 @@ public boolean isActive() {
     // ----    SECURE BUNDLE        ----
     // ---------------------------------
 
+    /**
+     * Download SecureBundle for a specific data center.
+     * @return
+     *      binary content.
+     */
+    public byte[] downloadDefaultSecureConnectBundle() {
+        return Utils.downloadFile(getDefaultSecureConnectBundleUrl());
+    }
+
     /**
      * Download SecureBundle for a specific data center
      *
@@ -120,14 +129,39 @@ public boolean isActive() {
     public void downloadDefaultSecureConnectBundle(String destination) {
         // Parameters Validation
         Assert.hasLength(destination, "destination");
+        // Download binary in target folder
+        Utils.downloadFile(getDefaultSecureConnectBundleUrl(), destination);
+    }
+
+    /**
+     * This utility method retrieve the binary content for the bundle.
+     *
+     * @return
+     *      secure connect bundle binary content.
+     */
+    private String getDefaultSecureConnectBundleUrl() {
         if (!isActive())
             throw new IllegalStateException("Database '" + databaseId + "' is not available.");
         // Get list of urls
         ApiResponseHttp res = POST(getEndpointDatabase() + "/secureBundleURL");
         // Mapping
-        String url = (String) JsonUtils.unmarshallBean(res.getBody(), Map.class).get("downloadURL");
-        // Download binary in target folder
-        Utils.downloadFile(url, destination);
+        return (String) JsonUtils.unmarshallBean(res.getBody(), Map.class).get("downloadURL");
+    }
+
+    /**
+     * Download SecureBundle for a specific data center.
+     * @return
+     *      binary content.
+     */
+    public byte[] downloadSecureConnectBundle(String region) {
+        Assert.hasLength(region, "region");
+        Database db = get();
+        return downloadSecureConnectBundle(db.getInfo()
+                .getDatacenters()
+                .stream()
+                .filter(d -> region.equalsIgnoreCase(d.getRegion()))
+                .findFirst()
+                .orElseThrow(() -> new RegionNotFoundException(region, databaseId)));
     }
 
     /**
@@ -150,6 +184,16 @@ public void downloadSecureConnectBundle(String region, String destination) {
                 .orElseThrow(() -> new RegionNotFoundException(region, databaseId)), destination);
     }
 
+    /**
+     * Download SCB for a database and a datacenter in target location.
+     *
+     * @param dc
+     *         current region
+     */
+    private byte[] downloadSecureConnectBundle(Datacenter dc) {
+        return Utils.downloadFile(dc.getSecureBundleUrl());
+    }
+
     /**
      * Download SCB for a database and a datacenter in target location.
      *

astra-sdk-devops/src/main/java/com/dtsx/astra/sdk/utils/Utils.java

@@ -17,8 +17,10 @@
 package com.dtsx.astra.sdk.utils;
 
 import java.io.BufferedInputStream;
+import java.io.ByteArrayOutputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
+import java.io.InputStream;
 import java.net.HttpURLConnection;
 import java.net.MalformedURLException;
 import java.net.URL;
@@ -57,7 +59,32 @@ public static boolean hasAllLength(String... lStr) {
         if (null == lStr) return false;
         return Arrays.stream(lStr).allMatch(Utils::hasLength);
     }
-    
+
+    /**
+     * Download File Content.
+     *
+     * @param fileUrl
+     *      current file URL
+     * @return
+     *      the file content
+     */
+    public static byte[] downloadFile(String fileUrl) {
+        try {
+            URL url = new URL(fileUrl);
+            try (InputStream in = new BufferedInputStream(url.openStream());
+                 ByteArrayOutputStream out = new ByteArrayOutputStream()) {
+                byte[] buffer = new byte[1024];
+                int bytesRead;
+                while ((bytesRead = in.read(buffer, 0, buffer.length)) != -1) {
+                    out.write(buffer, 0, bytesRead);
+                }
+                return out.toByteArray();
+            }
+        } catch(IOException ioe) {
+            throw new IllegalArgumentException("Cannot download file",ioe);
+        }
+    }
+
     /**
      * downloadFile
      * 

astra-sdk-devops/src/test/java/com/dtsx/astra/sdk/db/DatabaseClientTest.java

@@ -78,6 +78,10 @@ public void shouldDownloadDefaultScbTest() {
         // Then
         Assertions.assertTrue(new File(randomFile).exists());
         getSdkTestDatabaseClient().downloadSecureConnectBundle(SDK_TEST_DB_REGION, randomFile);
+        // When
+        byte[] data = getSdkTestDatabaseClient().downloadDefaultSecureConnectBundle();
+        // Then
+        Assertions.assertTrue(data != null && data.length > 1000);
     }
 
     @Test
@@ -93,8 +97,13 @@ public void shouldDownloadRegionScbTest() {
         // When
         getSdkTestDatabaseClient().downloadSecureConnectBundle(SDK_TEST_DB_REGION, randomFile);
         Assertions.assertTrue(new File(randomFile).exists());
+        // When
+        byte[] data = getSdkTestDatabaseClient().downloadSecureConnectBundle(SDK_TEST_DB_REGION);
+        // Then
+        Assertions.assertTrue(data != null && data.length > 1000);
     }
 
+
     @Test
     @Order(5)
     @DisplayName("05. Download All Cloud Secured Bundle")

astra-spring-boot-3x-autoconfigure/pom.xml

@@ -12,7 +12,7 @@
   </parent>
 
   <properties>
-    <spring-boot.version>3.1.0</spring-boot.version>
+    <spring-boot.version>3.1.1</spring-boot.version>
   </properties>
 
   <dependencies>