src/org/sonews/daemon/ConnectionWorker.java
author František Kučera <franta-hg@frantovo.cz>
Sat, 08 Oct 2011 23:32:18 +0200
changeset 63 d883d4ab7b9d
parent 37 74139325d305
permissions -rwxr-xr-x
Drupal: kostry tříd
chris@1
     1
/*
chris@1
     2
 *   SONEWS News Server
chris@1
     3
 *   see AUTHORS for the list of contributors
chris@1
     4
 *
chris@1
     5
 *   This program is free software: you can redistribute it and/or modify
chris@1
     6
 *   it under the terms of the GNU General Public License as published by
chris@1
     7
 *   the Free Software Foundation, either version 3 of the License, or
chris@1
     8
 *   (at your option) any later version.
chris@1
     9
 *
chris@1
    10
 *   This program is distributed in the hope that it will be useful,
chris@1
    11
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
chris@1
    12
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
chris@1
    13
 *   GNU General Public License for more details.
chris@1
    14
 *
chris@1
    15
 *   You should have received a copy of the GNU General Public License
chris@1
    16
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
chris@1
    17
 */
chris@1
    18
chris@1
    19
package org.sonews.daemon;
chris@1
    20
chris@1
    21
import org.sonews.util.Log;
chris@1
    22
import java.nio.ByteBuffer;
chris@1
    23
import java.nio.channels.SocketChannel;
chris@1
    24
import java.util.concurrent.ArrayBlockingQueue;
chris@1
    25
chris@1
    26
/**
chris@1
    27
 * Does most of the work: parsing input, talking to client and Database.
chris@1
    28
 * @author Christian Lins
chris@1
    29
 * @since sonews/0.5.0
chris@1
    30
 */
chris@1
    31
class ConnectionWorker extends AbstractDaemon
chris@1
    32
{
chris@1
    33
cli@37
    34
	// 256 pending events should be enough
cli@37
    35
	private static ArrayBlockingQueue<SocketChannel> pendingChannels = new ArrayBlockingQueue<SocketChannel>(256, true);
chris@1
    36
cli@37
    37
	/**
cli@37
    38
	 * Registers the given channel for further event processing.
cli@37
    39
	 * @param channel
cli@37
    40
	 */
cli@37
    41
	public static void addChannel(SocketChannel channel)
cli@37
    42
		throws InterruptedException
cli@37
    43
	{
cli@37
    44
		pendingChannels.put(channel);
cli@37
    45
	}
chris@1
    46
cli@37
    47
	/**
cli@37
    48
	 * Processing loop.
cli@37
    49
	 */
cli@37
    50
	@Override
cli@37
    51
	public void run()
cli@37
    52
	{
cli@37
    53
		while (isRunning()) {
cli@37
    54
			try {
cli@37
    55
				// Retrieve and remove if available, otherwise wait.
cli@37
    56
				SocketChannel channel = pendingChannels.take();
cli@37
    57
cli@37
    58
				if (channel != null) {
cli@37
    59
					// Connections.getInstance().get() MAY return null
cli@37
    60
					NNTPConnection conn = Connections.getInstance().get(channel);
cli@37
    61
cli@37
    62
					// Try to lock the connection object
cli@37
    63
					if (conn != null && conn.tryReadLock()) {
cli@37
    64
						ByteBuffer buf = conn.getBuffers().nextInputLine();
cli@37
    65
						while (buf != null) // Complete line was received
cli@37
    66
						{
cli@37
    67
							final byte[] line = new byte[buf.limit()];
cli@37
    68
							buf.get(line);
cli@37
    69
							ChannelLineBuffers.recycleBuffer(buf);
cli@37
    70
cli@37
    71
							// Here is the actual work done
cli@37
    72
							conn.lineReceived(line);
cli@37
    73
cli@37
    74
							// Read next line as we could have already received the next line
cli@37
    75
							buf = conn.getBuffers().nextInputLine();
cli@37
    76
						}
cli@37
    77
						conn.unlockReadLock();
cli@37
    78
					} else {
cli@37
    79
						addChannel(channel);
cli@37
    80
					}
cli@37
    81
				}
cli@37
    82
			} catch (InterruptedException ex) {
cli@37
    83
				Log.get().info("ConnectionWorker interrupted: " + ex);
cli@37
    84
			} catch (Exception ex) {
cli@37
    85
				Log.get().severe("Exception in ConnectionWorker: " + ex);
cli@37
    86
				ex.printStackTrace();
cli@37
    87
			}
cli@37
    88
		} // end while(isRunning())
cli@37
    89
	}
chris@1
    90
}