package com.atlassian.cache.ehcache.cluster;
   
   import java.util.Date;
   import java.util.List;
   import java.util.Properties;
   
   import com.google.common.collect.ImmutableList;
   
   import net.sf.ehcache.CacheException;
  import net.sf.ehcache.CacheManager;
  import net.sf.ehcache.Ehcache;
  import net.sf.ehcache.distribution.CachePeer;
  import net.sf.ehcache.distribution.RMICacheManagerPeerProvider;
  
  import static com.atlassian.cache.ehcache.cluster.EhCacheConfigurationFactory.BASE_PORT;
  import static com.atlassian.cache.ehcache.cluster.EhCacheConfigurationFactory.LOCALHOST;
  
  /**
   * @since v2.4.5
   */
  public class EhCachePeerProvider extends RMICacheManagerPeerProvider
  {
      static final String THIS_NODE_ID = "thisNodeId";
      static final String MAX_NODE_ID = "maxNodeId";
  
      private final int thisNodeId;
      private final int maxNodeId;
  
      public EhCachePeerProvider(CacheManager cacheManager, Properties properties)
      {
          super(cacheManager);
          this.thisNodeId = Integer.parseInt(properties.getProperty(THIS_NODE_ID));
          this.maxNodeId = Integer.parseInt(properties.getProperty(MAX_NODE_ID));
      }
  
      @Override
      public void init()
      {
      }
  
      @Override
      public long getTimeForClusterToForm()
      {
          return 0;
      }
  
      @Override
      public final void registerPeer(String rmiUrl)
      {
          //Nothing to do here
      }
  
      @Override
      public List<CachePeer> listRemoteCachePeers(Ehcache cache) throws CacheException
      {
          final ImmutableList.Builder<CachePeer> list = ImmutableList.builder();
          for (int otherNodeId = 1; otherNodeId <= maxNodeId; ++otherNodeId)
          {
              if (thisNodeId != otherNodeId)
              {
                  try
                  {
                      final String rmiUrl = buildBaseUrl(otherNodeId, cache.getName());
                      list.add(lookupRemoteCachePeer(rmiUrl));
                  }
                  catch (Exception e)
                  {
                      System.err.println("No binding from node" + thisNodeId + " to node" + otherNodeId +
                              " for cache '" + cache.getName() + '\'');
                  }
              }
          }
          return list.build();
      }
  
      @Override
      protected boolean stale(Date date)
      {
          return false;
      }
  
      private static String buildBaseUrl(final int otherNodeId, final String cacheName)
      {
          final int port = BASE_PORT + otherNodeId;
          return "//" + LOCALHOST + ':' + port + '/' + cacheName;
      }
  
  }