Migrate to Jakarta Equivalents

build.gradle

@@ -41,6 +41,7 @@ apply plugin: 'com.palantir.baseline'
 apply plugin: 'com.palantir.consistent-versions'
 apply plugin: 'com.palantir.baseline-java-versions'
 apply plugin: 'com.palantir.jdks.latest'
+apply plugin: 'com.palantir.consistent-versions'
 
 repositories {
     mavenCentral() { metadataSources { mavenPom(); ignoreGradleMetadataRedirection() } }

settings.gradle

@@ -4,9 +4,15 @@ include 'tracing'
 include 'tracing-api'
 include 'tracing-benchmarks'
 include 'tracing-demos'
-include 'tracing-jaxrs'
-include 'tracing-jersey'
 include 'tracing-okhttp3'
-include 'tracing-servlet'
 include 'tracing-test-utils'
-include 'tracing-undertow'
+
+withJakarta('tracing-servlet')
+withJakarta('tracing-undertow')
+withJakarta('tracing-jaxrs')
+withJakarta('tracing-jersey')
+
+def withJakarta(name) {
+  include name
+  include "${name}-jakarta"
+}

tracing-jaxrs-jakarta/build.gradle

@@ -0,0 +1,31 @@
+/*
+ * (c) Copyright 2018 Palantir Technologies Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+apply plugin: 'org.inferred.processors'
+
+apply plugin: 'com.palantir.external-publish-jar'
+apply plugin: 'com.palantir.revapi'
+
+dependencies {
+    api project(":tracing")
+    api "jakarta.ws.rs:jakarta.ws.rs-api"
+
+    testImplementation "ch.qos.logback:logback-classic"
+    testImplementation "junit:junit"
+    testImplementation "org.assertj:assertj-core"
+    testImplementation "org.jmock:jmock"
+    testImplementation "org.mockito:mockito-core"
+}

tracing-jaxrs-jakarta/src/main/java/com/palantir/tracing/jaxrs/JaxRsTracers.java

@@ -0,0 +1,55 @@
+/*
+ * (c) Copyright 2018 Palantir Technologies Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.palantir.tracing.jaxrs;
+
+import com.palantir.tracing.CloseableSpan;
+import com.palantir.tracing.Detached;
+import com.palantir.tracing.DetachedSpan;
+import com.palantir.tracing.Tracers;
+import jakarta.ws.rs.WebApplicationException;
+import jakarta.ws.rs.core.StreamingOutput;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.concurrent.Callable;
+
+public final class JaxRsTracers {
+
+    private JaxRsTracers() {}
+
+    /** Like {@link Tracers#wrap(Callable)}, but for StreamingOutputs. */
+    public static StreamingOutput wrap(StreamingOutput delegate) {
+        return new TracingAwareStreamingOutput(delegate);
+    }
+
+    private static class TracingAwareStreamingOutput implements StreamingOutput {
+
+        private final StreamingOutput delegate;
+        private final Detached detached;
+
+        TracingAwareStreamingOutput(StreamingOutput delegate) {
+            this.delegate = delegate;
+            this.detached = DetachedSpan.detach();
+        }
+
+        @Override
+        public void write(OutputStream output) throws IOException, WebApplicationException {
+            try (CloseableSpan ignored = detached.childSpan("streaming-output")) {
+                delegate.write(output);
+            }
+        }
+    }
+}

tracing-jaxrs-jakarta/src/test/java/com/palantir/tracing/jaxrs/JaxRsTracersTest.java

@@ -0,0 +1,84 @@
+/*
+ * (c) Copyright 2018 Palantir Technologies Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.palantir.tracing.jaxrs;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import com.palantir.tracing.AlwaysSampler;
+import com.palantir.tracing.Tracer;
+import jakarta.ws.rs.core.StreamingOutput;
+import java.io.ByteArrayOutputStream;
+import org.junit.Test;
+
+public final class JaxRsTracersTest {
+
+    @Test
+    public void testWrappingStreamingOutput_streamingOutputTraceIsIsolated_sampled() throws Exception {
+        Tracer.setSampler(AlwaysSampler.INSTANCE);
+        Tracer.getAndClearTrace();
+
+        Tracer.fastStartSpan("outside");
+        StreamingOutput streamingOutput = JaxRsTracers.wrap(_os -> {
+            Tracer.fastStartSpan("inside"); // never completed
+        });
+        streamingOutput.write(new ByteArrayOutputStream());
+        assertThat(Tracer.completeSpan().get().getOperation()).isEqualTo("outside");
+    }
+
+    @Test
+    public void testWrappingStreamingOutput_streamingOutputTraceIsIsolated_unsampled() throws Exception {
+        Tracer.setSampler(() -> false);
+        Tracer.getAndClearTrace();
+
+        Tracer.fastStartSpan("outside");
+        StreamingOutput streamingOutput = JaxRsTracers.wrap(_os -> {
+            Tracer.fastStartSpan("inside"); // never completed
+        });
+        streamingOutput.write(new ByteArrayOutputStream());
+        assertThat(Tracer.hasTraceId()).isTrue();
+        Tracer.fastCompleteSpan();
+        assertThat(Tracer.hasTraceId()).isFalse();
+    }
+
+    @Test
+    public void testWrappingStreamingOutput_traceStateIsCapturedAtConstructionTime_sampled() throws Exception {
+        Tracer.setSampler(AlwaysSampler.INSTANCE);
+        Tracer.getAndClearTrace();
+
+        Tracer.fastStartSpan("before-construction");
+        StreamingOutput streamingOutput = JaxRsTracers.wrap(_os -> {
+            assertThat(Tracer.completeSpan().get().getOperation()).isEqualTo("streaming-output");
+        });
+        Tracer.fastStartSpan("after-construction");
+        streamingOutput.write(new ByteArrayOutputStream());
+    }
+
+    @Test
+    public void testWrappingStreamingOutput_traceStateIsCapturedAtConstructionTime_unsampled() throws Exception {
+        Tracer.setSampler(() -> false);
+        Tracer.getAndClearTrace();
+
+        Tracer.fastStartSpan("before-construction");
+        StreamingOutput streamingOutput = JaxRsTracers.wrap(_os -> {
+            assertThat(Tracer.hasTraceId()).isTrue();
+            Tracer.fastCompleteSpan();
+            assertThat(Tracer.hasTraceId()).isFalse();
+        });
+        Tracer.fastStartSpan("after-construction");
+        streamingOutput.write(new ByteArrayOutputStream());
+    }
+}

tracing-jaxrs/build.gradle

@@ -19,9 +19,13 @@ apply plugin: 'org.inferred.processors'
 apply plugin: 'com.palantir.external-publish-jar'
 apply plugin: 'com.palantir.revapi'
 
+versionsLock {
+    disableJavaPluginDefaults()
+}
+
 dependencies {
     api project(":tracing")
-    api "jakarta.ws.rs:jakarta.ws.rs-api"
+    api "javax.ws.rs:javax.ws.rs-api:2.1.1"
 
     testImplementation "ch.qos.logback:logback-classic"
     testImplementation "junit:junit"

tracing-jersey-jakarta/build.gradle

@@ -0,0 +1,34 @@
+/*
+ * (c) Copyright 2018 Palantir Technologies Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+apply plugin: 'com.palantir.external-publish-jar'
+apply plugin: 'com.palantir.revapi'
+
+dependencies {
+    api "org.glassfish.jersey.core:jersey-server"
+    api project(":tracing")
+
+    implementation 'com.google.guava:guava'
+    implementation 'jakarta.ws.rs:jakarta.ws.rs-api'
+    implementation project(':tracing-api')
+
+    testImplementation "io.dropwizard:dropwizard-testing"
+    testImplementation "org.junit.jupiter:junit-jupiter"
+    testImplementation "org.assertj:assertj-core"
+    testImplementation "org.hamcrest:hamcrest-all"
+    testImplementation "org.mockito:mockito-core"
+    testImplementation "org.mockito:mockito-junit-jupiter"
+}