Addressing comments from PR #29

- Corrected a bad refactoring referencing `isColdStart`
- Added Copyright headers.

Author: agarbutt

src/main/java/com/newrelic/opentracing/aws/EnhancedSpanBuilder.java

@@ -1,3 +1,8 @@
+/*
+ * Copyright 2019 New Relic Corporation. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
 package com.newrelic.opentracing.aws;
 
 import io.opentracing.Span;

src/main/java/com/newrelic/opentracing/aws/SpanUtil.java

@@ -13,21 +13,25 @@
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.atomic.AtomicBoolean;
 
 public class SpanUtil {
 
   private SpanUtil() {}
 
   static <Input> Span buildSpan(
-      Input input, Context context, Tracer tracer, SpanContext spanContext) {
+      Input input,
+      Context context,
+      Tracer tracer,
+      SpanContext spanContext,
+      AtomicBoolean isColdStart) {
     return EnhancedSpanBuilder.basedOn(tracer, "handleRequest")
         .asChildOf(spanContext)
         .withTag("aws.requestId", context.getAwsRequestId())
         .withTag("aws.lambda.arn", context.getInvokedFunctionArn())
         .optionallyWithTag(
             "aws.lambda.eventSource.arn", EventSourceParser.parseEventSourceArn(input))
-        .optionallyWithTag(
-            "aws.lambda.coldStart", TracingRequestHandler.isColdStart.getAndSet(false))
+        .optionallyWithTag("aws.lambda.coldStart", isColdStart.getAndSet(false))
         .start();
   }
 

src/main/java/com/newrelic/opentracing/aws/TracingRequestHandler.java

@@ -43,7 +43,7 @@ default Output handleRequest(Input input, Context context) {
     final Tracer tracer = GlobalTracer.get();
     final SpanContext spanContext = extractContext(tracer, input);
 
-    Span span = SpanUtil.buildSpan(input, context, tracer, spanContext);
+    Span span = SpanUtil.buildSpan(input, context, tracer, spanContext, isColdStart);
     try (Scope scope = tracer.activateSpan(span)) {
       try {
         Output output = doHandleRequest(input, context);

src/main/java/com/newrelic/opentracing/aws/TracingRequestStreamHandler.java

@@ -42,7 +42,7 @@ default void handleRequest(InputStream input, OutputStream output, Context conte
     final Tracer tracer = GlobalTracer.get();
     final SpanContext spanContext = extractContext(tracer, input);
 
-    Span span = SpanUtil.buildSpan(input, context, tracer, spanContext);
+    Span span = SpanUtil.buildSpan(input, context, tracer, spanContext, isColdStart);
     try (Scope scope = tracer.activateSpan(span)) {
       try {
         doHandleRequest(input, output, context);

src/test/java/com/newrelic/opentracing/aws/EnhancedSpanBuilderTest.java

@@ -1,3 +1,8 @@
+/*
+ * Copyright 2019 New Relic Corporation. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
 package com.newrelic.opentracing.aws;
 
 import static io.opentracing.propagation.Format.Builtin.TEXT_MAP;