org.analogweb.netty.AnalogwebChannelInitializer Maven / Gradle / Ivy
package org.analogweb.netty;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.ssl.SslContext;
import io.netty.handler.ssl.util.SelfSignedCertificate;
import io.netty.util.concurrent.DefaultEventExecutorGroup;
import io.netty.util.concurrent.EventExecutorGroup;
import java.security.cert.CertificateException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import javax.net.ssl.SSLException;
import org.analogweb.Application;
import org.analogweb.ApplicationContext;
import org.analogweb.ApplicationProperties;
import org.analogweb.core.ApplicationRuntimeException;
import org.analogweb.util.Assertion;
import org.analogweb.util.ClassCollector;
import org.analogweb.util.FileClassCollector;
import org.analogweb.util.JarClassCollector;
import org.analogweb.util.StringUtils;
/**
* @author snowgooseyk
*/
public class AnalogwebChannelInitializer extends ChannelInitializer {
protected static final boolean SSL = System.getProperty("ssl") != null;
protected static final int DEFAULT_AGGREGATION_SIZE = 65535;
protected static final String MAX_AGGREGATION_SIZE = "analogweb.netty.max.aggregation.size";
private final SslContext sslCtx;
private final Application app;
private final ApplicationProperties properties;
private final EventExecutorGroup handlerSpecificExecutorGroup = new DefaultEventExecutorGroup(8);
public AnalogwebChannelInitializer(SslContext ssl, Application app,
ApplicationContext contextResolver, ApplicationProperties props) {
Assertion.notNull(app, Application.class.getName());
this.sslCtx = ssl == null ? resolveSslContext() : ssl;
this.properties = props;
this.app = app;
getApplication().run(contextResolver, getApplicationProperties(), getClassCollectors(),
getClassLoader());
}
protected SslContext resolveSslContext() {
final SslContext sslCtx;
if (SSL) {
try {
final SelfSignedCertificate ssc = new SelfSignedCertificate();
sslCtx = SslContext.newServerContext(ssc.certificate(), ssc.privateKey());
} catch (final SSLException e) {
throw new ApplicationRuntimeException(e) {
private static final long serialVersionUID = 1L;
};
} catch (final CertificateException e) {
throw new ApplicationRuntimeException(e) {
private static final long serialVersionUID = 1L;
};
}
} else {
sslCtx = null;
}
return sslCtx;
}
@Override
public void initChannel(SocketChannel ch) throws Exception {
final ChannelPipeline pipeline = ch.pipeline();
final SslContext ssl = getSslContext();
if (ssl != null) {
pipeline.addLast(ssl.newHandler(ch.alloc()));
}
pipeline.addLast(new HttpServerCodec());
pipeline.addLast(new HttpObjectAggregator(getMaxAggregationSize(getApplicationProperties())));
pipeline.addLast(getHandlerSpecificExecutorGroup(), createServerHandler());
}
private int getMaxAggregationSize(ApplicationProperties applicationProperties) {
String size = applicationProperties.getStringProperty(MAX_AGGREGATION_SIZE);
if (StringUtils.isNotEmpty(size)) {
try {
return Integer.parseInt(size);
} catch (NumberFormatException e) {
// nop.
}
}
return DEFAULT_AGGREGATION_SIZE;
}
protected ChannelHandler createServerHandler() {
return new AnalogwebChannelInboundHandler(getApplication(), getApplicationProperties());
}
protected SslContext getSslContext() {
return this.sslCtx;
}
protected Application getApplication() {
return this.app;
}
protected ApplicationProperties getApplicationProperties() {
return this.properties;
}
protected ClassLoader getClassLoader() {
return Thread.currentThread().getContextClassLoader();
}
protected EventExecutorGroup getHandlerSpecificExecutorGroup() {
return this.handlerSpecificExecutorGroup;
}
protected List getClassCollectors() {
final List list = new ArrayList();
list.add(new JarClassCollector());
list.add(new FileClassCollector());
return Collections.unmodifiableList(list);
}
}