package com.hoo.framework.spring.interceptor;

import java.lang.reflect.Proxy;

import org.aopalliance.intercept.MethodInterceptor;

import org.aopalliance.intercept.MethodInvocation;

import org.apache.commons.lang.ClassUtils;

import org.springframework.beans.factory.InitializingBean;

import com.hoo.framework.log.ApplicationLogging;
 

import com.hoo.framework.spring.support.CustomerContextHolder;

/**

 * <b>function:</b> 動態設定資料來源攔截器

 * @author hoojo

 * @createDate 2013-9-27 下午02:00:13

 * @file DataSourceMethodInterceptor.java

 * @package com.hoo.framework.spring.interceptor

 * @project SHMB

 * @blog http://blog.csdn.net/IBM_hoojo
 

 * @email [email protected]

 * @version 1.0

 */

public class DataSourceMethodInterceptor extends ApplicationLogging implements MethodInterceptor, InitializingBean {

    @Override

    public Object invoke(MethodInvocation invocation) throws Throwable {

        Class<?> clazz = invocation.getThis().getClass();

        String className = clazz.getName();

        if (ClassUtils.isAssignable(clazz, Proxy.class)) {

            className = invocation.getMethod().getDeclaringClass().getName();

        }

        String methodName = invocation.getMethod().getName();

        Object[] arguments = invocation.getArguments();

        trace("execute {}.{}({})", className, methodName, arguments);

        if (className.contains("MySQL")) {

            CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_MYSQL);

        } else if (className.contains("Oracle")) {

            CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_ORACLE);

        } else if (methodName.contains("MySQL")) {

            CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_MYSQL);

        } else if (methodName.contains("Oracle")) {

            CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_ORACLE);

        } else {

            CustomerContextHolder.clearCustomerType();

        }

        /*

        if (className.contains("MySQL") || methodName.contains("MySQL")) {

            CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_MYSQL);

        } else if (className.contains("Oracle") || methodName.contains("Oracle")) {

            CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_ORACLE);

        } else {

            CustomerContextHolder.clearCustomerType();

        }

        */

        Object result = invocation.proceed();

        return result;

    }

    @Override

    public void afterPropertiesSet() throws Exception {

        log.trace("afterPropertiesSet……");

    }

}