diff --git a/examples/extension/src/main/java/com/example/javaagent/config/EnvironmentConfig.java b/examples/extension/src/main/java/com/example/javaagent/config/EnvironmentConfig.java index 45eff59d652d..937566e84a69 100644 --- a/examples/extension/src/main/java/com/example/javaagent/config/EnvironmentConfig.java +++ b/examples/extension/src/main/java/com/example/javaagent/config/EnvironmentConfig.java @@ -9,7 +9,7 @@ public class EnvironmentConfig { public static final String MW_PROFILING_SERVER_URL = - System.getenv().getOrDefault("MW_PROFILING_SERVER_URL", "https://profiling.middleware.io"); + System.getenv().get("MW_PROFILING_SERVER_URL"); public static final String MW_PROFILING_ALLOC = System.getenv().getOrDefault("MW_PROFILING_ALLOC", "512k"); diff --git a/examples/extension/src/main/java/com/example/profile/PyroscopeProfile.java b/examples/extension/src/main/java/com/example/profile/PyroscopeProfile.java index 1a94688806a5..afdb2215b56a 100644 --- a/examples/extension/src/main/java/com/example/profile/PyroscopeProfile.java +++ b/examples/extension/src/main/java/com/example/profile/PyroscopeProfile.java @@ -33,13 +33,17 @@ public static void startProfiling() { try { String tenantId = authenticateAndGetTenantId(); if (tenantId != null) { + String profilingServerUrl = EnvironmentConfig.MW_PROFILING_SERVER_URL; + if (profilingServerUrl == null) { + profilingServerUrl = "https://" + tenantId + ".middleware.io/profiling"; + } PyroscopeAgent.start( new Config.Builder() .setApplicationName(SystemProperties.SERVICE_NAME) .setProfilingEvent(EventType.ITIMER) .setProfilingAlloc(EnvironmentConfig.MW_PROFILING_ALLOC) .setProfilingLock(EnvironmentConfig.MW_PROFILING_LOCK) - .setServerAddress(EnvironmentConfig.MW_PROFILING_SERVER_URL) + .setServerAddress(profilingServerUrl) .setTenantID(tenantId) .build()); } else {