/*
* Copyright (C) 2012-2013 Facebook, Inc.
*
* 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 com.facebook.nifty.core;
import com.facebook.nifty.codec.DefaultThriftFrameCodecFactory;
import com.facebook.nifty.codec.ThriftFrameCodecFactory;
import com.facebook.nifty.duplex.TDuplexProtocolFactory;
import com.facebook.nifty.processor.NiftyProcessor;
import com.facebook.nifty.processor.NiftyProcessorFactory;
import io.airlift.units.Duration;
import org.apache.thrift.TProcessor;
import org.apache.thrift.TProcessorFactory;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocolFactory;
import org.apache.thrift.transport.TTransport;
import java.util.concurrent.Executor;
import java.util.concurrent.atomic.AtomicInteger;
import static com.facebook.nifty.processor.NiftyProcessorAdapters.factoryFromTProcessorFactory;
import static com.google.common.base.Preconditions.checkState;
/**
* Builder for the Thrift Server descriptor. Example :
* <code>
* new ThriftServerDefBuilder()
* .listen(config.getServerPort())
* .limitFrameSizeTo(config.getMaxFrameSize())
* .withProcessor(new FacebookService.Processor(new MyFacebookBase()))
* .using(Executors.newFixedThreadPool(5))
* .build();
* <p/>
* <p/>
* You can then pass ThriftServerDef to guice via a multibinder.
* <p/>
* </code>
*/
public class ThriftServerDefBuilder
{
private static final AtomicInteger ID = new AtomicInteger(1);
private ThriftFrameCodecFactory thriftFrameCodecFactory;
private int serverPort;
private int maxFrameSize;
private int maxConnections;
private int queuedResponseLimit;
private NiftyProcessorFactory niftyProcessorFactory;
private TProcessorFactory thriftProcessorFactory;
private TDuplexProtocolFactory duplexProtocolFactory;
private Executor executor;
private String name = "nifty-" + ID.getAndIncrement();
private Duration clientIdleTimeout;
/**
* The default maximum allowable size for a single incoming thrift request or outgoing thrift
* response. A server can configure the actual maximum to be much higher (up to 0x7FFFFFFF or
* almost 2 GB). This default could also be safely bumped up, but 64MB is chosen simply
* because it seems reasonable that if you are sending requests or responses larger than
* that, it should be a conscious decision (something you must manually configure).
*/
private static final int MAX_FRAME_SIZE = 64 * 1024 * 1024;
/**
* Create a ThriftServerDefBuilder with common defaults
*/
public ThriftServerDefBuilder()
{
this.serverPort = 8080;
this.maxFrameSize = MAX_FRAME_SIZE;
this.maxConnections = 0;
this.queuedResponseLimit = 16;
this.duplexProtocolFactory = TDuplexProtocolFactory.fromSingleFactory(new TBinaryProtocol.Factory(true, true));
this.executor = new Executor()
{
@Override
public void execute(Runnable runnable)
{
runnable.run();
}
};
this.clientIdleTimeout = null;
this.thriftFrameCodecFactory = new DefaultThriftFrameCodecFactory();
}
/**
* Give the endpoint a more meaningful name.
*/
public ThriftServerDefBuilder name(String name)
{
this.name = name;
return this;
}
/**
* Listen to this port.
*/
public ThriftServerDefBuilder listen(int serverPort)
{
this.serverPort = serverPort;
return this;
}
/**
* Specify protocolFactory for both input and output
*/
public ThriftServerDefBuilder protocol(TDuplexProtocolFactory tProtocolFactory)
{
this.duplexProtocolFactory = tProtocolFactory;
return this;
}
public ThriftServerDefBuilder protocol(TProtocolFactory tProtocolFactory)
{
this.duplexProtocolFactory = TDuplexProtocolFactory.fromSingleFactory(tProtocolFactory);
return this;
}
/**
* Specify the TProcessor.
*/
public ThriftServerDefBuilder withProcessor(final NiftyProcessor processor)
{
this.niftyProcessorFactory = new NiftyProcessorFactory() {
@Override
public NiftyProcessor getProcessor(TTransport transport)
{
return processor;
}
};
return this;
}
public ThriftServerDefBuilder withProcessor(TProcessor processor)
{
this.thriftProcessorFactory = new TProcessorFactory(processor);
return this;
}
/**
* Anohter way to specify the TProcessor.
*/
public ThriftServerDefBuilder withProcessorFactory(NiftyProcessorFactory processorFactory)
{
this.niftyProcessorFactory = processorFactory;
return this;
}
/**
* Anohter way to specify the TProcessor.
*/
public ThriftServerDefBuilder withProcessorFactory(TProcessorFactory processorFactory)
{
this.thriftProcessorFactory = processorFactory;
return this;
}
/**
* Set frame size limit. Default is MAX_FRAME_SIZE
*/
public ThriftServerDefBuilder limitFrameSizeTo(int maxFrameSize)
{
this.maxFrameSize = maxFrameSize;
return this;
}
/**
* Set maximum number of connections. Default is 0 (unlimited)
*/
public ThriftServerDefBuilder limitConnectionsTo(int maxConnections)
{
this.maxConnections = maxConnections;
return this;
}
/**
* Limit number of queued responses per connection, before pausing reads
* to catch up.
*/
public ThriftServerDefBuilder limitQueuedResponsesPerConnection(int queuedResponseLimit)
{
this.queuedResponseLimit = queuedResponseLimit;
return this;
}
/**
* Specify timeout during which if connected client doesn't send a message, server
* will disconnect the client
*/
public ThriftServerDefBuilder clientIdleTimeout(Duration clientIdleTimeout)
{
this.clientIdleTimeout = clientIdleTimeout;
return this;
}
public ThriftServerDefBuilder thriftFrameCodecFactory(ThriftFrameCodecFactory thriftFrameCodecFactory)
{
this.thriftFrameCodecFactory = thriftFrameCodecFactory;
return this;
}
/**
* Specify an executor for thrift processor invocations ( i.e. = THaHsServer )
* By default invocation happens in Netty single thread
* ( i.e. = TNonBlockingServer )
*/
public ThriftServerDefBuilder using(Executor exe)
{
this.executor = exe;
return this;
}
/**
* Build the ThriftServerDef
*/
public ThriftServerDef build()
{
checkState(niftyProcessorFactory != null || thriftProcessorFactory != null,
"Processor not defined!");
checkState(niftyProcessorFactory == null || thriftProcessorFactory == null,
"TProcessors will be automatically adapted to NiftyProcessors, don't specify both");
checkState(maxConnections >= 0, "maxConnections should be 0 (for unlimited) or positive");
if (niftyProcessorFactory == null)
{
niftyProcessorFactory = factoryFromTProcessorFactory(thriftProcessorFactory);
}
return new ThriftServerDef(
name,
serverPort,
maxFrameSize,
queuedResponseLimit,
maxConnections,
niftyProcessorFactory,
duplexProtocolFactory,
clientIdleTimeout,
thriftFrameCodecFactory,
executor);
}
}