空间换时间，轻松提高性能100倍

空间换时间的最常用场景就是缓存，为了提高性能可以设置不同类型的缓存。下面是我遇到的一个小问题。

问题描述

在统计股票的历史收益时，涉及到对交易日的操作，比如获取某个日期偏移一定天数后的交易日。由于交易日是比较固定的数据，在系统启动启动之初，就加载到缓存中，保存在一个ArrayList中，定义了一个方法getDate(String from, int offset)实现偏移取值。原始代码如下：

    public String getDateFast(String fromDate, int offset) {
        if(tradingList.isEmpty()){
            initialize();
        }

        int index = tradingList.indexOf(fromDate);
        if(index == -1){
            return null;
        }

        index -= offset;
        return tradingList.get(index);
    }

上述代码省略了和技术无关的业务细节。

在某一个业务场景下，需要对两万多条数据调用该方法，打印的日志如下：

从日志看出，两万条数据竟然花费了9秒多，这的是不可以接受的。

问题分析

因为tradingList是用ArrayList存储，那么每次的indexOf都是需要做一个全循环，所以indexOf操作是性能的瓶颈。

解决方法

找到了原因也就有解决办法。最快的查找是O(1)，符合这个条件的存储结构是HashMap。所以在初始化时，除了用ArrayList存储日期序列，用HashMap存储日期所在的索引。代码如下：

    private void initialize(){
        tradingList.clear();
        dateIndexMap.clear();

        //从数据库中加载交易日的序列
        tradingList = marketCalendarDao.findDateByIsTradingSortDateDesc(true);

        //建立日期和位置的对应关系
        for(int index = 0; index < tradingList.size(); index += 1){
            dateIndexMap.put(tradingList.get(index), index);
        }
    }

    public String getDateFast(String fromDate, int offset) {
        if(tradingList.isEmpty()){
            initialize();
        }

        if(dateIndexMap.containsKey(fromDate)){
            int index = dateIndexMap.get(fromDate);
            index -= offset;
            return tradingList.get(index);
        }
        return null;
    }

最后的运行效果：