net.bytebuddy.implementation.bind.annotation.SuperCall Java Examples

private static Object executeMethod(@SuperCall Callable<?> zuper, String uri, String src, String dst, String fsAction, String username) throws Exception {
    long startTime = System.nanoTime();
    DataAccessEventProtos.FsEvent.Status status = DataAccessEventProtos.FsEvent.Status.SUCCESS;
    try {
        Object result = zuper.call();
        if (Boolean.FALSE.equals(result)) {
            status = DataAccessEventProtos.FsEvent.Status.FAILURE;
        }
        return result;
    } catch (Exception e) {
        status = DataAccessEventProtos.FsEvent.Status.FAILURE;
        throw e;
    } finally {
        long elapsedTime = (System.nanoTime() - startTime) / NANOSECONDS_PER_MILLISECOND;
        sendFsEvent(uri, src, dst, fsAction, username, elapsedTime, status);
    }
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) Object dst) throws Exception {
    return callDistributedFileSystem(zuper, o, null, dst.toString(), FsAction.READ.name());
}
 

/**
 * The interception method to be applied.
 *
 * @param zuper A proxy to call the super method to validate the functioning og creating an auxiliary type.
 * @return The value to be returned by the instrumented {@link Object#toString()} method.
 * @throws Exception If an exception occurs.
 */
public static String intercept(@SuperCall Callable<String> zuper) throws Exception {
    String toString = zuper.call();
    if (toString.equals(FOO)) {
        throw new IllegalStateException("Super call proxy invocation did not derive in its value");
    }
    return FOO;
}
 

@RuntimeType
@BindingPriority( BindingPriority.DEFAULT * 3 )
public Object interceptSuper( @SuperCall final Callable<?> zuper, @This final Object receiver, @Origin Method method,
        @AllArguments final Object[] parameters,
        @FieldProxy( INTERCEPTOR_FIELD_NAME ) StepInterceptorGetterSetter stepInterceptorGetter )
        throws Throwable{

    StepInterceptor interceptor = (StepInterceptor) stepInterceptorGetter.getValue();

    if( interceptor == null ) {
        return zuper.call();
    }

    return interceptor.intercept( receiver, method, parameters, zuper::call );
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) Object dst) throws Exception {
    return callDistributedFileSystem(zuper, o, null, dst.toString(), FsAction.DELETE.name());
}
 

@RuntimeType
public static Object intercept(@Origin Method method,
                               @SuperCall Callable<?> callable) throws Exception {
    long start = System.nanoTime();
    try {
        // 原有函数执行
        return callable.call();
    } finally {
        PerformanceLogger.printMethodUsedTime(method.getDeclaringClass().getName() + "#" + method.getName(), (System.nanoTime() - start));
    }
}
 

private static Object callDistributedFileSystem(@SuperCall Callable<?> zuper, @This Object o, String src, String dst, String fsAction) throws Exception {
    ClassLoader classLoader = o.getClass().getClassLoader();
    Field dfsField = getField(classLoader, o.getClass().getName(), "dfs");
    Object dfs = dfsField.get(o);
    Field ugiField = getField(classLoader, "org.apache.hadoop.hdfs.DFSClient", "ugi");
    Object ugi = ugiField.get(dfs);
    Method getShortUserName = getMethod(classLoader, "org.apache.hadoop.security.UserGroupInformation", "getShortUserName");
    Method getUri = getMethod(classLoader, o.getClass().getName(), "getUri");
    return executeMethod(zuper, getUri.invoke(o).toString(), src, dst, fsAction, (String) getShortUserName.invoke(ugi));
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) String dst) throws Exception {

    ClassLoader classLoader = o.getClass().getClassLoader();
    Field field = getField(classLoader, "org.apache.hadoop.hdfs.protocolPB.ClientNamenodeProtocolTranslatorPB", "rpcProxy");
    Object rpcProxy = field.get(o);

    Method getServerAddress = getMethod(classLoader, "org.apache.hadoop.ipc.RPC", "getServerAddress", Object.class);
    InetSocketAddress inetSocketAddress = (InetSocketAddress) getServerAddress.invoke(o, rpcProxy);
    return executeMethod(zuper, "hdfs://" + inetSocketAddress.getHostString() + ":" + inetSocketAddress.getPort(),
        null, dst, FsAction.ADD_BLOCK.name(), null);
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) Object dst) throws Exception {
    return callDistributedFileSystem(zuper, o, null, dst.toString(), FsAction.GET_CONTENT_SUMMARY.name());
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) Object dst) throws Exception {
    return callDistributedFileSystem(zuper, o, null, dst.toString(), FsAction.LIST_STATUS.name());
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) Object dst) throws Exception {
    return callDistributedFileSystem(zuper, o, null, dst.toString(), FsAction.APPEND.name());
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) Object dst) throws Exception {
    return callDistributedFileSystem(zuper, o, null, dst.toString(), FsAction.WRITE.name());
}
 

@RuntimeType
public static Object intercept(
    @SuperCall Callable<?> zuper,
    @This Object o,
    @Argument(0) Object src,
    @Argument(1) Object dst) throws Exception {
    return callDistributedFileSystem(zuper, o, src.toString(), dst.toString(), FsAction.RENAME.name());
}
 

public static String intercept(@SuperCall Callable<String> zuper) throws Exception {
    return zuper.call() + BAR;
}
 

@RuntimeType
public static Object foo(@SuperCall(fallbackToDefault = false) Callable<?> zuper) throws Exception {
    return null;
}
 

public static String bar(@SuperCall(serializableProxy = true) Callable<String> callable) throws Exception {
    assertThat(callable, instanceOf(Serializable.class));
    return callable.call();
}
 

public static String bar(@SuperCall String value) throws Exception {
    return value;
}
 

@RuntimeType
public static Object foo(@SuperCall Callable<?> zuper) throws Exception {
    return zuper.call();
}
 

public static Object foo(@SuperCall Callable<?> zuper) throws Exception {
    return zuper.call();
}
 

public static String bar(@SuperCall Callable<String> callable) throws Exception {
    assertThat(callable, CoreMatchers.not(instanceOf(Serializable.class)));
    return callable.call();
}
 

public static void foo(@SuperCall Runnable runnable) {
    assertThat(runnable, CoreMatchers.not(instanceOf(Serializable.class)));
    runnable.run();
}
 

public String intercept(@SuperCall Callable<String> zuper) throws Exception {
    return zuper.call() + BAR;
}
 

@SuppressWarnings("unused")
@RuntimeType
public Object invoke(@SuperCall Callable<Object> superMethod, @AllArguments @RuntimeType Object[] args) throws Throwable {
    return superMethod.call();
}