diff --git a/kubernetes-manifests/adservice.yaml b/kubernetes-manifests/adservice.yaml index dc0879f..7d4d619 100644 --- a/kubernetes-manifests/adservice.yaml +++ b/kubernetes-manifests/adservice.yaml @@ -31,6 +31,10 @@ spec: env: - name: PORT value: "9555" + - name: JAEGER_ENABLED + value: "false" + - name: PROMETHEUS_ENABLED + value: "false" resources: requests: cpu: 200m diff --git a/src/adservice/build.gradle b/src/adservice/build.gradle index 59b0313..690018e 100644 --- a/src/adservice/build.gradle +++ b/src/adservice/build.gradle @@ -45,6 +45,7 @@ dependencies { compile fileTree(dir: offlineCompile, include: '*.jar') } else { compile "com.google.api.grpc:proto-google-common-protos:1.11.0", + "io.opencensus:opencensus-exporter-trace-jaeger:${opencensusVersion}", "io.opencensus:opencensus-exporter-stats-prometheus:${opencensusVersion}", "io.opencensus:opencensus-exporter-stats-stackdriver:${opencensusVersion}", "io.opencensus:opencensus-exporter-trace-stackdriver:${opencensusVersion}", diff --git a/src/adservice/src/main/java/hipstershop/AdService.java b/src/adservice/src/main/java/hipstershop/AdService.java index 1224efe..74cda60 100644 --- a/src/adservice/src/main/java/hipstershop/AdService.java +++ b/src/adservice/src/main/java/hipstershop/AdService.java @@ -32,6 +32,7 @@ import io.opencensus.common.Duration; import io.opencensus.common.Scope; import io.opencensus.contrib.grpc.metrics.RpcViews; import io.opencensus.exporter.stats.prometheus.PrometheusStatsCollector; +import io.opencensus.exporter.trace.jaeger.JaegerTraceExporter; import io.opencensus.exporter.trace.logging.LoggingTraceExporter; import io.opencensus.exporter.stats.stackdriver.StackdriverStatsConfiguration; import io.opencensus.exporter.stats.stackdriver.StackdriverStatsExporter; @@ -43,6 +44,7 @@ import io.opencensus.trace.SpanBuilder; import io.opencensus.trace.Tracer; import io.opencensus.trace.Tracing; import io.opencensus.trace.samplers.Samplers; +import io.prometheus.client.exporter.HTTPServer; import java.io.IOException; import java.util.Collection; import java.util.ArrayList; @@ -201,11 +203,6 @@ public class AdService { public static void initStackdriver() { logger.info("Initialize StackDriver"); - // Registers all RPC views. - RpcViews.registerAllViews(); - - // Registers logging trace exporter. - LoggingTraceExporter.register(); long sleepTime = 10; /* seconds */ int maxAttempts = 5; boolean statsExporterRegistered = false; @@ -243,10 +240,38 @@ public class AdService { logger.info("StackDriver initialization complete."); } + static void initJaeger() { + boolean enabled = Boolean.parseBoolean(System.getenv("JAEGER_ENABLED")); + if (enabled) { + // Register Jaeger Tracing. + JaegerTraceExporter.createAndRegister("http://jaeger-collector:14268/api/traces", "adservice"); + logger.info("Jaeger initialization complete."); + } else { + logger.info("Jaeger initialization disabled."); + } + } + + static void initPrometheus() throws IOException { + boolean enabled = Boolean.parseBoolean(System.getenv("PROMETHEUS_ENABLED")); + if (enabled) { + PrometheusStatsCollector.createAndRegister(); + HTTPServer prometheusServer = new HTTPServer(9090, true); + logger.info("Prometheus initialization complete."); + } else { + logger.info("Prometheus initialization disabled."); + } + } + /** Main launches the server from the command line. */ public static void main(String[] args) throws IOException, InterruptedException { // Add final keyword to pass checkStyle. + // Registers all RPC views. + RpcViews.registerAllViews(); + + // Registers logging trace exporter. + LoggingTraceExporter.register(); + new Thread( new Runnable() { public void run(){ initStackdriver(); @@ -254,7 +279,10 @@ public class AdService { }).start(); // Register Prometheus exporters and export metrics to a Prometheus HTTPServer. - PrometheusStatsCollector.createAndRegister(); + initPrometheus(); + + // Register Jaeger + initJaeger(); // Start the RPC server. You shouldn't see any output from gRPC before this. logger.info("AdService starting."); @@ -262,4 +290,4 @@ public class AdService { service.start(); service.blockUntilShutdown(); } -} \ No newline at end of file +}