Merge 1f1581e into 38bba80

turnerlabs · Mar 20, 2022 · 15c7784 · 15c7784
2 parents 38bba80 + 1f1581e
commit 15c7784
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/main/java/com/warnermedia/kplserver/App.java b/src/main/java/com/warnermedia/kplserver/App.java
@@ -25,7 +25,7 @@ public static void main(String[] args) throws Exception {
     ServerSocket errSocket = new ServerSocket(port);
     errSocket.setSoTimeout(100);
 
-    KinesisEventPublisher kinesisEventPublisher = new KinesisEventPublisher(stream, getRegion(), errSocket);
+    KinesisEventPublisher kinesisEventPublisher = new KinesisEventPublisher(stream, getRegion(), getMetricsLevel(), errSocket);
 
     // graceful shutdowns
     Runtime.getRuntime().addShutdownHook(new Thread() {
@@ -79,4 +79,13 @@ static String getKinesisStream() {
   static String getRegion() {
     return System.getenv("AWS_DEFAULT_REGION");
   }
+
+  static String getMetricsLevel() {
+    String p = System.getenv("METRICS_LEVEL");
+    if (p == null || p.equals("")) {
+      return "detailed";
+    }
+    return p;
+  }
+
 }
diff --git a/src/main/java/com/warnermedia/kplserver/KinesisEventPublisher.java b/src/main/java/com/warnermedia/kplserver/KinesisEventPublisher.java
@@ -12,6 +12,7 @@
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 
+import javax.security.auth.login.Configuration;
 import java.io.DataOutputStream;
 import java.io.IOException;
 import java.math.BigInteger;
@@ -33,10 +34,11 @@ public class KinesisEventPublisher {
   ServerSocket errSocket;
   Socket errClient;
 
-  public KinesisEventPublisher(String stream, String region, ServerSocket errSocket) {
+  public KinesisEventPublisher(String stream, String region, String metricsLevel, ServerSocket errSocket) {
     this.stream = stream;
     kinesis = new KinesisProducer(new KinesisProducerConfiguration()
-      .setRegion(region));
+      .setRegion(region)
+      .setMetricsLevel(metricsLevel));
     this.errSocket = errSocket;
   }