update profiling server url for v2 (#13)

middleware-labs · Jun 24, 2024 · a8f1e8e · a8f1e8e
1 parent 657b853
commit a8f1e8e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
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
@@ -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
@@ -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 {