chris@1: /*
chris@1:  *   SONEWS News Server
chris@1:  *   see AUTHORS for the list of contributors
chris@1:  *
chris@1:  *   This program is free software: you can redistribute it and/or modify
chris@1:  *   it under the terms of the GNU General Public License as published by
chris@1:  *   the Free Software Foundation, either version 3 of the License, or
chris@1:  *   (at your option) any later version.
chris@1:  *
chris@1:  *   This program is distributed in the hope that it will be useful,
chris@1:  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
chris@1:  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
chris@1:  *   GNU General Public License for more details.
chris@1:  *
chris@1:  *   You should have received a copy of the GNU General Public License
chris@1:  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
chris@1:  */
chris@1: 
chris@1: package org.sonews.daemon;
chris@1: 
chris@3: import org.sonews.config.Config;
chris@1: import org.sonews.util.Log;
chris@1: import java.io.IOException;
chris@1: import java.net.InetSocketAddress;
chris@1: import java.net.Socket;
chris@1: import java.nio.channels.SocketChannel;
chris@1: import java.util.ArrayList;
chris@1: import java.util.HashMap;
chris@1: import java.util.List;
chris@1: import java.util.ListIterator;
chris@1: import java.util.Map;
chris@1: import org.sonews.util.Stats;
chris@1: 
chris@1: /**
chris@1:  * Daemon thread collecting all NNTPConnection instances. The thread
chris@1:  * checks periodically if there are stale/timed out connections and
chris@1:  * removes and purges them properly.
chris@1:  * @author Christian Lins
chris@1:  * @since sonews/0.5.0
chris@1:  */
chris@3: public final class Connections extends AbstractDaemon
chris@1: {
chris@1: 
chris@1:   private static final Connections instance = new Connections();
chris@1:   
chris@1:   /**
chris@1:    * @return Active Connections instance.
chris@1:    */
chris@1:   public static Connections getInstance()
chris@1:   {
chris@1:     return Connections.instance;
chris@1:   }
chris@1:   
chris@1:   private final List<NNTPConnection> connections 
chris@1:     = new ArrayList<NNTPConnection>();
chris@1:   private final Map<SocketChannel, NNTPConnection> connByChannel 
chris@1:     = new HashMap<SocketChannel, NNTPConnection>();
chris@1:   
chris@1:   private Connections()
chris@1:   {
chris@1:     setName("Connections");
chris@1:   }
chris@1:   
chris@1:   /**
chris@1:    * Adds the given NNTPConnection to the Connections management.
chris@1:    * @param conn
chris@1:    * @see org.sonews.daemon.NNTPConnection
chris@1:    */
chris@1:   public void add(final NNTPConnection conn)
chris@1:   {
chris@1:     synchronized(this.connections)
chris@1:     {
chris@1:       this.connections.add(conn);
chris@3:       this.connByChannel.put(conn.getSocketChannel(), conn);
chris@1:     }
chris@1:   }
chris@1:   
chris@1:   /**
chris@1:    * @param channel
chris@1:    * @return NNTPConnection instance that is associated with the given
chris@1:    * SocketChannel.
chris@1:    */
chris@1:   public NNTPConnection get(final SocketChannel channel)
chris@1:   {
chris@1:     synchronized(this.connections)
chris@1:     {
chris@1:       return this.connByChannel.get(channel);
chris@1:     }
chris@1:   }
chris@1: 
chris@1:   int getConnectionCount(String remote)
chris@1:   {
chris@1:     int cnt = 0;
chris@1:     synchronized(this.connections)
chris@1:     {
chris@1:       for(NNTPConnection conn : this.connections)
chris@1:       {
chris@1:         assert conn != null;
chris@3:         assert conn.getSocketChannel() != null;
chris@1: 
chris@3:         Socket socket = conn.getSocketChannel().socket();
chris@1:         if(socket != null)
chris@1:         {
chris@1:           InetSocketAddress sockAddr = (InetSocketAddress)socket.getRemoteSocketAddress();
chris@1:           if(sockAddr != null)
chris@1:           {
chris@1:             if(sockAddr.getHostName().equals(remote))
chris@1:             {
chris@1:               cnt++;
chris@1:             }
chris@1:           }
chris@1:         } // if(socket != null)
chris@1:       }
chris@1:     }
chris@1:     return cnt;
chris@1:   }
chris@1:   
chris@1:   /**
chris@1:    * Run loops. Checks periodically for timed out connections and purged them
chris@1:    * from the lists.
chris@1:    */
chris@1:   @Override
chris@1:   public void run()
chris@1:   {
chris@1:     while(isRunning())
chris@1:     {
chris@3:       int timeoutMillis = 1000 * Config.inst().get(Config.TIMEOUT, 180);
chris@1:       
chris@1:       synchronized (this.connections)
chris@1:       {
chris@1:         final ListIterator<NNTPConnection> iter = this.connections.listIterator();
chris@1:         NNTPConnection conn;
chris@1: 
chris@1:         while (iter.hasNext())
chris@1:         {
chris@1:           conn = iter.next();
chris@1:           if((System.currentTimeMillis() - conn.getLastActivity()) > timeoutMillis)
chris@1:           {
chris@1:             // A connection timeout has occurred so purge the connection
chris@1:             iter.remove();
chris@1: 
chris@1:             // Close and remove the channel
chris@3:             SocketChannel channel = conn.getSocketChannel();
chris@1:             connByChannel.remove(channel);
chris@1:             
chris@1:             try
chris@1:             {
chris@1:               // Close the channel; implicitely cancels all selectionkeys
chris@1:               channel.close();
chris@1:               Log.msg("Disconnected: " + channel.socket().getRemoteSocketAddress() +
chris@1:                 " (timeout)", true);
chris@1:             }
chris@1:             catch(IOException ex)
chris@1:             {
chris@1:               Log.msg("Connections.run(): " + ex, false);
chris@1:             }
chris@1: 
chris@1:             // Recycle the used buffers
chris@1:             conn.getBuffers().recycleBuffers();
chris@1:             
chris@1:             Stats.getInstance().clientDisconnect();
chris@1:           }
chris@1:         }
chris@1:       }
chris@1: 
chris@1:       try
chris@1:       {
chris@1:         Thread.sleep(10000); // Sleep ten seconds
chris@1:       }
chris@1:       catch(InterruptedException ex)
chris@1:       {
chris@1:         Log.msg("Connections Thread was interrupted: " + ex.getMessage(), false);
chris@1:       }
chris@1:     }
chris@1:   }
chris@1:   
chris@1: }