/*
* Copyright (C) 2012 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.swift.service.async;
import com.facebook.nifty.client.FramedClientConnector;
import com.facebook.nifty.client.HttpClientConnector;
import com.facebook.swift.codec.ThriftCodecManager;
import com.facebook.swift.service.ThriftClient;
import com.facebook.swift.service.ThriftClientConfig;
import com.facebook.swift.service.ThriftClientManager;
import com.facebook.swift.service.ThriftServer;
import com.facebook.swift.service.ThriftServerConfig;
import com.facebook.swift.service.ThriftServiceProcessor;
import com.google.common.net.HostAndPort;
import com.google.common.util.concurrent.ListenableFuture;
import io.airlift.units.DataSize;
import io.airlift.units.Duration;
import org.apache.thrift.TException;
import org.apache.thrift.transport.TTransportException;
import java.net.URI;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
public class AsyncTestBase
{
public static final int MAX_FRAME_SIZE = 0x3fffffff;
protected ThriftCodecManager codecManager;
protected ThriftClientManager clientManager;
protected <T> ListenableFuture<T> createClient(Class<T> clientClass, ThriftServer server)
throws TTransportException, InterruptedException, ExecutionException
{
return createClient(clientClass, server.getPort());
}
protected <T> ListenableFuture<T> createClient(Class<T> clientClass, int serverPort)
throws TTransportException, InterruptedException, ExecutionException
{
HostAndPort address = HostAndPort.fromParts("localhost", serverPort);
ThriftClientConfig config = new ThriftClientConfig().setConnectTimeout(new Duration(1, TimeUnit.SECONDS))
.setReadTimeout(new Duration(1, TimeUnit.SECONDS))
.setWriteTimeout(new Duration(1, TimeUnit.SECONDS));
FramedClientConnector connector = new FramedClientConnector(address);
return new ThriftClient<>(clientManager, clientClass, config, "asyncTestClient").open(connector);
}
protected <T> ListenableFuture<T> createHttpClient(Class<T> clientClass, int serverPort)
throws TTransportException, InterruptedException, ExecutionException
{
ThriftClientConfig config = new ThriftClientConfig().setConnectTimeout(new Duration(1, TimeUnit.SECONDS))
.setReadTimeout(new Duration(1, TimeUnit.SECONDS))
.setWriteTimeout(new Duration(1, TimeUnit.SECONDS));
HttpClientConnector connector =
new HttpClientConnector(URI.create("http://localhost:" + serverPort + "/thrift/"));
return new ThriftClient<>(clientManager, clientClass, config, "asyncTestClient").open(connector);
}
protected ThriftServer createAsyncServer()
throws InstantiationException, IllegalAccessException, TException
{
DelayedMapAsyncHandler handler = new DelayedMapAsyncHandler();
handler.putValueSlowly(0, TimeUnit.MILLISECONDS, "testKey", "default");
return createServerFromHandler(handler);
}
protected ThriftServer createServerFromHandler(Object handler)
throws IllegalAccessException, InstantiationException
{
ThriftServiceProcessor processor = new ThriftServiceProcessor(codecManager, handler);
ThriftServerConfig config = new ThriftServerConfig();
config.setMaxFrameSize(new DataSize(MAX_FRAME_SIZE, DataSize.Unit.BYTE));
return new ThriftServer(processor, config).start();
}
}