package com.atlassian.vcache.internal.core;
   
   import com.atlassian.vcache.internal.RequestContext;
   
   import java.util.Map;
   import java.util.Optional;
   import java.util.concurrent.ConcurrentHashMap;
   import java.util.function.Supplier;
   
  import static java.util.Objects.requireNonNull;
  
  /**
   * An implementation of the {@link RequestContext}.
   *
   * @since 1.0
   */
  public class DefaultRequestContext implements RequestContext {
      private final String partitionId;
      private final Map<Object, Object> map = new ConcurrentHashMap<>();
  
      public DefaultRequestContext(String partitionId) {
          this.partitionId = requireNonNull(partitionId);
      }
  
      @Override
      public String partitionIdentifier() {
          return partitionId;
      }
  
      @SuppressWarnings("unchecked")
      @Override
      public <T> T computeIfAbsent(Object key, Supplier<T> supplier) {
          return (T) map.computeIfAbsent(requireNonNull(key), o -> requireNonNull(supplier.get()));
      }
  
      @SuppressWarnings("unchecked")
      @Override
      public <T> Optional<T> get(Object key) {
          return Optional.ofNullable((T) map.get(key));
      }
  }