Enable Micronaut OpenTelemetry spans (#179)

Author: skykanin

pom.xml

@@ -132,6 +132,11 @@
         <artifactId>micronaut-tracing-annotation</artifactId>
         <version>${micronaut.tracing.version}</version>
     </dependency>
+    <dependency>
+      <groupId>io.micronaut.tracing</groupId>
+      <artifactId>micronaut-tracing-opentelemetry</artifactId>
+      <version>${micronaut.tracing.version}</version>
+    </dependency>
     <dependency>
       <groupId>io.micronaut.cache</groupId>
       <artifactId>micronaut-cache-core</artifactId>

src/main/java/no/ssb/dlp/pseudo/service/pseudo/PseudoController.java

@@ -12,6 +12,7 @@
 import io.micronaut.tracing.annotation.NewSpan;
 import io.micronaut.tracing.annotation.SpanTag;
 import io.reactivex.Flowable;
+import io.opentelemetry.api.trace.Span;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.media.Schema;
 import io.swagger.v3.oas.annotations.tags.Tag;
@@ -73,6 +74,12 @@ public HttpResponse<Flowable<byte[]>> pseudonymizeField(
             @SpanTag("pseudonymize column request") @Schema(implementation = PseudoFieldRequest.class) String request
     ) {
         PseudoFieldRequest req = Json.toObject(PseudoFieldRequest.class, request);
+        Span currentSpan = Span.current();
+        if (currentSpan.getSpanContext().isValid() && req != null) {
+            currentSpan.setAttribute("pseudo.field", req.getName());
+            currentSpan.setAttribute("pseudo.pattern", req.getPattern());
+            currentSpan.setAttribute("pseudo.values.count", req.getValues() == null ? 0 : req.getValues().size());
+        }
         log.info(Strings.padEnd(String.format("*** Pseudonymize field: %s ", req.getName()), 80, '*'));
         PseudoField pseudoField = new PseudoField(req.getName(), req.getPattern(), req.getPseudoFunc(), req.getKeyset());
         try {

src/main/java/no/ssb/dlp/pseudo/service/pseudo/PseudoField.java

@@ -1,7 +1,6 @@
 package no.ssb.dlp.pseudo.service.pseudo;
 
 import com.google.common.base.Stopwatch;
-import io.micronaut.tracing.annotation.ContinueSpan;
 import io.reactivex.Completable;
 import io.reactivex.Flowable;
 import lombok.AccessLevel;
@@ -81,7 +80,6 @@ public PseudoField(String name, String pattern, String pseudoFunc, EncryptedKeys
      * @param values                 The values to be processed.
      * @return A Flowable stream that processes the field values by applying the configured pseudo rules, and returns them as a lists of strings.
      */
-    @ContinueSpan
     public Flowable<String> process(PseudoConfigSplitter pseudoConfigSplitter,
                                           RecordMapProcessorFactory recordProcessorFactory,
                                           List<String> values,
@@ -182,4 +180,4 @@ protected Completable getPreprocessor(List<String> values, RecordMapProcessor<Ps
             return Completable.complete();
         }
     }
-}
+}

src/main/java/no/ssb/dlp/pseudo/service/pseudo/RecordMapProcessorFactory.java

@@ -66,6 +66,7 @@ public RecordMapProcessor<PseudoMetadataProcessor> newPseudonymizeRecordProcesso
         return new RecordMapProcessor<>(chain, metadataProcessor);
     }
 
+    @ContinueSpan
     public RecordMapProcessor<PseudoMetadataProcessor> newDepseudonymizeRecordProcessor(List<PseudoConfig> pseudoConfigs, String correlationId) {
         ValueInterceptorChain chain = new ValueInterceptorChain();
         PseudoMetadataProcessor metadataProcessor = new PseudoMetadataProcessor(correlationId);
@@ -80,6 +81,7 @@ public RecordMapProcessor<PseudoMetadataProcessor> newDepseudonymizeRecordProces
         return new RecordMapProcessor<>(chain, metadataProcessor);
     }
 
+    @ContinueSpan
     public RecordMapProcessor<PseudoMetadataProcessor> newRepseudonymizeRecordProcessor(PseudoConfig sourcePseudoConfig,
                                                                PseudoConfig targetPseudoConfig, String correlationId) {
         final PseudoFuncs fieldDepseudonymizer = newPseudoFuncs(sourcePseudoConfig.getRules(),