Java ThreadLocal 妙用

前言

最近公司需要做一个功能：导出报表的数据到 Excel 中，要求按报表内容分多个 sheet 导出；我用开源的 easy-excel 导出数据，easy-excel 写excel 的时候，设置多 sheet 时有两个属性：sheetNo 和 sheetName ，分别表示 sheet 号 和 sheet 名称；easy-excel 框架要求这两个属性不能重复，如果重复则会覆盖掉重复的那个 sheet 。

sheetNo 取值

sheetNo 是一个 Integer ，如果解决重复可以通过引入 for 循环通过遍历的下标来解决，这种固然很好，但若想用 Java 8 提供的函数式边城就哑火了，可以考虑放到 ThreadLocal 里 ，每次从 ThreadLocal 取，没有就从 0 开始，有就加 1 放到 ThreadLocal 返回，下面附上伪代码：

   private Integer getSheetNo(){
        Integer sheetNo = ThreadLocalUtils.get(key);
        sheetNo = Objects.isNull(sheetNo) ? 0 : ++sheetNo;
        ThreadLocalUtils.put(key, sheetNo);
        return sheetNo;
    }

sheetName 重复解决

在我的场景中我组成的 sheetName 会重复的，产品经理给的要求是从第一个重复的 sheetName 后面加序号，从 1 开始，这对我们来说想到的解决方案: 借用 Map 来实现，但是这会显式写在我们的代码里，如果你的方法是嵌套的，你可能需要从方法入参出传进来，解决方案同样我们可以借用 ThreadLocal 解决，毕竟 ThreadLocal 根本上也是一个 map, 解决方案: 先按照业务规则生成 sheetName ，然后判断是否在 ThreadLocal 有重复：若没有则就用原 sheetName，把当前 sheetName 作为key, 0 为 value 放入 ThreadLocal ；若有就用sheetName + 从 ThreadLocal 取出的序号+1，把序号加 1 放入ThreadLocal 。下面是伪代码：

public String getSheetName(String sourceSheetName){
    Integer index = ThreadLocalUtils.get(prefix + sourceSheetName);
    if(Objects.isNull(index)){
        index = 0;
        ThreadLocalUtils.put(prefix + sourceSheetName, index);
    } else {
        ThreadLocalUtils.put(prefix + sourceSheetName, ++index)
    }
    return index;
}

ps: 一般情况下我们都是用工程里封装好的上下文工具，无需担心用完未清理，但是若是自己新建的 ThreadLocal 记得在请求或者用完 clear 一下，防止数据污染以及内存泄漏的危害