io.r2dbc.proxy.callback

Class ConnectionCallbackHandler

java.lang.Object

io.r2dbc.proxy.callback.ConnectionCallbackHandler

All Implemented Interfaces:

CallbackHandler

public final class ConnectionCallbackHandler
extends Object

Proxy callback handler for Connection.

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
ConnectionCallbackHandler(io.r2dbc.spi.Connection connection, ConnectionInfo connectionInfo, 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

ConnectionCallbackHandler

public ConnectionCallbackHandler(io.r2dbc.spi.Connection connection,
                                 ConnectionInfo connectionInfo,
                                 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