package com.atlassian.vcache.internal.core;
   
   import java.util.HashMap;
   import java.util.Map;
   import java.util.Optional;
   
   import com.atlassian.vcache.internal.RequestContext;
   
   import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  import static java.util.Objects.requireNonNull;
  
  /**
   * Default implementation of {@link TransactionControlManager}.
   */
  public class DefaultTransactionControlManager implements TransactionControlManager
  {
      private static final Logger log = LoggerFactory.getLogger(DefaultTransactionControlManager.class);
      private final Object myKey = new Object();
  
      private final Instrumentor instrumentor;
  
      public DefaultTransactionControlManager(Instrumentor instrumentor)
      {
          this.instrumentor = requireNonNull(instrumentor);
      }
  
      @Override
      public void registerTransactionalExternalCache(RequestContext requestContext, String cacheName, TransactionControl control)
      {
          final Map<String, TransactionControl> map = requestContext.computeIfAbsent(myKey, HashMap::new);
          if (!map.containsKey(cacheName))
          {
              log.trace("Registering {}", cacheName);
              map.put(cacheName, instrumentor.wrap(control, cacheName));
          }
      }
  
      @Override
      public void syncAll(RequestContext requestContext)
      {
          log.trace("Synchronising all");
          final Optional<Map<String, TransactionControl>> optionalMap = requestContext.get(myKey);
          if (optionalMap.isPresent())
          {
              optionalMap.get().entrySet().forEach(e -> {
                  log.trace("Syncing {}", e.getKey());
                  e.getValue().transactionSync();
              });
          }
      }
  }