diff --git a/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralAPIClient.java b/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralAPIClient.java index 711cc646faa3..c1a505cf5efa 100644 --- a/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralAPIClient.java +++ b/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralAPIClient.java @@ -83,6 +83,7 @@ import static org.ballerinalang.central.client.CentralClientConstants.APPLICATION_OCTET_STREAM; import static org.ballerinalang.central.client.CentralClientConstants.AUTHORIZATION; import static org.ballerinalang.central.client.CentralClientConstants.BALA_URL; +import static org.ballerinalang.central.client.CentralClientConstants.BALLERINA_CENTRAL_TELEMETRY_DISABLED; import static org.ballerinalang.central.client.CentralClientConstants.BALLERINA_PLATFORM; import static org.ballerinalang.central.client.CentralClientConstants.CONTENT_DISPOSITION; import static org.ballerinalang.central.client.CentralClientConstants.CONTENT_TYPE; @@ -99,6 +100,7 @@ import static org.ballerinalang.central.client.CentralClientConstants.USER_AGENT; import static org.ballerinalang.central.client.CentralClientConstants.VERSION; import static org.ballerinalang.central.client.Utils.ProgressRequestBody; +import static org.ballerinalang.central.client.Utils.SET_TEST_MODE_ACTIVE; import static org.ballerinalang.central.client.Utils.checkHash; import static org.ballerinalang.central.client.Utils.createBalaInHomeRepo; import static org.ballerinalang.central.client.Utils.getAsList; @@ -1681,16 +1683,14 @@ protected void closeClient(OkHttpClient client) throws IOException { * @return Http request builder */ protected Request.Builder getNewRequest(String supportedPlatform, String ballerinaVersion) { - if (this.accessToken.isEmpty()) { - return new Request.Builder() - .addHeader(BALLERINA_PLATFORM, supportedPlatform) - .addHeader(USER_AGENT, ballerinaVersion); - } else { - return new Request.Builder() - .addHeader(BALLERINA_PLATFORM, supportedPlatform) - .addHeader(USER_AGENT, ballerinaVersion) - .addHeader(AUTHORIZATION, getBearerToken(this.accessToken)); + Request.Builder requestBuilder = new Request.Builder() + .addHeader(BALLERINA_PLATFORM, supportedPlatform) + .addHeader(USER_AGENT, ballerinaVersion) + .addHeader(BALLERINA_CENTRAL_TELEMETRY_DISABLED, String.valueOf(SET_TEST_MODE_ACTIVE)); + if (!this.accessToken.isEmpty()) { + requestBuilder.addHeader(AUTHORIZATION, getBearerToken(this.accessToken)); } + return requestBuilder; } public String accessToken() { diff --git a/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralClientConstants.java b/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralClientConstants.java index 933ebaa160f2..5328e70727c2 100644 --- a/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralClientConstants.java +++ b/cli/central-client/src/main/java/org/ballerinalang/central/client/CentralClientConstants.java @@ -60,5 +60,6 @@ private CentralClientConstants() { public static final int UPDATE_INTERVAL_MILLIS = 1000; public static final String SHA256 = "sha-256="; public static final String SHA256_ALGORITHM = "SHA-256"; - + public static final String TEST_MODE_ACTIVE = "TEST_MODE_ACTIVE"; + public static final String BALLERINA_CENTRAL_TELEMETRY_DISABLED = "Ballerina-Central-Telemetry-Disabled"; } diff --git a/cli/central-client/src/main/java/org/ballerinalang/central/client/Utils.java b/cli/central-client/src/main/java/org/ballerinalang/central/client/Utils.java index cee307464019..1f968a104cc6 100644 --- a/cli/central-client/src/main/java/org/ballerinalang/central/client/Utils.java +++ b/cli/central-client/src/main/java/org/ballerinalang/central/client/Utils.java @@ -72,6 +72,7 @@ import static org.ballerinalang.central.client.CentralClientConstants.SHA256; import static org.ballerinalang.central.client.CentralClientConstants.SHA256_ALGORITHM; import static org.ballerinalang.central.client.CentralClientConstants.STAGING_REPO; +import static org.ballerinalang.central.client.CentralClientConstants.TEST_MODE_ACTIVE; import static org.ballerinalang.central.client.CentralClientConstants.UPDATE_INTERVAL_MILLIS; /** @@ -85,6 +86,7 @@ public class Utils { System.getenv(BALLERINA_STAGE_CENTRAL)); public static final boolean SET_BALLERINA_DEV_CENTRAL = Boolean.parseBoolean( System.getenv(BALLERINA_DEV_CENTRAL)); + public static final boolean SET_TEST_MODE_ACTIVE = Boolean.parseBoolean(System.getenv(TEST_MODE_ACTIVE)); private Utils() { }