Add YAML config parsing from file

Author: majanjua-amzn

awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AwsApplicationSignalsCustomizerProvider.java

@@ -47,6 +47,11 @@
 import io.opentelemetry.sdk.trace.SpanProcessor;
 import io.opentelemetry.sdk.trace.export.SpanExporter;
 import io.opentelemetry.sdk.trace.samplers.Sampler;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
 import java.time.Duration;
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -487,6 +492,18 @@ static AwsXrayAdaptiveSamplingConfig parseConfigString(String config)
     if (config == null) {
       return null;
     }
+
+    // Check if the config is a file path and the file exists
+    Path path = Paths.get(config);
+    if (Files.exists(path)) {
+      try {
+        config = String.join("\n", Files.readAllLines(path, StandardCharsets.UTF_8));
+      } catch (IOException e) {
+        throw new IllegalArgumentException(
+            "Failed to read adaptive sampling configuration file: " + e.getMessage(), e);
+      }
+    }
+
     ObjectMapper yamlMapper = new ObjectMapper(new YAMLFactory());
     Map<String, Object> configMap =
         yamlMapper.readValue(config, new TypeReference<Map<String, Object>>() {});

awsagentprovider/src/test/java/software/amazon/opentelemetry/javaagent/providers/AwsApplicationSignalsCustomizerProviderTest.java

@@ -20,6 +20,10 @@
 import static org.assertj.core.api.Assertions.assertThatNoException;
 
 import com.fasterxml.jackson.core.JsonProcessingException;
+import io.opentelemetry.contrib.awsxray.AwsXrayAdaptiveSamplingConfig;
+import java.io.File;
+import java.net.URISyntaxException;
+import java.net.URL;
 import org.junit.jupiter.api.Test;
 
 class AwsApplicationSignalsCustomizerProviderTest {
@@ -57,4 +61,42 @@ void setAdaptiveSamplingConfigFromString_invalidYaml() {
                 AwsApplicationSignalsCustomizerProvider.parseConfigString(
                     "{version: 1, invalid: yaml: structure}"));
   }
+
+  @Test
+  void setAdaptiveSamplingConfigFromFile_validYaml()
+      throws JsonProcessingException, URISyntaxException {
+    // Get the resource file path
+    URL resourceUrl =
+        getClass().getClassLoader().getResource("adaptive-sampling-config-valid.yaml");
+    assertThat(resourceUrl).isNotNull();
+
+    // Get the absolute file path
+    File configFile = new File(resourceUrl.toURI());
+    String absolutePath = configFile.getAbsolutePath();
+
+    // Parse the config using the file path
+    AwsXrayAdaptiveSamplingConfig config =
+        AwsApplicationSignalsCustomizerProvider.parseConfigString(absolutePath);
+
+    // Assert the configuration was parsed correctly
+    assertThat(config).isNotNull();
+    assertThat(config.getVersion()).isEqualTo(1);
+    assertThat(config.getErrorCaptureLimit().getErrorSpansPerSecond()).isEqualTo(10);
+  }
+
+  @Test
+  void setAdaptiveSamplingConfigFromFile_invalidYaml() throws URISyntaxException {
+    // Get the resource file path
+    URL resourceUrl =
+        getClass().getClassLoader().getResource("adaptive-sampling-config-invalid.yaml");
+    assertThat(resourceUrl).isNotNull();
+
+    // Get the absolute file path
+    File configFile = new File(resourceUrl.toURI());
+    String absolutePath = configFile.getAbsolutePath();
+
+    // Parse the config using the file path
+    assertThatException()
+        .isThrownBy(() -> AwsApplicationSignalsCustomizerProvider.parseConfigString(absolutePath));
+  }
 }

awsagentprovider/src/test/resources/adaptive-sampling-config-invalid.yaml

@@ -0,0 +1,13 @@
+version: 1.0
+anomalyConditions:
+  - errorCodeRegex: "^5\\d\\d$"
+    usage: both
+  - errorCodeRegex: "^4\\d\\d$"
+    usage: both
+  - errorCodeRegex: "^3\\d\\d$"
+    usage: both
+  - errorCodeRegex: "^2\\d\\d$"
+    operations: invalid part of config
+    usage: both
+errorCaptureLimit:
+  errorSpansPerSecond: 10

awsagentprovider/src/test/resources/adaptive-sampling-config-valid.yaml

@@ -0,0 +1,12 @@
+version: 1.0
+anomalyConditions:
+  - errorCodeRegex: "^5\\d\\d$"
+    usage: both
+  - errorCodeRegex: "^4\\d\\d$"
+    usage: both
+  - errorCodeRegex: "^3\\d\\d$"
+    usage: both
+  - errorCodeRegex: "^2\\d\\d$"
+    usage: both
+errorCaptureLimit:
+  errorSpansPerSecond: 10