io.r2dbc.proxy.callback

Class ConnectionFactoryCallbackHandler

java.lang.Object

io.r2dbc.proxy.callback.ConnectionFactoryCallbackHandler

All Implemented Interfaces:

CallbackHandler

public final class ConnectionFactoryCallbackHandler
extends Object

Proxy callback handler for ConnectionFactory.

Author:

Tadaya Tsuyukubo

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	CallbackHandlerSupport.MethodInvocationStrategy Strategy to invoke the original instance(non-proxy) and retrieve result.

Field Summary

Fields

Modifier and Type	Field and Description
protected static CallbackHandlerSupport.MethodInvocationStrategy	DEFAULT_INVOCATION_STRATEGY
protected CallbackHandlerSupport.MethodInvocationStrategy	methodInvocationStrategy
protected ProxyConfig	proxyConfig

Constructor Summary

Constructors

Constructor and Description
ConnectionFactoryCallbackHandler(io.r2dbc.spi.ConnectionFactory connectionFactory, ProxyConfig proxyConfig)

Method Summary

Modifier and Type	Method and Description
protected Object	handleCommonMethod(String methodName, Object original, Object[] args, io.r2dbc.spi.Connection originalConnection)
protected Flux<? extends io.r2dbc.spi.Result>	interceptQueryExecution(Publisher<? extends io.r2dbc.spi.Result> publisher, io.r2dbc.proxy.callback.MutableQueryExecutionInfo executionInfo) Augment query execution result to hook up listener lifecycle.
Object	invoke(Object proxy, Method method, Object[] args) When proxy is invoked, actual implementation of the proxy handler delegates the invocation to this method.
protected boolean	isCommonMethod(String methodName)
protected Object	proceedExecution(Method method, Object target, Object[] args, ProxyExecutionListener listener, ConnectionInfo connectionInfo, Consumer<MethodExecutionInfo> onComplete) Augment method invocation and call method listener.
void	setMethodInvocationStrategy(CallbackHandlerSupport.MethodInvocationStrategy methodInvocationStrategy) Set CallbackHandlerSupport.MethodInvocationStrategy to invoke the original instance(non-proxy) and retrieve result.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

DEFAULT_INVOCATION_STRATEGY

protected static final CallbackHandlerSupport.MethodInvocationStrategy DEFAULT_INVOCATION_STRATEGY

proxyConfig

protected final ProxyConfig proxyConfig

methodInvocationStrategy

protected CallbackHandlerSupport.MethodInvocationStrategy methodInvocationStrategy

Constructor Detail

ConnectionFactoryCallbackHandler

public ConnectionFactoryCallbackHandler(io.r2dbc.spi.ConnectionFactory connectionFactory,
                                        ProxyConfig proxyConfig)

Method Detail

invoke

public Object invoke(Object proxy,
                     Method method,
                     @Nullable
                     Object[] args)
              throws Throwable

Description copied from interface: CallbackHandler

When proxy is invoked, actual implementation of the proxy handler delegates the invocation to this method.

Parameters:

proxy - the proxy instance that the method was invoked on

method - the method that has invoked on the proxy instance

args - an array of objects that has passed to the method invocation. this can be null when method is invoked with no argument.

Returns:

result returned from the method invocation on the proxy instance. (can be null.)

Throws:

Throwable - the exception thrown from the method invocation on the proxy instance.

IllegalArgumentException - if proxy is null

isCommonMethod

protected boolean isCommonMethod(String methodName)

handleCommonMethod

@Nullable
protected Object handleCommonMethod(String methodName,
                                              Object original,
                                              @Nullable
                                              Object[] args,
                                              @Nullable
                                              io.r2dbc.spi.Connection originalConnection)

proceedExecution

protected Object proceedExecution(Method method,
                                  Object target,
                                  @Nullable
                                  Object[] args,
                                  ProxyExecutionListener listener,
                                  @Nullable
                                  ConnectionInfo connectionInfo,
                                  @Nullable
                                  Consumer<MethodExecutionInfo> onComplete)
                           throws Throwable

Augment method invocation and call method listener.

Parameters:

method - method to invoke on target

target - an object being invoked

args - arguments for the method. null if the method doesn't take any arguments.

listener - listener that before/after method callbacks will be called

connectionInfo - current connection information. null when invoked operation is not associated to the Connection.

onComplete - a callback that will be invoked at successful termination(onComplete) of the result publisher.

Returns:

result of invoking the original object

Throws:

Throwable - thrown exception during the invocation

IllegalArgumentException - if method is null

IllegalArgumentException - if target is null

IllegalArgumentException - if listener is null

interceptQueryExecution

protected Flux<? extends io.r2dbc.spi.Result> interceptQueryExecution(Publisher<? extends io.r2dbc.spi.Result> publisher,
                                                                      io.r2dbc.proxy.callback.MutableQueryExecutionInfo executionInfo)

Augment query execution result to hook up listener lifecycle.

Parameters:

publisher - query invocation result publisher

executionInfo - query execution context info

Returns:

query invocation result flux

Throws:

IllegalArgumentException - if flux is null

IllegalArgumentException - if executionInfo is null

setMethodInvocationStrategy

public void setMethodInvocationStrategy(CallbackHandlerSupport.MethodInvocationStrategy methodInvocationStrategy)

Set CallbackHandlerSupport.MethodInvocationStrategy to invoke the original instance(non-proxy) and retrieve result.

Parameters:

methodInvocationStrategy - strategy for method invocation

Throws:

IllegalArgumentException - if methodInvocationStrategy is null

See Also:

CallbackHandlerSupport.MethodInvocationStrategy