本博客来自我的新书Java性能优化(暂定名),第5章的Java代码优化技巧节选2,也欢迎阅读我的新书 《Spring Boot 2 精髓 》
5.2 高速Map存取
使用EnumMap来存取Key是Enum的,会有较快的速度,如下是一个网关返回对象Result的的状态属性,是一个枚举类
1 public static enum Status { 2 SUCCESS(1,"成功"),FAIL(2,"处理失败"),DEGRADE(98,"成功降级"),UNKOWN(99,"未知异常"); 3 4 private int code; 5 String msg; 6 Status(int code,String msg){ 7 this.code = code; 8 this.msg = msg; 9 } 10 11 public int getCode() { 12 return code; 13 } 14 15 public String getMsg() { 16 return msg; 17 } 18 }
考虑到定义微服务网关返回对象,应该尽量使用java自带类型,以避免各种序列化,反序列化问题,网关不返回此枚举值,而是返回msg字段,因此可以构造一个EnumMap,Key为Status枚举类型,Value为Status.msg 属性
1 Map<Status,String> enumMap =null; 2 private void initEnumMap(){ 3 enumMap = new EnumMap<Status,String>(Status.class); 4 for(Status status:Status.values()) { 5 enumMap.put(status,status.msg); 6 } 7 }
构造EnumMap的时候,内部实际上通过一个数组保存了所有的枚举值,索引是枚举的ordinal得到 当要根据Enum来操作EnumMap,只需要先调用ordinal,得到其索引,然后直接操作数组即可。操作一维数组有这最快的速度
有些场景下Key是int类型,这时候可以参考第二章的IntMap
有很多Key-Value使用场景,都可以转化为根据索引对数组的存取,我们学过的C语言,操作的是变量名,但实际上还是根据指针获取到内存中的值,Beetl模板语言,对变量的访问,也不像其他脚本语言那样,通过变量名访问Map获取其值,而是在编译期间就为这个变量分配好了索引值,所有变量都保存在一个一维数组里,这样的存取,相比于Map存取,有十倍以上性能提高。
如下一段脚本语言
1 var a = 1; 2 var b = 2+a;
有些语言引擎会翻译成类似如下java代码
1 context.put("a",1); 2 context.put("b",context.get("a")+2);
这里context是一个Map。从Map里通过Key存取尽管很快,但是Beetl还是在解析脚本语言的时候给变量设置了数组所在索引,因此如上脚本在Beetl中翻译如下
1 Object[] vars = context.vars; 2 vars[0] =1 ; 3 vars[1] = vars[0]+2
这里为变量a,b 分别设置了在变量表中的索引是0和1;
曾优化过一个电商的基础组件,电商系统每天调用这个组件的次数高达10+万亿次,这个组件是用来统计方法调用的时长,收集一段时间后,定期发送到 分析系统,用于查找和分析方法的性能,其中有一部分代码是记录以调用时长分类,记录条用次数,下面的代码
1 Watch watch = Watch.instance("orderByWx"); //初始化,从微信来的订单 2 //调用其他业务逻辑..... 3 Profile.add(watch.endWatch());//记录一次
Watch类定义如下
1 public class Watch { 2 String key; 3 long start; 4 long millis =-1; 5 private Watch(String key){ 6 this.key = key; 7 this.start = System.nanoTime(); 8 } 9 public static Watch instance(String key){ 10 return new Watch(key); 11 } 12 13 public Watch endWatch(){ 14 millis = millisConsume(); 15 return this; 16 } 17 18 /** 19 * 返回方法调用消耗的毫秒 20 * [@return](https://my.oschina.net/u/556800) 21 */ 22 private long millisConsume(){ 23 return TimeUnit.NANOSECONDS.toMillis(System.nanoTime()-start); 24 } 25 26 }
Watch的key属性记录调用类型,如订单调用,商品查询信息等,可以为任意值,start属性记录了调用时候的时间点,millis会在调用endWatch后记录调用消耗的毫秒数。
Profile类用来记录监控信息,并通过其他后台线程发送到性能分析中心,例子做了一定简化,只呈现保存部分
1 public class Profile { 2 //调用时长和调用次数 3 static Map<Integer, AtomicInteger> countMap = new ConcurrentHashMap<>(); 4 /** 5 * 对调用时间计数 6 * [@param](https://my.oschina.net/u/2303379) watch 7 */ 8 public static void addWatch(Watch watch){ 9 int consumeTime = (int)watch.millis; 10 AtomicInteger count = countMap.get(consumeTime); 11 if(count==null){ 12 count = new AtomicInteger(); 13 AtomicInteger old = countMap.putIfAbsent(consumeTime,count); 14 if(old!=null){ 15 count = old; 16 } 17 } 18 count.incrementAndGet(); 19 } 20 }
Profile会初始化一个ConcurrentHashMap用于计数,Key为Integer类型,表示调用时长,Value为AtomicInteger,用来计数,每次调用,都会自增一个
Profile性能有一点优化空间,如果从业务角度考虑,大部分需要监控的方法或者代码块,执行时间并不长,假设不超过32毫秒(这是一个假设值,根据系统运维统计分析后得出),因此,可以考虑用一个32长度的数组来存放32毫秒以内的所有计数,超过32毫秒的再沿用以前的方法
1 static Map<Integer, AtomicInteger> countMap = new ConcurrentHashMap<>(); 2 static final int MAX = 32; 3 //保存消耗时间为32毫秒的调用次数 4 static AtomicInteger[] counts = new AtomicInteger[MAX]; 5 static{ 6 for(int i=0;i<MAX;i++){ 7 counts[i] = new AtomicInteger(); 8 } 9 } 10 11 /** 12 * 对调用时间计数 13 * [@param](https://my.oschina.net/u/2303379) watch 14 */ 15 public static void addWatch(Watch watch){ 16 int consumeTime = (int)watch.millis; 17 if(consumeTime<MAX){ 18 counts[consumeTime].incrementAndGet(); 19 return ; 20 } 21 AtomicInteger count = countMap.get(consumeTime); 22 //原有的Profile.addWatch逻辑,在此忽略 23 24 }
新完善的代码使用counts数组记录32毫秒以内调用计数,因此当addWatch被调用的时候,先判断millis是否小于32毫秒,如果是,直接用数组获取计数器,然后自增。否则,沿用以前的逻辑
优化后,通过JMH测试(com.ibeetl.code.ch05.WatchTest),性能略有提升,如下输出:
1 Benchmark Mode Samples Score Score error Units 2 c.i.c.c.WatchTest.better thrpt 20 16447.171 2195.344 ops/ms 3 c.i.c.c.WatchTest.general thrpt 20 11566.545 601.579 ops/ms
性能优化提高了40%,尽管看着不如本书其他例子性能提升那么明显,但考虑这是一个基础工具,性能提升会对所有系统都有帮助,实际上,作者优化为跟此性能监控工具后,对业务系统的有提升5%的性能提升,在拥有数十万台服务器的大型电商系统,这个系统提升还是有意义的。