package com.atlassian.sal.core.net;
   
   import org.apache.http.HttpHost;
   import org.apache.http.auth.AuthScheme;
   import org.apache.http.client.AuthCache;
   import org.apache.http.impl.client.BasicAuthCache;
   
   /**
    * An AuthCache implementation allowing setting Auth for a host on all ports
   *
   * When getting a value for a host, we try first to find exact match and then
   * a match for all ports
   *
   * @since v3.0
   */
  class AllPortsAuthCache extends BasicAuthCache implements AuthCache {
      @Override
      protected HttpHost getKey(final HttpHost host) {
          return host;
      }
  
      @Override
      public AuthScheme get(final HttpHost host) {
          // looking for exact match
          AuthScheme authScheme = super.get(host);
  
          if (authScheme != null) {
              return authScheme;
          }
  
          // Exact match not found, looking for all ports match
          return super.get(new HttpHost(host.getHostName()));
      }
  }