AOP使用举例

 1.背景

不同的日期查询数据，需要到不同的库去查：

 

本系统采用AOP的方法，自动切换数据源。

当前库：今天的数据

历史库：前三个月的数据

归档库：三个月之前的数据

 

 2.代码示例

本系统AOP代码：

（1）由ChangeDataSource.class进入AOP

@CloudComponent
@TargetDataSource(location = ChangeDataSource.class)
public class IOptionPortfolioStrategyTentrustServiceImpl implements IOptionPortfolioStrategyTentrustService {

    @Autowired
    private IOptionPortfolioStrategyTentrustMapper iOptionPortfolioStrategyTentrustMapper;


    /**
     * 日期类型切分
     */
    private static final String DATETYPE_1 = "1" ;
    private static final String DATETYPE_3 = "3" ;


    /**
     * 当前与历史查询
     * @param req
     * @return
     */
    @Override
    public PubstockOptionPortfolioStrategyTentrustResponse getTentrust(PubstockOptionPortfolioStrategyTentrustRequest req) {
        ValidUtils.notEmpty(req.getLoginCompany(), "公司序号不能为空");
        ValidUtils.notEmpty(req.getBeginDate(), "起始日期不能为空");
        ValidUtils.notEmpty(req.getEndDate(), "结束日期不能为空");
        PubstockOptionPortfolioStrategyTentrustResponse resp = new PubstockOptionPortfolioStrategyTentrustResponse();
        if (DATETYPE_1.equals(req.getDateType())) {
            MysqlPageHelper.startPage(req.getStartRow(), req.getPageSize(),false);
            List<OpubstockPtionPortfolioStrategyTentrust> rows = iOptionPortfolioStrategyTentrustMapper.tentrustQuery(req);
            MysqlPageHelper.endPage();
            // 添加判空操作
            if (rows != null && rows.size() > 0) {
                resp.setRows(rows);
            }
        } else {
            MysqlPageHelper.startPage(req.getStartRow(), req.getPageSize(),false);
            List<OpubstockPtionPortfolioStrategyTentrust> rows2 = iOptionPortfolioStrategyTentrustMapper.tentrustHisQuery(req);
            MysqlPageHelper.endPage();
            // 添加判空操作
            if (rows2 != null && rows2.size() > 0) {
                resp.setRows(rows2);
            }
        }
        return resp;
    }
}

根据begin_date,end_date 切换日期 end_date/business_date < 归档日cxhis， locationKey = dbarch .   否则 locationKey = default 默认数据库

public class ChangeDataSource implements DsLocationable{


    /**
     * Get locationkey string
     *
     * @param args
     * @return string
     */
    @Override
    public String getLocationkey(Object[] args) {
        // TODO Auto-generated method stub
        String locationKey = DataSourceConstants.DEFAULT;
        AresUtil aresUtil = SpringUtils.getBean(AresUtil.class);
        IFinancepubService financePubService = aresUtil.getFinancepubService();
        IAresService iAresService = aresUtil.getAresService();
        for (Object obj : args) {
            CxServiceAspect cxAspect = new CxServiceAspect();
            YhpubPretradedayGetRequest yhpubPretradedayGetRequest = new YhpubPretradedayGetRequest();
            yhpubPretradedayGetRequest.setMarketNo(5);
            YhpubPretradedayGetResponse yhpubPretradedayGetResponse = iAresService.pretradedayGet(yhpubPretradedayGetRequest);
            //获取上一个业务日期
            Integer prevBusinessDate = yhpubPretradedayGetResponse!=null?yhpubPretradedayGetResponse.getPrevBusinessDate():0;
            //查询日期小于等于cxhis 归档日，数据全部移动到了归档库
            BatclearPrebusindatecalcRequest req = new BatclearPrebusindatecalcRequest();
            req.setBusinessDate(prevBusinessDate);
            BatclearPrebusindatecalcResponse response = financePubService.prebusindatecalc(req);
            int cxhis = response.getPrevBusinessDate();
//            int cxhis = this.cxhisDate(prevBusinessDate);

            if (obj instanceof BaseDTO) {
                BaseDTO baseDto = (BaseDTO) obj;
                baseDto.setCxhis(cxhis);
            }
            /****** START报表查询逻辑处理START *********************/
            if (obj instanceof OcxCommonquery) {
                /** 1.数据权限控制 **/
                OcxCommonquery ocxCommonquery = (OcxCommonquery) obj;
                Integer companyId = ocxCommonquery.getCompanyId();

                /****** 3 数据源切换 START *********************/
                // 切换到归档数据查询
                int endDate = ocxCommonquery.getEndDate() == null ? 0 : ocxCommonquery.getEndDate().intValue();
                int beginDate = ocxCommonquery.getBeginDate() == null ? 0 : ocxCommonquery.getBeginDate().intValue();

                if (endDate > 0 && endDate <= cxhis) {
                    ocxCommonquery.setDateType("3");
                    locationKey = DataSourceConstants.DBARCH;
                }
                if (beginDate == 0 && endDate == 0) { // 如果只传入一个日期businessDate
                    Object value = cxAspect.getFieldValueByName("businessDate", obj);
                    // 如果入参中没有这个参数或者这个参数没有值,就默认查询当前库（dbtrade）
                    if (null == value) {
                        ocxCommonquery.setDateType("1");
                        // dbAspect.dataSourceExchange("1");
                        locationKey = DataSourceConstants.DEFAULT;
                    } else {
                        int businessDate = Integer.parseInt(String.valueOf(value));
                        if (businessDate > 0 && businessDate <= cxhis) {
                            ocxCommonquery.setDateType("3");
                            // dbAspect.dataSourceExchange("3");
                            locationKey = DataSourceConstants.DBARCH;
                        }
                    }
                }
            }

        }
        return locationKey;
    }
}