Example usage for org.springframework.cglib.proxy Enhancer Enhancer

List of usage examples for org.springframework.cglib.proxy Enhancer Enhancer

  1. HOME
  2. Java
  3. org.springframework
  4. org.springframework.cglib.proxy.*
  5. Enhancer
  6. Enhancer

Introduction

In this page you can find the example usage for org.springframework.cglib.proxy Enhancer Enhancer.

Prototype

public Enhancer()

Source Link

Document

Create a new Enhancer.

Usage

From source file:net.kaczmarzyk.spring.data.jpa.web.EnhancerUtil.java

@SuppressWarnings("unchecked")
static <T> T wrapWithIfaceImplementation(final Class<T> iface, final Specification<Object> targetSpec) {
    Enhancer enhancer = new Enhancer();
    enhancer.setInterfaces(new Class[] { iface });
    enhancer.setCallback(new MethodInterceptor() {
        @Override/*ww w. j  a  v  a  2s.com*/
        public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
            if ("toString".equals(method.getName())) {
                return iface.getSimpleName() + "[" + proxy.invoke(targetSpec, args) + "]";
            }
            return proxy.invoke(targetSpec, args);
        }
    });

    return (T) enhancer.create();
}

From source file:com.mongodb.DBCollectionProxyFactory.java

/** Get the enhanced DB collection from the provided one 
 * @param db db name//  w ww .  j  a  va  2 s .c o m
 * @param name collection name
 * @return the enhanced collection
 */
@SuppressWarnings("deprecation")
public static DBCollection get(final DB db, final String name, final boolean is_mock) {

    Enhancer collectionEnhancer = new Enhancer();
    collectionEnhancer
            .setSuperclass(is_mock ? com.mongodb.FongoDBCollection.class : com.mongodb.DBCollectionImpl.class);
    MethodInterceptor collectionMi = new MethodInterceptor() {
        boolean _top_level = true;

        @Override
        public Object intercept(final Object object, final Method method, final Object[] args,
                final MethodProxy methodProxy) throws Throwable {
            if (_top_level) {
                try {
                    _top_level = false;
                    for (int count = 0;; count++) {
                        //DEBUG
                        //System.out.println("intercepted method: " + method.toString() + ", loop=" + count + ");

                        try {
                            Object o = methodProxy.invokeSuper(object, args);

                            if (o instanceof DBCursor) {
                                o = getCursor((DBCursor) o);
                            }
                            return o;
                        } catch (com.mongodb.MongoException e) {
                            if (count < 60) {
                                continue;
                            }
                            throw e;
                        }
                    }
                } finally {
                    _top_level = true;
                }
            } else {
                return methodProxy.invokeSuper(object, args);
            }
        }

    };
    collectionEnhancer.setCallback(collectionMi);
    return (DBCollection) collectionEnhancer
            .create(is_mock ? new Class[] { com.mongodb.FongoDB.class, String.class }
                    : new Class[] { com.mongodb.DBApiLayer.class, String.class }, new Object[] { db, name });
}

From source file:org.reindeer.redis.jedis.JedisFactoryBean.java

@Override
public Jedis getObject() throws Exception {
    if (jedis != null) {
        return jedis;
    }//w ww  .ja va  2  s.c  o  m
    Enhancer en = new Enhancer();
    en.setSuperclass(Jedis.class);
    en.setCallbackFilter(finalizeFilter);
    en.setCallbacks(new Callback[] { NoOp.INSTANCE, jedisCallback });
    jedis = (Jedis) en.create(new Class[] { String.class }, new Object[] { "JedisProxy" });
    return jedis;
}

From source file:org.reindeer.redis.shard.ShardedJedisFactoryBean.java

@Override
public ShardedJedis getObject() throws Exception {
    if (jedis != null) {
        return jedis;
    }/*from www  . j  a  v a 2  s.co m*/
    Enhancer en = new Enhancer();
    en.setSuperclass(ShardedJedis.class);
    en.setCallbackFilter(finalizeFilter);
    en.setCallbacks(new Callback[] { NoOp.INSTANCE, jedisCallback });
    jedis = (ShardedJedis) en.create(new Class[] { List.class },
            new Object[] { Arrays.asList(new JedisShardInfo("shardedJedisProxy")) });
    return jedis;
}

From source file:cat.albirar.framework.proxy.ProxyFactory.java

/**
 * Create a proxy for the indicated type.
 * @param handler The handler//w  ww.  ja v a 2  s.c o m
 * @param type The type, should to be a concrete class type
 * @return The proxy
 */
@SuppressWarnings("unchecked")
private <T> T newProxyForConcreteClass(org.springframework.cglib.proxy.Callback handler, Class<T> type) {
    Enhancer enhancer;

    Assert.isTrue(!Modifier.isAbstract(type.getModifiers()), "The type should to be a concrete class");

    enhancer = new Enhancer();
    enhancer.setSuperclass(type);
    enhancer.setClassLoader(type.getClassLoader());
    enhancer.setCallback(handler);
    return (T) enhancer.create();
}

From source file:com.watchrabbit.executor.spring.ExecutorAnnotationBeanPostProcessor.java

private Enhancer createProxy(Class<?> targetClass, Executor classAnnotation,
        Map<Method, Executor> annotatedMethods, final Object bean) {
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(targetClass);
    enhancer.setCallback((InvocationHandler) (Object proxy, Method method, Object[] args) -> {
        if (annotatedMethods.containsKey(method) || classAnnotation != null) {
            Executor annotation = annotatedMethods.get(method);
            if (annotation != null) {
                return generateExecutor(annotation, method, bean, args);
            } else {
                return generateExecutor(classAnnotation, method, bean, args);
            }/* w ww  .ja v a2 s .  c o  m*/
        } else {
            return method.invoke(bean, args);
        }
    });
    return enhancer;
}

From source file:io.syndesis.runtime.Recordings.java

static public <T> T recorder(Object object, Class<T> as) {
    if (as.isInterface()) {
        // If it's just an interface, use standard java reflect proxying
        return as.cast(Proxy.newProxyInstance(as.getClassLoader(), new Class[] { as },
                new RecordingInvocationHandler(object)));
    }//from ww w. ja  v a2  s . co m

    // If it's a class then use gclib to implement a subclass to implement proxying
    RecordingInvocationHandler ih = new RecordingInvocationHandler(object);
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(as);
    enhancer.setInterfaces(new Class[] { RecordingProxy.class });
    enhancer.setCallback(new org.springframework.cglib.proxy.InvocationHandler() {
        @Override
        public Object invoke(Object o, Method method, Object[] objects) throws Throwable {
            return ih.invoke(o, method, objects);
        }
    });
    return as.cast(enhancer.create());
}

From source file:com.mongodb.DBCollectionProxyFactory.java

protected static DBCursor getCursor(final DBCursor from) {
    Enhancer dbcursorEnhancer = new Enhancer();
    dbcursorEnhancer.setSuperclass(com.mongodb.DBCursor.class);

    MethodInterceptor collectionMi = new MethodInterceptor() {
        boolean _top_level = true;

        @Override//from  www.j  av a  2s . c  o m
        public Object intercept(Object object, Method method, Object[] args, MethodProxy methodProxy)
                throws Throwable {
            if (_top_level) {
                try {
                    _top_level = false;
                    for (int count = 0;; count++) {
                        //DEBUG
                        //System.out.println("intercepted method: " + method.toString() + ", loop=" + count + ");

                        try {
                            Object o = methodProxy.invokeSuper(object, args);
                            return o;
                        } catch (com.mongodb.MongoException e) {
                            if (count < 60) {
                                continue;
                            }
                            throw e;
                        }
                    }
                } finally {
                    _top_level = true;
                }
            } else {
                return methodProxy.invokeSuper(object, args);
            }
        }
    };
    dbcursorEnhancer.setCallback(collectionMi);
    return (DBCursor) dbcursorEnhancer.create(
            new Class[] { DBCollection.class, DBObject.class, DBObject.class, ReadPreference.class },
            new Object[] { from.getCollection(), from.getQuery(), from.getKeysWanted(),
                    from.getReadPreference() });
}

From source file:com.newtranx.util.cassandra.spring.AccessorScannerConfigurer.java

@Override
public void postProcessBeanFactory(ConfigurableListableBeanFactory context) throws BeansException {
    ClassPathScanningCandidateComponentProvider scanner = new ClassPathScanningCandidateComponentProvider(
            false) {/*from  w w  w  .j a  va2 s. c o  m*/

        @Override
        protected boolean isCandidateComponent(AnnotatedBeanDefinition beanDefinition) {
            return beanDefinition.getMetadata().isInterface();
        }

    };
    scanner.addIncludeFilter(new AnnotationTypeFilter(Accessor.class));
    for (BeanDefinition bd : scanner.findCandidateComponents(basePackage)) {
        Class<?> accessorCls;
        try {
            accessorCls = Class.forName(bd.getBeanClassName());
        } catch (ClassNotFoundException e) {
            throw new AssertionError(e);
        }
        log.info("Creating proxy accessor: " + accessorCls.getName());
        MethodInterceptor interceptor = new MethodInterceptor() {

            private final Lazy<?> target = new Lazy<>(() -> {
                log.info("Creating actual accessor: " + accessorCls.getName());
                Session session;
                if (AccessorScannerConfigurer.this.session == null)
                    session = mainContext.getBean(Session.class);
                else
                    session = AccessorScannerConfigurer.this.session;
                MappingManager mappingManager = new MappingManager(session);
                return mappingManager.createAccessor(accessorCls);
            });

            @Override
            public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy)
                    throws Throwable {
                if ("toString".equals(method.getName())) {
                    return accessorCls.getName();
                }
                return method.invoke(target.get(), args);
            }

        };
        Enhancer enhancer = new Enhancer();
        enhancer.setInterfaces(new Class<?>[] { accessorCls });
        enhancer.setCallback(interceptor);
        Object bean = enhancer.create();
        String beanName = StringUtils.uncapitalize(accessorCls.getSimpleName());
        context.registerSingleton(beanName, bean);
        log.info("Bean registed, name=" + beanName + ", bean=" + bean.toString());
    }
}

From source file:com.newtranx.util.cassandra.spring.MapperScannerConfigurer.java

@SuppressWarnings("unchecked")
private static <T> T createProxy(final Class<?> classToMock, final MethodInterceptor interceptor) {
    final Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(classToMock);
    enhancer.setCallbackType(interceptor.getClass());
    final Class<?> proxyClass = enhancer.createClass();
    Enhancer.registerCallbacks(proxyClass, new Callback[] { interceptor });
    return (T) ObjenesisHelper.newInstance(proxyClass);
}