背景
前提:抖音小程序有qps的监控,如果说qps过低就会导致小程序被下架掉。
业务代码非常的简单 一个easy的查询 但是当并非达到 20就 会发现qps降低了10倍
业务需求实现大概这么一个链路
ok 那么此前我们在认识一下 computeIfAbsent 方法(大佬可以跳过)
当我们想要在map中取值并判断,如果map中不存在那进行写入我们代码会怎么写?
不出意外的话 大家应该都会这么实现吧。但是实际上jdk早就提我们想好解决方案了。我们可以通过 computeIfAbsent 来进行实现。
让我们在来看一下跑出来的结果吧
ConcurrentHashMap.computeIfAbsent 方法性能问题
ok 话不多说 直接上测试代码
使用JMH来进行测试 代码如下
测试代码
package cn.ideamake.im.auth.service;import org.openjdk.jmh.annotations.Benchmark;
import org.openjdk.jmh.annotations.Fork;
import org.openjdk.jmh.annotations.Level;
import org.openjdk.jmh.annotations.Measurement;
import org.openjdk.jmh.annotations.Scope;
import org.openjdk.jmh.annotations.Setup;
import org.openjdk.jmh.annotations.State;
import org.openjdk.jmh.annotations.Warmup;
import org.openjdk.jmh.results.format.ResultFormatType;
import org.openjdk.jmh.runner.Runner;
import org.openjdk.jmh.runner.RunnerException;
import org.openjdk.jmh.runner.options.Options;
import org.openjdk.jmh.runner.options.OptionsBuilder;import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;/*** @author Barcke* @version 1.0* @projectName im-auth* @className Test* @date 2024/4/30 16:30* @slogan: 源于生活 高于生活* @description:**/
@Warmup(iterations = 3, time = 5)
@Fork(2)
@Measurement(iterations = 3, time = 5)
@State(Scope.Benchmark)
public class Test {private static final String KEY = "barcke";private static final Object VALUE = new Object();private final Map<String, Object> concurrentMap = new ConcurrentHashMap<>(1, 1);@Setup(Level.Iteration)public void setup() {concurrentMap.clear();}@Benchmarkpublic Object benchGetBeforeComputeIfAbsent() {Object result = concurrentMap.get(KEY);if (null == result) {result = concurrentMap.put(KEY, VALUE);}return result;}@Benchmarkpublic Object benchComputeIfAbsent() {return concurrentMap.computeIfAbsent(KEY, (k) -> VALUE);}public static void main(String[] args) throws RunnerException {Options opt = new OptionsBuilder().include(Test.class.getSimpleName()).result("result.json").resultFormat(ResultFormatType.JSON).build();new Runner(opt).run();}}
让我们直接看看平均运行时间吧!
可以看到 平均运行时间是提升了一倍!!
优化方案总结
1、升级JDK(1.9之后JDK已经处理了此问题)
2、主动调优通过util方法来处理 computeIfAbsent
public static <K, V> V computeIfAbsent(Map<K, V> map, K key, Function<? super K, ? extends V> mappingFunction) {V value = map.get(key);if (null == value) {map.putIfAbsent(key, mappingFunction.apply(key));value = map.get(key);}return value;}