Cloud API

agent-bridge/src/main/java/com/newrelic/agent/bridge/AgentBridge.java

@@ -42,6 +42,8 @@ public final class AgentBridge {
 
     public static volatile AsyncApi asyncApi = new NoOpAsyncApi();
 
+    public static volatile CloudApi cloud = NoOpCloud.INSTANCE;
+
     public static volatile CollectionFactory collectionFactory = new DefaultCollectionFactory();
 
     /**

agent-bridge/src/main/java/com/newrelic/agent/bridge/CloudApi.java

@@ -0,0 +1,31 @@
+/*
+ *
+ *  * Copyright 2024 New Relic Corporation. All rights reserved.
+ *  * SPDX-License-Identifier: Apache-2.0
+ *
+ */
+
+package com.newrelic.agent.bridge;
+
+import com.newrelic.api.agent.Cloud;
+import com.newrelic.api.agent.CloudAccountInfo;
+
+/**
+ * Internal Cloud API. This extends the public Cloud API and adds methods
+ * for retrieving the data set by the public API methods.
+ */
+public interface CloudApi extends Cloud {
+
+    /**
+     * Return the general account information of the provided type.
+     * This data is either set by {@link Cloud#setAccountInfo(CloudAccountInfo, String)}
+     * or the agent config.
+     */
+    String getAccountInfo(CloudAccountInfo cloudAccountInfo);
+
+    /**
+     * Retrieves the account information for a cloud service SDK client.
+     * If no data was recorded for the SDK client, the general account information will be returned.
+     */
+    String getAccountInfo(Object sdkClient, CloudAccountInfo cloudAccountInfo);
+}

agent-bridge/src/main/java/com/newrelic/agent/bridge/NoOpAgent.java

@@ -8,6 +8,7 @@
 package com.newrelic.agent.bridge;
 
 import com.newrelic.api.agent.AiMonitoring;
+import com.newrelic.api.agent.Cloud;
 import com.newrelic.api.agent.Config;
 import com.newrelic.api.agent.ErrorApi;
 import com.newrelic.api.agent.Insights;
@@ -73,6 +74,11 @@ public AiMonitoring getAiMonitoring() {
         return NoOpAiMonitoring.INSTANCE;
     }
 
+    @Override
+    public Cloud getCloud() {
+        return NoOpCloud.INSTANCE;
+    }
+
     @Override
     public ErrorApi getErrorApi() {
         return NoOpErrorApi.INSTANCE;

agent-bridge/src/main/java/com/newrelic/agent/bridge/NoOpCloud.java

@@ -0,0 +1,37 @@
+/*
+ *
+ *  * Copyright 2024 New Relic Corporation. All rights reserved.
+ *  * SPDX-License-Identifier: Apache-2.0
+ *
+ */
+
+package com.newrelic.agent.bridge;
+
+import com.newrelic.api.agent.CloudAccountInfo;
+
+public class NoOpCloud implements CloudApi {
+
+    public static final CloudApi INSTANCE = new NoOpCloud();
+
+    private NoOpCloud() {
+        // only instance should be the INSTANCE
+    }
+
+    @Override
+    public void setAccountInfo(CloudAccountInfo cloudAccountInfo, String value) {
+    }
+
+    @Override
+    public void setAccountInfo(Object sdkClient, CloudAccountInfo cloudAccountInfo, String value) {
+    }
+
+    @Override
+    public String getAccountInfo(CloudAccountInfo cloudAccountInfo) {
+        return null;
+    }
+
+    @Override
+    public String getAccountInfo(Object sdkClient, CloudAccountInfo cloudAccountInfo) {
+        return null;
+    }
+}

functional_test/src/test/java/com/newrelic/agent/extension/FakeExtensionAgent.java

@@ -11,6 +11,7 @@
 import com.newrelic.agent.bridge.TracedMethod;
 import com.newrelic.agent.bridge.Transaction;
 import com.newrelic.api.agent.AiMonitoring;
+import com.newrelic.api.agent.Cloud;
 import com.newrelic.api.agent.Config;
 import com.newrelic.api.agent.ErrorApi;
 import com.newrelic.api.agent.Insights;
@@ -44,6 +45,11 @@ public AiMonitoring getAiMonitoring() {
         return null;
     }
 
+    @Override
+    public Cloud getCloud() {
+        return null;
+    }
+
     @Override
     public ErrorApi getErrorApi() { throw new RuntimeException(); }
 

newrelic-agent/src/main/java/com/newrelic/agent/AgentImpl.java

@@ -17,6 +17,7 @@
 import com.newrelic.agent.service.ServiceFactory;
 import com.newrelic.agent.tracers.Tracer;
 import com.newrelic.api.agent.AiMonitoring;
+import com.newrelic.api.agent.Cloud;
 import com.newrelic.api.agent.ErrorApi;
 import com.newrelic.api.agent.Insights;
 import com.newrelic.api.agent.Logger;
@@ -144,6 +145,11 @@ public AiMonitoring getAiMonitoring() {
         return new AiMonitoringImpl();
     }
 
+    @Override
+    public Cloud getCloud() {
+        return AgentBridge.cloud;
+    }
+
     @Override
     public Logs getLogSender() {
         return ServiceFactory.getServiceManager().getLogSenderService();

newrelic-agent/src/main/java/com/newrelic/agent/MetricNames.java

@@ -361,6 +361,11 @@ public class MetricNames {
     public static final String SUPPORTABILITY_API_SET_ACCOUNT_NAME = "SetAccountName";
     public static final String SUPPORTABILITY_API_SET_USER_ID = "SetUserId";
 
+    // Cloud API
+    public static final String SUPPORTABILITY_API_CLOUD_SET_ACCOUNT_INFO_CLIENT = "Cloud/SetAccountInfoClient/";
+    public static final String SUPPORTABILITY_API_CLOUD_SET_ACCOUNT_INFO = "Cloud/SetAccountInfo/";
+    public static final String SUPPORTABILITY_CONFIG_AWS_ACCOUNT_ID = "Supportability/Cloud/ConfigAccountInfo/aws_account_id";
+
     //Transaction supportability metrics
     public static final String SUPPORTABILITY_TRANSACTION_STARTED = "Supportability/Transaction/StartedCount";
     public static final String SUPPORTABILITY_TRANSACTION_FINISHED = "Supportability/Transaction/FinishedCount";

newrelic-agent/src/main/java/com/newrelic/agent/cloud/CloudAccountInfoCache.java

@@ -0,0 +1,74 @@
+/*
+ *
+ *  * Copyright 2024 New Relic Corporation. All rights reserved.
+ *  * SPDX-License-Identifier: Apache-2.0
+ *
+ */
+
+package com.newrelic.agent.cloud;
+
+import com.github.benmanes.caffeine.cache.Caffeine;
+import com.github.benmanes.caffeine.cache.LoadingCache;
+import com.newrelic.agent.MetricNames;
+import com.newrelic.agent.config.AgentConfig;
+import com.newrelic.agent.service.ServiceFactory;
+import com.newrelic.api.agent.CloudAccountInfo;
+import com.newrelic.api.agent.NewRelic;
+
+import java.util.Collections;
+import java.util.EnumMap;
+import java.util.Map;
+
+/**
+ * This class implements the account info methods from the Cloud API.
+ */
+public class CloudAccountInfoCache {
+    private final LoadingCache<Object, Map<CloudAccountInfo, String>> cache;
+    // this object is used to store data that is not related to a specific sdk client
+    private static final Object NULL_CLIENT = new Object();
+
+    CloudAccountInfoCache() {
+        cache = Caffeine.newBuilder()
+                .initialCapacity(4)
+                .weakKeys()
+                .executor(Runnable::run)
+                .build((key) -> Collections.synchronizedMap(new EnumMap<>(CloudAccountInfo.class)));
+    }
+
+    public void setAccountInfo(CloudAccountInfo cloudAccountInfo, String value) {
+        setAccountInfo(NULL_CLIENT, cloudAccountInfo, value);
+    }
+
+    public void setAccountInfo(Object sdkClient, CloudAccountInfo cloudAccountInfo, String value) {
+        Map<CloudAccountInfo, String> accountInfo = cache.get(sdkClient);
+        accountInfo.put(cloudAccountInfo, value);
+    }
+
+    public String getAccountInfo(CloudAccountInfo cloudAccountInfo) {
+        Map<CloudAccountInfo, String> accountInfo = cache.getIfPresent(NULL_CLIENT);
+        if (accountInfo == null) {
+            return null;
+        }
+        return accountInfo.get(cloudAccountInfo);
+    }
+
+    public String getAccountInfo(Object sdkClient, CloudAccountInfo cloudAccountInfo) {
+        if (sdkClient == null) {
+            return getAccountInfo(cloudAccountInfo);
+        }
+        Map<CloudAccountInfo, String> accountInfo = cache.getIfPresent(sdkClient);
+        if (accountInfo == null) {
+            return getAccountInfo(cloudAccountInfo);
+        }
+        return accountInfo.get(cloudAccountInfo);
+    }
+
+    void retrieveDataFromConfig() {
+        AgentConfig agentConfig = ServiceFactory.getConfigService().getDefaultAgentConfig();
+        Object awsAccountId = agentConfig.getValue("cloud.aws.account_id");
+        if (awsAccountId != null) {
+            NewRelic.incrementCounter(MetricNames.SUPPORTABILITY_CONFIG_AWS_ACCOUNT_ID);
+            setAccountInfo(CloudAccountInfo.AWS_ACCOUNT_ID, awsAccountId.toString());
+        }
+    }
+}

newrelic-agent/src/main/java/com/newrelic/agent/cloud/CloudApiImpl.java

@@ -0,0 +1,61 @@
+/*
+ *
+ *  * Copyright 2024 New Relic Corporation. All rights reserved.
+ *  * SPDX-License-Identifier: Apache-2.0
+ *
+ */
+
+package com.newrelic.agent.cloud;
+
+import com.newrelic.agent.MetricNames;
+import com.newrelic.agent.bridge.AgentBridge;
+import com.newrelic.agent.bridge.CloudApi;
+import com.newrelic.api.agent.CloudAccountInfo;
+
+/**
+ * Facade for the Cloud API.
+ */
+public class CloudApiImpl implements CloudApi {
+
+    private final CloudAccountInfoCache accountInfoCache;
+
+    private CloudApiImpl() {
+        this(new CloudAccountInfoCache());
+        accountInfoCache.retrieveDataFromConfig();
+    }
+
+    // for testing
+    CloudApiImpl(CloudAccountInfoCache accountInfoCache) {
+        this.accountInfoCache = accountInfoCache;
+    }
+
+    // calling this method more than once will invalidate any Cloud API calls to set account info
+    public static void initialize() {
+        AgentBridge.cloud = new CloudApiImpl();
+    }
+
+    @Override
+    public void setAccountInfo(CloudAccountInfo cloudAccountInfo, String value) {
+        MetricNames.recordApiSupportabilityMetric(MetricNames.SUPPORTABILITY_API_CLOUD_SET_ACCOUNT_INFO + cloudAccountInfo.toString());
+        accountInfoCache.setAccountInfo(cloudAccountInfo, value);
+    }
+
+    @Override
+    public void setAccountInfo(Object sdkClient, CloudAccountInfo cloudAccountInfo, String value) {
+        MetricNames.recordApiSupportabilityMetric(MetricNames.SUPPORTABILITY_API_CLOUD_SET_ACCOUNT_INFO_CLIENT + cloudAccountInfo.toString());
+        accountInfoCache.setAccountInfo(sdkClient, cloudAccountInfo, value);
+    }
+
+    @Override
+    public String getAccountInfo(CloudAccountInfo cloudAccountInfo) {
+        // not recording metrics because this is for the internal API
+        return accountInfoCache.getAccountInfo(cloudAccountInfo);
+    }
+
+    @Override
+    public String getAccountInfo(Object sdkClient, CloudAccountInfo cloudAccountInfo) {
+        // not recording metrics because this is for the internal API
+        return accountInfoCache.getAccountInfo(sdkClient, cloudAccountInfo);
+    }
+
+}

newrelic-agent/src/main/java/com/newrelic/agent/core/CoreServiceImpl.java

@@ -13,6 +13,7 @@
 import com.newrelic.agent.MetricNames;
 import com.newrelic.agent.PrivateApiImpl;
 import com.newrelic.agent.TransactionService;
+import com.newrelic.agent.cloud.CloudApiImpl;
 import com.newrelic.agent.config.AgentConfig;
 import com.newrelic.agent.config.ConfigService;
 import com.newrelic.agent.logging.AgentLogManager;
@@ -70,6 +71,7 @@ protected void doStart() {
     private void initializeBridgeApis() {
         NewRelicApiImplementation.initialize();
         PrivateApiImpl.initialize(Agent.LOG);
+        CloudApiImpl.initialize();
     }
 
     /**

newrelic-agent/src/test/java/com/newrelic/agent/cloud/CloudAccountInfoCacheTest.java

@@ -0,0 +1,74 @@
+/*
+ *
+ *  * Copyright 2024 New Relic Corporation. All rights reserved.
+ *  * SPDX-License-Identifier: Apache-2.0
+ *
+ */
+
+package com.newrelic.agent.cloud;
+
+import com.newrelic.agent.service.ServiceFactory;
+import com.newrelic.agent.service.ServiceManager;
+import org.junit.Test;
+
+import static com.newrelic.api.agent.CloudAccountInfo.AWS_ACCOUNT_ID;
+import static org.junit.Assert.*;
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public class CloudAccountInfoCacheTest {
+
+    @Test
+    public void accountInfo() {
+        CloudAccountInfoCache cache = new CloudAccountInfoCache();
+
+        assertNull(cache.getAccountInfo(AWS_ACCOUNT_ID));
+
+        String accountId = "123456789";
+        cache.setAccountInfo(AWS_ACCOUNT_ID, accountId);
+
+        assertEquals(accountId, cache.getAccountInfo(AWS_ACCOUNT_ID));
+    }
+
+    @Test
+    public void accountInfoClient() {
+        CloudAccountInfoCache cache = new CloudAccountInfoCache();
+        Object sdkClient = new Object();
+
+        assertNull(cache.getAccountInfo(sdkClient, AWS_ACCOUNT_ID));
+
+        String accountId = "123456789";
+        cache.setAccountInfo(sdkClient, AWS_ACCOUNT_ID, accountId);
+
+        assertEquals(accountId, cache.getAccountInfo(sdkClient, AWS_ACCOUNT_ID));
+
+        Object anotherSdkClient = new Object();
+        assertNull(cache.getAccountInfo(anotherSdkClient, AWS_ACCOUNT_ID));
+    }
+
+    @Test
+    public void accountInfoClientFallback() {
+        CloudAccountInfoCache cache = new CloudAccountInfoCache();
+        String accountId = "123456789";
+        cache.setAccountInfo(AWS_ACCOUNT_ID, accountId);
+
+        Object sdkClient = new Object();
+        assertEquals(accountId, cache.getAccountInfo(sdkClient, AWS_ACCOUNT_ID));
+    }
+
+
+    @Test
+    public void retrieveDataFromConfigAccountInfo() {
+        CloudAccountInfoCache cache = new CloudAccountInfoCache();
+        String accountId = "123456789";
+
+        ServiceManager serviceManager = mock(ServiceManager.class, RETURNS_DEEP_STUBS);
+        ServiceFactory.setServiceManager(serviceManager);
+        when(serviceManager.getConfigService().getDefaultAgentConfig().getValue("cloud.aws.account_id"))
+                .thenReturn(accountId);
+        cache.retrieveDataFromConfig();
+
+        assertEquals(accountId, cache.getAccountInfo(AWS_ACCOUNT_ID));
+    }
+}