diff --git a/tools/src/main/java/org/apache/kafka/tools/automq/GenerateS3UrlCmd.java b/tools/src/main/java/org/apache/kafka/tools/automq/GenerateS3UrlCmd.java index 9a342e50f2..3e1c3811b3 100644 --- a/tools/src/main/java/org/apache/kafka/tools/automq/GenerateS3UrlCmd.java +++ b/tools/src/main/java/org/apache/kafka/tools/automq/GenerateS3UrlCmd.java @@ -60,8 +60,14 @@ static class Parameter { this.s3SecretKey = res.getString("s3-secret-key"); this.s3Region = res.getString("s3-region"); String endpointProtocolStr = res.get("s3-endpoint-protocol"); + String currentEndpoint = res.getString("s3-endpoint"); + if (currentEndpoint.startsWith("http://") || currentEndpoint.startsWith("https://")) { + this.s3Endpoint = currentEndpoint.replace("http://", "").replace("https://", ""); + endpointProtocolStr = currentEndpoint.substring(0, currentEndpoint.indexOf("://")); + } else { + this.s3Endpoint = currentEndpoint; + } this.endpointProtocol = EndpointProtocol.getByName(endpointProtocolStr); - this.s3Endpoint = res.getString("s3-endpoint"); this.s3DataBucket = res.getString("s3-data-bucket"); String s3OpsBucketFromArg = res.getString("s3-ops-bucket"); if (s3OpsBucketFromArg == null) { @@ -144,10 +150,8 @@ public String run() { System.out.println(); //precheck - String s3Endpoint = parameter.s3Endpoint; - var context = S3Utils.S3Context.builder().setEndpoint( - s3Endpoint.startsWith("https://") || s3Endpoint.startsWith("http://") ? s3Endpoint : - parameter.endpointProtocol.getName() + "://" + s3Endpoint) + var context = S3Utils.S3Context.builder() + .setEndpoint(parameter.endpointProtocol.getName() + "://" + parameter.s3Endpoint) .setCredentialsProviders(List.of(() -> AwsBasicCredentials.create(parameter.s3AccessKey, parameter.s3SecretKey))) .setBucketName(parameter.s3DataBucket) .setRegion(parameter.s3Region) diff --git a/tools/src/main/java/org/apache/kafka/tools/automq/GenerateStartCmdCmd.java b/tools/src/main/java/org/apache/kafka/tools/automq/GenerateStartCmdCmd.java index 05283005ab..7954551aad 100644 --- a/tools/src/main/java/org/apache/kafka/tools/automq/GenerateStartCmdCmd.java +++ b/tools/src/main/java/org/apache/kafka/tools/automq/GenerateStartCmdCmd.java @@ -45,7 +45,7 @@ static class Parameter { final boolean controllerOnlyMode; Parameter(Namespace res) { - this.s3Url = res.getString("s3-url"); + this.s3Url = res.getString("s3-url").replace("http://", "").replace("https://", ""); this.brokerList = res.getString("broker-list"); this.controllerList = res.getString("controller-list"); this.networkBaselineBandwidthMB = res.getString("network-baseline-bandwidth-mb");