MSPF-6: Builder type fixes to make thrift endpoint creation more handy

This commit is contained in:
Vladimir Pankrashkin 2016-06-02 11:33:09 +03:00
parent ab0284d36a
commit 863f3e775e
2 changed files with 22 additions and 0 deletions

View File

@ -1,6 +1,7 @@
package com.rbkmoney.woody.thrift.impl.http; package com.rbkmoney.woody.thrift.impl.http;
import com.rbkmoney.woody.api.AbstractClientBuilder; import com.rbkmoney.woody.api.AbstractClientBuilder;
import com.rbkmoney.woody.api.ClientBuilder;
import com.rbkmoney.woody.api.event.ClientEventListener; import com.rbkmoney.woody.api.event.ClientEventListener;
import com.rbkmoney.woody.api.generator.IdGenerator; import com.rbkmoney.woody.api.generator.IdGenerator;
import com.rbkmoney.woody.api.interceptor.CommonInterceptor; import com.rbkmoney.woody.api.interceptor.CommonInterceptor;
@ -28,6 +29,7 @@ import org.apache.thrift.transport.TTransport;
import java.lang.reflect.Constructor; import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException; import java.lang.reflect.InvocationTargetException;
import java.net.URI;
import java.util.Arrays; import java.util.Arrays;
import java.util.Optional; import java.util.Optional;
@ -49,6 +51,21 @@ public class THClientBuilder extends AbstractClientBuilder {
return this; return this;
} }
@Override
public THClientBuilder withAddress(URI address) {
return (THClientBuilder) super.withAddress(address);
}
@Override
public THClientBuilder withEventListener(ClientEventListener listener) {
return (THClientBuilder) super.withEventListener(listener);
}
@Override
public THClientBuilder withIdGenerator(IdGenerator generator) {
return (THClientBuilder) super.withIdGenerator(generator);
}
@Override @Override
protected MethodCallTracer getOnCallMetadataExtender(Class iface) { protected MethodCallTracer getOnCallMetadataExtender(Class iface) {
return new EmptyTracer() { return new EmptyTracer() {

View File

@ -1,6 +1,7 @@
package com.rbkmoney.woody.thrift.impl.http; package com.rbkmoney.woody.thrift.impl.http;
import com.rbkmoney.woody.api.AbstractServiceBuilder; import com.rbkmoney.woody.api.AbstractServiceBuilder;
import com.rbkmoney.woody.api.ServiceBuilder;
import com.rbkmoney.woody.api.event.ServiceEventListener; import com.rbkmoney.woody.api.event.ServiceEventListener;
import com.rbkmoney.woody.api.interceptor.CommonInterceptor; import com.rbkmoney.woody.api.interceptor.CommonInterceptor;
import com.rbkmoney.woody.api.interceptor.CompositeInterceptor; import com.rbkmoney.woody.api.interceptor.CompositeInterceptor;
@ -30,6 +31,10 @@ import java.util.Optional;
*/ */
public class THServiceBuilder extends AbstractServiceBuilder<Servlet> { public class THServiceBuilder extends AbstractServiceBuilder<Servlet> {
@Override
public THServiceBuilder withEventListener(ServiceEventListener listener) {
return (THServiceBuilder) super.withEventListener(listener);
}
@Override @Override
protected MethodCallTracer getOnCallMetadataExtender(Class serviceInterface) { protected MethodCallTracer getOnCallMetadataExtender(Class serviceInterface) {