package com.atlassian.logging;
   
   import com.atlassian.workcontext.api.ImmutableWorkContextReference;
   import com.atlassian.workcontext.api.WorkContextAvailable;
   
   import javax.annotation.Nonnull;
   import javax.annotation.Nullable;
   import java.util.Optional;
   
  /**
   * Provides a data exchange point for logging data that is be associated with the WorkContext lifecycle.
   */
  public class WorkContextLoggingData {
  
      private static ImmutableWorkContextReference<Optional<String>> traceId = new ImmutableWorkContextReference<>();
      private static ImmutableWorkContextReference<Optional<String>> tenantId = new ImmutableWorkContextReference<>();
  
      /**
       * Returns an Optional that will contain the traceId if it has been set, otherwise it will return {@link Optional#empty}.
       *
       * @return An {@link Optional} containing the traceId or {@link Optional#empty}.
       */
      @Nonnull
      public static Optional<String> getTraceId() {
          if (WorkContextAvailable.isWorkContextAvailableStrict() && traceId.isPresent()) {
              return traceId.get();
          }
          return Optional.empty();
      }
  
      /**
       * Set the traceId for this WorkContext. Can be null to indicate that there is no traceId available.
       *
       * @param traceId The traceId associated with this WorkContext.
       */
      public static void setTraceId(@Nullable String traceId) {
          if (WorkContextAvailable.isWorkContextAvailableStrict()) {
              WorkContextLoggingData.traceId.initialise(Optional.ofNullable(traceId));
          }
      }
  
      /**
       * Returns an {@link String} containing the tenantId. If the tenantId has not been set, then we will return
       * {@link Optional#empty}.
       *
       * @return The actual tenantId or {@link Optional#empty}
       */
      @Nonnull
      public static Optional<String> getTenantId() {
          if (WorkContextAvailable.isWorkContextAvailableStrict() && tenantId.isPresent()) {
              return tenantId.get();
          }
          // We have not set the tenantId, as opposed to those times where the tenantId is set, but not present.
          return Optional.empty();
      }
  
      /**
       * Set the tenantId.
       *
       * @param tenantId The tenantId that is associated with this WorkContext. If null is passed, then the WorkContext
       *                 is considered to be explicitly untenanted.
       */
      public static void setTenantId(@Nullable String tenantId) {
          if (WorkContextAvailable.isWorkContextAvailableStrict()) {
              WorkContextLoggingData.tenantId.initialise(Optional.ofNullable(tenantId));
          }
      }
  }