package org.apache.commons.jcs.auxiliary.lateral.socket.tcp;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
   * with the License.  You may obtain a copy of the License at
   *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  
  import java.io.EOFException;
  import java.io.IOException;
  import java.io.ObjectInputStream;
  import java.io.ObjectOutputStream;
  import java.io.Serializable;
  import java.net.InetAddress;
  import java.net.ServerSocket;
  import java.net.Socket;
  import java.net.SocketException;
  import java.net.SocketTimeoutException;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.Set;
  import java.util.concurrent.ExecutorService;
  import java.util.concurrent.Executors;
  import java.util.concurrent.atomic.AtomicBoolean;
  
  import org.apache.commons.jcs.access.exception.CacheException;
  import org.apache.commons.jcs.auxiliary.lateral.LateralElementDescriptor;
  import org.apache.commons.jcs.auxiliary.lateral.behavior.ILateralCacheListener;
  import org.apache.commons.jcs.auxiliary.lateral.socket.tcp.behavior.ITCPLateralCacheAttributes;
  import org.apache.commons.jcs.engine.CacheInfo;
  import org.apache.commons.jcs.engine.behavior.ICacheElement;
  import org.apache.commons.jcs.engine.behavior.ICompositeCacheManager;
  import org.apache.commons.jcs.engine.behavior.IShutdownObserver;
  import org.apache.commons.jcs.engine.control.CompositeCache;
  import org.apache.commons.jcs.engine.control.CompositeCacheManager;
  import org.apache.commons.jcs.io.ObjectInputStreamClassLoaderAware;
  import org.apache.commons.jcs.utils.threadpool.DaemonThreadFactory;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  
  /**
   * Listens for connections from other TCP lateral caches and handles them. The initialization method
   * starts a listening thread, which creates a socket server. When messages are received they are
   * passed to a pooled executor which then calls the appropriate handle method.
   */
  public class LateralTCPListener<K, V>
      implements ILateralCacheListener<K, V>, IShutdownObserver
  {
      /** The logger */
      private static final Log log = LogFactory.getLog( LateralTCPListener.class );
  
      /** How long the server will block on an accept(). 0 is infinite. */
      private static final int acceptTimeOut = 1000;
  
      /** The CacheHub this listener is associated with */
      private transient ICompositeCacheManager cacheManager;
  
      /** Map of available instances, keyed by port */
      private static final HashMap<String, ILateralCacheListener<?, ?>> instances =
          new HashMap<String, ILateralCacheListener<?, ?>>();
  
      /** The socket listener */
      private ListenerThread receiver;
  
      /** Configuration attributes */
      private ITCPLateralCacheAttributes tcpLateralCacheAttributes;
  
      /** Listening port */
      private int port;
  
      /** The processor. We should probably use an event queue here. */
      private ExecutorService pooledExecutor;
  
      /** put count */
      private int putCnt = 0;
  
      /** remove count */
      private int removeCnt = 0;
  
      /** get count */
      private int getCnt = 0;
  
      /**
       * Use the vmid by default. This can be set for testing. If we ever need to run more than one
       * per vm, then we need a new technique.
       */
     private long listenerId = CacheInfo.listenerId;
 
     /** is this shut down? */
     private AtomicBoolean shutdown;
 
     /** is this terminated? */
     private AtomicBoolean terminated;
 
     /**
      * Gets the instance attribute of the LateralCacheTCPListener class.
      * <p>
      * @param ilca ITCPLateralCacheAttributes
      * @param cacheMgr
      * @return The instance value
      */
     public synchronized static <K, V> LateralTCPListener<K, V>
         getInstance( ITCPLateralCacheAttributes ilca, ICompositeCacheManager cacheMgr )
     {
         @SuppressWarnings("unchecked") // Need to cast because of common map for all instances
         LateralTCPListener<K, V> ins = (LateralTCPListener<K, V>) instances.get( String.valueOf( ilca.getTcpListenerPort() ) );
 
         if ( ins == null )
         {
             ins = new LateralTCPListener<K, V>( ilca );
 
             ins.init();
             ins.setCacheManager( cacheMgr );
 
             instances.put( String.valueOf( ilca.getTcpListenerPort() ), ins );
 
             if ( log.isInfoEnabled() )
             {
                 log.info( "Created new listener " + ilca.getTcpListenerPort() );
             }
         }
 
         return ins;
     }
 
     /**
      * Only need one since it does work for all regions, just reference by multiple region names.
      * <p>
      * @param ilca
      */
     protected LateralTCPListener( ITCPLateralCacheAttributes ilca )
     {
         this.setTcpLateralCacheAttributes( ilca );
     }
 
     /**
      * This starts the ListenerThread on the specified port.
      */
     @Override
     public synchronized void init()
     {
         try
         {
             this.port = getTcpLateralCacheAttributes().getTcpListenerPort();
 
             pooledExecutor = Executors.newCachedThreadPool(
                     new DaemonThreadFactory("JCS-LateralTCPListener-"));
             terminated = new AtomicBoolean(false);
             shutdown = new AtomicBoolean(false);
 
             log.info( "Listening on port " + port );
 
             ServerSocket serverSocket = new ServerSocket( port );
             serverSocket.setSoTimeout( acceptTimeOut );
 
             receiver = new ListenerThread(serverSocket);
             receiver.setDaemon( true );
             receiver.start();
         }
         catch ( IOException ex )
         {
             throw new IllegalStateException( ex );
         }
     }
 
     /**
      * Let the lateral cache set a listener_id. Since there is only one listener for all the
      * regions and every region gets registered? the id shouldn't be set if it isn't zero. If it is
      * we assume that it is a reconnect.
      * <p>
      * By default, the listener id is the vmid.
      * <p>
      * The service should set this value. This value will never be changed by a server we connect
      * to. It needs to be non static, for unit tests.
      * <p>
      * The service will use the value it sets in all send requests to the sender.
      * <p>
      * @param id The new listenerId value
      * @throws IOException
      */
     @Override
     public void setListenerId( long id )
         throws IOException
     {
         this.listenerId = id;
         if ( log.isDebugEnabled() )
         {
             log.debug( "set listenerId = " + id );
         }
     }
 
     /**
      * Gets the listenerId attribute of the LateralCacheTCPListener object
      * <p>
      * @return The listenerId value
      * @throws IOException
      */
     @Override
     public long getListenerId()
         throws IOException
     {
         return this.listenerId;
     }
 
     /**
      * Increments the put count. Gets the cache that was injected by the lateral factory. Calls put
      * on the cache.
      * <p>
      * @see org.apache.commons.jcs.engine.behavior.ICacheListener#handlePut(org.apache.commons.jcs.engine.behavior.ICacheElement)
      */
     @Override
     public void handlePut( ICacheElement<K, V> element )
         throws IOException
     {
         putCnt++;
         if ( log.isInfoEnabled() )
         {
             if ( getPutCnt() % 100 == 0 )
             {
                 log.info( "Put Count (port " + getTcpLateralCacheAttributes().getTcpListenerPort() + ") = "
                     + getPutCnt() );
             }
         }
 
         if ( log.isDebugEnabled() )
         {
             log.debug( "handlePut> cacheName=" + element.getCacheName() + ", key=" + element.getKey() );
         }
 
         getCache( element.getCacheName() ).localUpdate( element );
     }
 
     /**
      * Increments the remove count. Gets the cache that was injected by the lateral factory. Calls
      * remove on the cache.
      * <p>
      * @see org.apache.commons.jcs.engine.behavior.ICacheListener#handleRemove(java.lang.String,
      *      Object)
      */
     @Override
     public void handleRemove( String cacheName, K key )
         throws IOException
     {
         removeCnt++;
         if ( log.isInfoEnabled() )
         {
             if ( getRemoveCnt() % 100 == 0 )
             {
                 log.info( "Remove Count = " + getRemoveCnt() );
             }
         }
 
         if ( log.isDebugEnabled() )
         {
             log.debug( "handleRemove> cacheName=" + cacheName + ", key=" + key );
         }
 
         getCache( cacheName ).localRemove( key );
     }
 
     /**
      * Gets the cache that was injected by the lateral factory. Calls removeAll on the cache.
      * <p>
      * @see org.apache.commons.jcs.engine.behavior.ICacheListener#handleRemoveAll(java.lang.String)
      */
     @Override
     public void handleRemoveAll( String cacheName )
         throws IOException
     {
         if ( log.isDebugEnabled() )
         {
             log.debug( "handleRemoveAll> cacheName=" + cacheName );
         }
 
         getCache( cacheName ).localRemoveAll();
     }
 
     /**
      * Gets the cache that was injected by the lateral factory. Calls get on the cache.
      * <p>
      * @param cacheName
      * @param key
      * @return a ICacheElement
      * @throws IOException
      */
     public ICacheElement<K, V> handleGet( String cacheName, K key )
         throws IOException
     {
         getCnt++;
         if ( log.isInfoEnabled() )
         {
             if ( getGetCnt() % 100 == 0 )
             {
                 log.info( "Get Count (port " + getTcpLateralCacheAttributes().getTcpListenerPort() + ") = "
                     + getGetCnt() );
             }
         }
 
         if ( log.isDebugEnabled() )
         {
             log.debug( "handleGet> cacheName=" + cacheName + ", key = " + key );
         }
 
         return getCache( cacheName ).localGet( key );
     }
 
     /**
      * Gets the cache that was injected by the lateral factory. Calls get on the cache.
      * <p>
      * @param cacheName the name of the cache
      * @param pattern the matching pattern
      * @return Map
      * @throws IOException
      */
     public Map<K, ICacheElement<K, V>> handleGetMatching( String cacheName, String pattern )
         throws IOException
     {
         getCnt++;
         if ( log.isInfoEnabled() )
         {
             if ( getGetCnt() % 100 == 0 )
             {
                 log.info( "GetMatching Count (port " + getTcpLateralCacheAttributes().getTcpListenerPort() + ") = "
                     + getGetCnt() );
             }
         }
 
         if ( log.isDebugEnabled() )
         {
             log.debug( "handleGetMatching> cacheName=" + cacheName + ", pattern = " + pattern );
         }
 
         return getCache( cacheName ).localGetMatching( pattern );
     }
 
     /**
      * Gets the cache that was injected by the lateral factory. Calls getKeySet on the cache.
      * <p>
      * @param cacheName the name of the cache
      * @return a set of keys
      * @throws IOException
      */
     public Set<K> handleGetKeySet( String cacheName ) throws IOException
     {
     	return getCache( cacheName ).getKeySet(true);
     }
 
     /**
      * This marks this instance as terminated.
      * <p>
      * @see org.apache.commons.jcs.engine.behavior.ICacheListener#handleDispose(java.lang.String)
      */
     @Override
     public void handleDispose( String cacheName )
         throws IOException
     {
         if ( log.isInfoEnabled() )
         {
             log.info( "handleDispose > cacheName=" + cacheName + " | Ignoring message.  Do not dispose from remote." );
         }
 
         // TODO handle active deregistration, rather than passive detection
         terminated.set(true);
     }
 
     @Override
     public synchronized void dispose()
     {
         terminated.set(true);
         notify();
 
         pooledExecutor.shutdownNow();
     }
 
     /**
      * Gets the cacheManager attribute of the LateralCacheTCPListener object.
      * <p>
      * Normally this is set by the factory. If it wasn't set the listener defaults to the expected
      * singleton behavior of the cache manager.
      * <p>
      * @param name
      * @return CompositeCache
      */
     protected CompositeCache<K, V> getCache( String name )
     {
         if ( getCacheManager() == null )
         {
             // revert to singleton on failure
             try
             {
                 setCacheManager( CompositeCacheManager.getInstance() );
             }
             catch (CacheException e)
             {
                 throw new RuntimeException("Could not retrieve cache manager instance", e);
             }
 
             if ( log.isDebugEnabled() )
             {
                 log.debug( "cacheMgr = " + getCacheManager() );
             }
         }
 
         return getCacheManager().getCache( name );
     }
 
     /**
      * This is roughly the number of updates the lateral has received.
      * <p>
      * @return Returns the putCnt.
      */
     public int getPutCnt()
     {
         return putCnt;
     }
 
     /**
      * @return Returns the getCnt.
      */
     public int getGetCnt()
     {
         return getCnt;
     }
 
     /**
      * @return Returns the removeCnt.
      */
     public int getRemoveCnt()
     {
         return removeCnt;
     }
 
     /**
      * @param cacheMgr The cacheMgr to set.
      */
     @Override
     public void setCacheManager( ICompositeCacheManager cacheMgr )
     {
         this.cacheManager = cacheMgr;
     }
 
     /**
      * @return Returns the cacheMgr.
      */
     @Override
     public ICompositeCacheManager getCacheManager()
     {
         return cacheManager;
     }
 
     /**
      * @param tcpLateralCacheAttributes The tcpLateralCacheAttributes to set.
      */
     public void setTcpLateralCacheAttributes( ITCPLateralCacheAttributes tcpLateralCacheAttributes )
     {
         this.tcpLateralCacheAttributes = tcpLateralCacheAttributes;
     }
 
     /**
      * @return Returns the tcpLateralCacheAttributes.
      */
     public ITCPLateralCacheAttributes getTcpLateralCacheAttributes()
     {
         return tcpLateralCacheAttributes;
     }
 
     /**
      * Processes commands from the server socket. There should be one listener for each configured
      * TCP lateral.
      */
     public class ListenerThread
         extends Thread
     {
         /** The socket listener */
         private final ServerSocket serverSocket;
 
         /**
          * Constructor
          *
          * @param serverSocket
          */
         public ListenerThread(ServerSocket serverSocket)
         {
             super();
             this.serverSocket = serverSocket;
         }
 
         /** Main processing method for the ListenerThread object */
         @SuppressWarnings("synthetic-access")
         @Override
         public void run()
         {
             try
             {
                 ConnectionHandler handler;
 
                 outer: while ( true )
                 {
                     if ( log.isDebugEnabled() )
                     {
                         log.debug( "Waiting for clients to connect " );
                     }
 
                     Socket socket = null;
                     inner: while (true)
                     {
                         // Check to see if we've been asked to exit, and exit
                         if (terminated.get())
                         {
                             if (log.isDebugEnabled())
                             {
                                 log.debug("Thread terminated, exiting gracefully");
                             }
                             break outer;
                         }
 
                         try
                         {
                             socket = serverSocket.accept();
                             break inner;
                         }
                         catch (SocketTimeoutException e)
                         {
                             // No problem! We loop back up!
                             continue inner;
                         }
                     }
 
                     if ( socket != null && log.isDebugEnabled() )
                     {
                         InetAddress inetAddress = socket.getInetAddress();
                         log.debug( "Connected to client at " + inetAddress );
                     }
 
                     handler = new ConnectionHandler( socket );
                     pooledExecutor.execute( handler );
                 }
             }
             catch ( IOException e )
             {
                 log.error( "Exception caught in TCP listener", e );
             }
             finally
             {
             	if (serverSocket != null)
             	{
             		try
             		{
 						serverSocket.close();
 					}
             		catch (IOException e)
             		{
                         log.error( "Exception caught closing socket", e );
 					}
             	}
             }
         }
     }
 
     /**
      * A Separate thread that runs when a command comes into the LateralTCPReceiver.
      */
     public class ConnectionHandler
         implements Runnable
     {
         /** The socket connection, passed in via constructor */
         private final Socket socket;
 
         /**
          * Construct for a given socket
          * @param socket
          */
         public ConnectionHandler( Socket socket )
         {
             this.socket = socket;
         }
 
         /**
          * Main processing method for the LateralTCPReceiverConnection object
          */
         @Override
         @SuppressWarnings({"unchecked", // Need to cast from Object
             "synthetic-access" })
         public void run()
         {
             ObjectInputStream ois;
 
             try
             {
                 ois = new ObjectInputStreamClassLoaderAware( socket.getInputStream(), null );
             }
             catch ( Exception e )
             {
                 log.error( "Could not open ObjectInputStream on " + socket, e );
 
                 return;
             }
 
             LateralElementDescriptor<K, V> led;
 
             try
             {
                 while ( true )
                 {
                     led = (LateralElementDescriptor<K, V>) ois.readObject();
 
                     if ( led == null )
                     {
                         log.debug( "LateralElementDescriptor is null" );
                         continue;
                     }
                     if ( led.requesterId == getListenerId() )
                     {
                         log.debug( "from self" );
                     }
                     else
                     {
                         if ( log.isDebugEnabled() )
                         {
                             log.debug( "receiving LateralElementDescriptor from another" + "led = " + led
                                 + ", led.command = " + led.command + ", led.ce = " + led.ce );
                         }
 
                         handle( led );
                     }
                 }
             }
             catch ( EOFException e )
             {
                 log.info( "Caught java.io.EOFException closing connection." + e.getMessage() );
             }
             catch ( SocketException e )
             {
                 log.info( "Caught java.net.SocketException closing connection." + e.getMessage() );
             }
             catch ( Exception e )
             {
                 log.error( "Unexpected exception.", e );
             }
 
             try
             {
                 ois.close();
             }
             catch ( IOException e )
             {
                 log.error( "Could not close object input stream.", e );
             }
         }
 
         /**
          * This calls the appropriate method, based on the command sent in the Lateral element
          * descriptor.
          * <p>
          * @param led
          * @throws IOException
          */
         @SuppressWarnings("synthetic-access")
         private void handle( LateralElementDescriptor<K, V> led )
             throws IOException
         {
             String cacheName = led.ce.getCacheName();
             K key = led.ce.getKey();
             Serializable obj = null;
 
             switch (led.command)
             {
                 case UPDATE:
                     handlePut( led.ce );
                     break;
 
                 case REMOVE:
                     // if a hashcode was given and filtering is on
                     // check to see if they are the same
                     // if so, then don't remove, otherwise issue a remove
                     if ( led.valHashCode != -1 )
                     {
                         if ( getTcpLateralCacheAttributes().isFilterRemoveByHashCode() )
                         {
                             ICacheElement<K, V> test = getCache( cacheName ).localGet( key );
                             if ( test != null )
                             {
                                 if ( test.getVal().hashCode() == led.valHashCode )
                                 {
                                     if ( log.isDebugEnabled() )
                                     {
                                         log.debug( "Filtering detected identical hashCode [" + led.valHashCode
                                             + "], not issuing a remove for led " + led );
                                     }
                                     return;
                                 }
                                 else
                                 {
                                     if ( log.isDebugEnabled() )
                                     {
                                         log.debug( "Different hashcodes, in cache [" + test.getVal().hashCode()
                                             + "] sent [" + led.valHashCode + "]" );
                                     }
                                 }
                             }
                         }
                     }
                     handleRemove( cacheName, key );
                     break;
 
                 case REMOVEALL:
                     handleRemoveAll( cacheName );
                     break;
 
                 case GET:
                     obj = handleGet( cacheName, key );
                     break;
 
                 case GET_MATCHING:
                     obj = (Serializable) handleGetMatching( cacheName, (String) key );
                     break;
 
                 case GET_KEYSET:
                 	obj = (Serializable) handleGetKeySet(cacheName);
                     break;
 
                 default: break;
             }
 
             if (obj != null)
             {
                 ObjectOutputStream oos = new ObjectOutputStream( socket.getOutputStream() );
                 oos.writeObject( obj );
                 oos.flush();
             }
         }
     }
 
     /**
      * Shuts down the receiver.
      */
     @Override
     public void shutdown()
     {
         if ( shutdown.compareAndSet(false, true) )
         {
             if ( log.isInfoEnabled() )
             {
                 log.info( "Shutting down TCP Lateral receiver." );
             }
 
             receiver.interrupt();
         }
         else
         {
             if ( log.isDebugEnabled() )
             {
                 log.debug( "Shutdown already called." );
             }
         }
     }
 }