/*
 * BaseDynamicDataSourceHandlerImpl.java: This class is the base class for
 * all the handler classes which need the dynamic DataSource support. In each
 * subclass handling method, it should invoke the "setDataSourceContext(...)"
 * method to switch the DataSource(by passing the "mallId" argument from
 * the handling method's arguments).
 */
public abstract class BaseDynamicDataSourceHandlerImpl {
    ...

    /**
     * Set the DataSource context according to the current mall.
     *
     * @param mallId
     */
    public void setDataSourceContext(Integer mallId) {
        ...
    }

    ...
}



/*
 * CustomerFlowHandlerImpl.java: A subclass of "BaseDynamicDataSourceHandlerImpl"
 * class.
 */
@Component
public class CustomerFlowHandlerImpl extends BaseDynamicDataSourceHandlerImpl implements CustomerFlowHandler {
    ...

    @Override
    public DataResult<String, Integer> getCustomerFlowTrendHistory(Integer mallId, CustomerType customerType,
                                                                   Integer floorId, Integer historyDayCount) {
        setDataSourceContext(mallId);
        ...
    }

    @Override
    public DataResult<String, Integer> getCustomerFlowTrendAverage(Integer mallId, CustomerType customerType,
                                                                   Integer floorId, Integer historyDayCount) {
        setDataSourceContext(mallId);
        ...
    }

    @Override
    public DataResult<String, Integer> getEnterShopCustomerAmountHistory(Integer mallId, CustomerType customerType,
                                                                         Integer floorId, Integer historyDayCount) {
        setDataSourceContext(mallId);
        ...
    }

    @Override
    public DataResult<String, Integer> getEnterShopCustomerAmountAverage(Integer mallId, CustomerType customerType,
                                                                         Integer floorId, Integer historyDayCount) {
        setDataSourceContext(mallId);
        ...
    }

    ...
}

/*
 * DynamicDataSourceAspect.java: The Aspect implementation which defines the pointcut
 * for each public method in any subclass of "BaseDynamicDataSourceHandlerImpl", and
 * makes the invocation of the "setDataSourceContext"method.
 */
@Component
@Aspect
public class DynamicDataSourceAspect {
    @Before("execution(public * com.jcloud.zhike.handler.impl.BaseDynamicDataSourceHandlerImpl+.*(..)) && args(mallId,..)")
    public void setDynamicDataSource(JoinPoint joinPoint, Integer mallId) {
        BaseDynamicDataSourceHandlerImpl target = (BaseDynamicDataSourceHandlerImpl)joinPoint.getTarget();
        target.setDataSourceContext(mallId);
    }
}



/*
 * CustomerFlowHandlerImpl.java: Now the "setDataSourceContext(...)" invocation lines can be removed.
 */
@Component
public class CustomerFlowHandlerImpl extends BaseDynamicDataSourceHandlerImpl implements CustomerFlowHandler {
    ...

    @Override
    public DataResult<String, Integer> getCustomerFlowTrendHistory(Integer mallId, CustomerType customerType,
                                                                   Integer floorId, Integer historyDayCount) {
        ...
    }

    @Override
    public DataResult<String, Integer> getCustomerFlowTrendAverage(Integer mallId, CustomerType customerType,
                                                                   Integer floorId, Integer historyDayCount) {
        ...
    }

    @Override
    public DataResult<String, Integer> getEnterShopCustomerAmountHistory(Integer mallId, CustomerType customerType,
                                                                         Integer floorId, Integer historyDayCount) {
        ...
    }

    @Override
    public DataResult<String, Integer> getEnterShopCustomerAmountAverage(Integer mallId, CustomerType customerType,
                                                                         Integer floorId, Integer historyDayCount) {
        ...
    }

    ...
}