Skip to content

Commit 15c7784

Browse files
authored
Merge 1f1581e into 38bba80
2 parents 38bba80 + 1f1581e commit 15c7784

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

src/main/java/com/warnermedia/kplserver/App.java

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ public static void main(String[] args) throws Exception {
2525
ServerSocket errSocket = new ServerSocket(port);
2626
errSocket.setSoTimeout(100);
2727

28-
KinesisEventPublisher kinesisEventPublisher = new KinesisEventPublisher(stream, getRegion(), errSocket);
28+
KinesisEventPublisher kinesisEventPublisher = new KinesisEventPublisher(stream, getRegion(), getMetricsLevel(), errSocket);
2929

3030
// graceful shutdowns
3131
Runtime.getRuntime().addShutdownHook(new Thread() {
@@ -79,4 +79,13 @@ static String getKinesisStream() {
7979
static String getRegion() {
8080
return System.getenv("AWS_DEFAULT_REGION");
8181
}
82+
83+
static String getMetricsLevel() {
84+
String p = System.getenv("METRICS_LEVEL");
85+
if (p == null || p.equals("")) {
86+
return "detailed";
87+
}
88+
return p;
89+
}
90+
8291
}

src/main/java/com/warnermedia/kplserver/KinesisEventPublisher.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import org.apache.commons.logging.Log;
1313
import org.apache.commons.logging.LogFactory;
1414

15+
import javax.security.auth.login.Configuration;
1516
import java.io.DataOutputStream;
1617
import java.io.IOException;
1718
import java.math.BigInteger;
@@ -33,10 +34,11 @@ public class KinesisEventPublisher {
3334
ServerSocket errSocket;
3435
Socket errClient;
3536

36-
public KinesisEventPublisher(String stream, String region, ServerSocket errSocket) {
37+
public KinesisEventPublisher(String stream, String region, String metricsLevel, ServerSocket errSocket) {
3738
this.stream = stream;
3839
kinesis = new KinesisProducer(new KinesisProducerConfiguration()
39-
.setRegion(region));
40+
.setRegion(region)
41+
.setMetricsLevel(metricsLevel));
4042
this.errSocket = errSocket;
4143
}
4244

0 commit comments

Comments
 (0)