/**
*
* Copyright 2004 Hiram Chirino
*
* Licensed 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.
*
**/
package org.activeio.net;
import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import org.activeio.AsynchChannel;
import org.activeio.AsynchChannelFactory;
import org.activeio.AsynchChannelServer;
import org.activeio.adapter.SynchToAsynchChannelServerAdapter;
import org.activeio.filter.WriteBufferedAsynchChannel;
import org.activeio.packet.ByteBufferPacket;
/**
* A TcpAsynchChannelFactory creates {@see org.activeio.net.TcpAsynchChannel}
* and {@see org.activeio.net.TcpAsynchChannelServer} objects.
*
* @version $Revision$
*/
public class NIOAsynchChannelFactory implements AsynchChannelFactory {
protected static final int DEFAULT_BUFFER_SIZE = Integer.parseInt(System.getProperty("org.activeio.net.nio.BufferSize", ""+(64*1024)));
protected static final int DEFAULT_BACKLOG = 500;
boolean useDirectBuffers = true;
private final boolean createWriteBufferedChannels;
private int backlog = DEFAULT_BACKLOG;
public NIOAsynchChannelFactory() {
this(true);
}
public NIOAsynchChannelFactory(boolean createWriteBufferedChannels) {
this.createWriteBufferedChannels = createWriteBufferedChannels;
}
/**
* Uses the {@param location}'s host and port to create a tcp connection to a remote host.
*
* @see org.activeio.AsynchChannelFactory#openAsynchChannel(java.net.URI)
*/
public AsynchChannel openAsynchChannel(URI location) throws IOException {
SocketChannel channel = SocketChannel.open();
channel.connect(new InetSocketAddress(location.getHost(), location.getPort()));
return createAsynchChannel(channel);
}
/**
* @param channel
* @return
* @throws IOException
*/
protected AsynchChannel createAsynchChannel(SocketChannel socketChannel) throws IOException {
AsynchChannel channel = new NIOAsynchChannel(socketChannel, useDirectBuffers);
if( createWriteBufferedChannels ) {
channel = new WriteBufferedAsynchChannel(channel, ByteBufferPacket.createDefaultBuffer(useDirectBuffers), false);
}
return channel;
}
/**
* Binds a server socket a the {@param location}'s port.
*
* @see org.activeio.AsynchChannelFactory#bindAsynchChannel(java.net.URI)
*/
public AsynchChannelServer bindAsynchChannel(URI bindURI) throws IOException {
String host = bindURI.getHost();
InetSocketAddress address;
if ( host == null || host.length() == 0 || host.equals("localhost") || host.equals("0.0.0.0") ) {
address = new InetSocketAddress(bindURI.getPort());
} else {
address = new InetSocketAddress(bindURI.getHost(), bindURI.getPort());
}
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.socket().bind(address,backlog);
URI connectURI = bindURI;
try {
connectURI = URISupport.changeHost(connectURI, InetAddress.getLocalHost().getHostName());
connectURI = URISupport.changePort(connectURI, serverSocketChannel.socket().getLocalPort());
} catch (URISyntaxException e) {
throw (IOException)new IOException("Could not build connect URI: "+e).initCause(e);
}
// We won't use non blocking NIO for the server since you only need 1 thread for him anyways.
// Just resuing the SocketChannelSynchChannelServer.
return SynchToAsynchChannelServerAdapter.adapt(
new NIOAsynchChannelServer(serverSocketChannel, bindURI, connectURI, createWriteBufferedChannels, useDirectBuffers));
}
/**
* @return Returns the backlog.
*/
public int getBacklog() {
return backlog;
}
/**
* @param backlog
* The backlog to set.
*/
public void setBacklog(int backlog) {
this.backlog = backlog;
}
}