百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程网 > 正文

switch 的性能提升了 3 倍,我只用了这一招

yuyutoo 2024-12-22 21:45 1 浏览 0 评论

上一篇《if快还是switch快?解密switch背后的秘密》我们测试了 if 和 switch 的性能,得出了要尽量使用 switch 的结论,因为他的效率比 if 高很多,具体原因点击上文查看。

既然 switch 如此有魅力,那么有没有更好的方法,让 switch 变得更快一些呢

答案是有的,不然本文就不会诞生了不是?

在上篇 if 和 switch 性能对比的文章中有读者问到:String 类型的 switch 性能是否也比 if 高?先说答案,String 类型的条件判断 switch 的性能依旧比 if 好

口说无凭,先举个,测试代码如下:

import org.openjdk.jmh.annotations.*;
import org.openjdk.jmh.infra.Blackhole;
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.concurrent.TimeUnit;

@BenchmarkMode(Mode.AverageTime) // 测试完成时间
@OutputTimeUnit(TimeUnit.NANOSECONDS)
@Warmup(iterations = 2, time = 1, timeUnit = TimeUnit.SECONDS) // 预热 2 轮,每次 1s
@Measurement(iterations = 5, time = 3, timeUnit = TimeUnit.SECONDS) // 测试 5 轮,每次 3s
@Fork(1) // fork 1 个线程
@State(Scope.Thread) // 每个测试线程一个实例
public class SwitchOptimizeByStringTest {
    static String _STR = "Java中文社群";
    public static void main(String[] args) throws RunnerException {
        // 启动基准测试
        Options opt = new OptionsBuilder()
                .include(SwitchOptimizeByStringTest.class.getSimpleName()) // 要导入的测试类
                .build();
        new Runner(opt).run(); // 执行测试
    }

    @Benchmark
    public void switchTest(Blackhole blackhole) {
        String s1;
        switch (_STR) {
            case "java":
                s1 = "java";
                break;
            case "mysql":
                s1 = "mysql";
                break;
            case "oracle":
                s1 = "oracle";
                break;
            case "redis":
                s1 = "redis";
                break;
            case "mq":
                s1 = "mq";
                break;
            case "kafka":
                s1 = "kafka";
                break;
            case "rabbitmq":
                s1 = "rabbitmq";
                break;
            default:
                s1 = "default";
                break;
        }
        // 为了避免 JIT 忽略未被使用的结果计算,可以使用 Blackhole#consume 来保证方法被正常执行
        blackhole.consume(s1);
    }

    @Benchmark
    public void ifTest(Blackhole blackhole) {
        String s1;
        if ("java".equals(_STR)) {
            s1 = "java";
        } else if ("mysql".equals(_STR)) {
            s1 = "mysql";
        } else if ("oracle".equals(_STR)) {
            s1 = "oracle";
        } else if ("redis".equals(_STR)) {
            s1 = "redis";
        } else if ("mq".equals(_STR)) {
            s1 = "mq";
        } else if ("kafka".equals(_STR)) {
            s1 = "kafka";
        } else if ("rabbitmq".equals(_STR)) {
            s1 = "rabbitmq";
        } else {
            s1 = "default";
        }
        // 为了避免 JIT 忽略未被使用的结果计算,可以使用 Blackhole#consume 来保证方法被正常执行
        blackhole.consume(s1);
    }
}

特殊说明:本文使用的是 Oracle 官方提供的性能测试工具 JMH(Java Microbenchmark Harness,JAVA 微基准测试套件)进行测试的。

以上代码测试的结果如下:


从 Score 列(平均完成时间)可以看出 switch 的性能依旧比 if 的性能要高。

备注:本文的测试环境为:JDK 1.8 / Mac mini (2018) / Idea 2020.1

switch 性能优化

我们知道在 JDK 1.7 之前 switch 是不支持 String 的,实际上 switch 只支持 int 类型

在 JDK 1.7 中的 String 类型,其实在编译的时候会使用 hashCode 来作为 switch 的实际值,以上 switch 判断字符串的代码,编译为字节码实际结果如下:

public static void switchTest() {
    String var1 = _STR;
    byte var2 = -1;
    switch(var1.hashCode()) {
        case -1008861826:
            if (var1.equals("oracle")) {
                var2 = 2;
            }
            break;
        case -95168706:
            if (var1.equals("rabbitmq")) {
                var2 = 6;
            }
            break;
        case 3492:
            if (var1.equals("mq")) {
                var2 = 4;
            }
            break;
        case 3254818:
            if (var1.equals("java")) {
                var2 = 0;
            }
            break;
        case 101807910:
            if (var1.equals("kafka")) {
                var2 = 5;
            }
            break;
        case 104382626:
            if (var1.equals("mysql")) {
                var2 = 1;
            }
            break;
        case 108389755:
            if (var1.equals("redis")) {
                var2 = 3;
            }
    }
    // 忽略其他代码...
}

知道了 switch 实现的本质,那么优化就变得比较简单了。

从以上的字节码可以看出,如果要优化 switch 只需要把 String 类型变成 int 类型就可以了,这样就剩了每个 case 中进行 if 判断的性能消耗,最终的优化代码如下:

public void switchHashCodeTest() {
    String s1;
    switch (_STR.hashCode()) {
        case 3254818:
            s1 = "java";
            break;
        case 104382626:
            s1 = "mysql";
            break;
        case -1008861826:
            s1 = "oracle";
            break;
        case 108389755:
            s1 = "redis";
            break;
        case 3492:
            s1 = "mq";
            break;
        case 101807910:
            s1 = "kafka";
            break;
        case -95168706:
            s1 = "rabbitmq";
            break;
        default:
            s1 = "default";
            break;
    }
}

此时我们使用 JMH 进行实际的测试,测试代码如下:

import org.openjdk.jmh.annotations.*;
import org.openjdk.jmh.infra.Blackhole;
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.concurrent.TimeUnit;

@BenchmarkMode(Mode.AverageTime) // 测试完成时间
@OutputTimeUnit(TimeUnit.NANOSECONDS)
@Warmup(iterations = 2, time = 1, timeUnit = TimeUnit.SECONDS) // 预热 2 轮,每次 1s
@Measurement(iterations = 5, time = 3, timeUnit = TimeUnit.SECONDS) // 测试 5 轮,每次 3s
@Fork(1) // fork 1 个线程
@State(Scope.Thread) // 每个测试线程一个实例
public class SwitchOptimizeByStringTest {
    static String _STR = "Java中文社群";
    public static void main(String[] args) throws RunnerException {
        // 启动基准测试
        Options opt = new OptionsBuilder()
                .include(SwitchOptimizeByStringTest.class.getSimpleName()) // 要导入的测试类
                .build();
        new Runner(opt).run(); // 执行测试
    }

    @Benchmark
    public void switchHashCodeTest(Blackhole blackhole) {
        String s1;
        switch (_STR.hashCode()) {
            case 3254818:
                s1 = "java";
                break;
            case 104382626:
                s1 = "mysql";
                break;
            case -1008861826:
                s1 = "oracle";
                break;
            case 108389755:
                s1 = "redis";
                break;
            case 3492:
                s1 = "mq";
                break;
            case 101807910:
                s1 = "kafka";
                break;
            case -95168706:
                s1 = "rabbitmq";
                break;
            default:
                s1 = "default";
                break;
        }
        // 为了避免 JIT 忽略未被使用的结果计算,可以使用 Blackhole#consume 来保证方法被正常执行
        blackhole.consume(s1);
    }

    @Benchmark
    public void switchTest(Blackhole blackhole) {
        String s1;
        switch (_STR) {
            case "java":
                s1 = "java";
                break;
            case "mysql":
                s1 = "mysql";
                break;
            case "oracle":
                s1 = "oracle";
                break;
            case "redis":
                s1 = "redis";
                break;
            case "mq":
                s1 = "mq";
                break;
            case "kafka":
                s1 = "kafka";
                break;
            case "rabbitmq":
                s1 = "rabbitmq";
                break;
            default:
                s1 = "default";
                break;
        }
        // 为了避免 JIT 忽略未被使用的结果计算,可以使用 Blackhole#consume 来保证方法被正常执行
        blackhole.consume(s1);
    }

    @Benchmark
    public void ifTest(Blackhole blackhole) {
        String s1;
        if ("java".equals(_STR)) {
            s1 = "java";
        } else if ("mysql".equals(_STR)) {
            s1 = "mysql";
        } else if ("oracle".equals(_STR)) {
            s1 = "oracle";
        } else if ("redis".equals(_STR)) {
            s1 = "redis";
        } else if ("mq".equals(_STR)) {
            s1 = "mq";
        } else if ("kafka".equals(_STR)) {
            s1 = "kafka";
        } else if ("rabbitmq".equals(_STR)) {
            s1 = "rabbitmq";
        } else {
            s1 = "default";
        }
        // 为了避免 JIT 忽略未被使用的结果计算,可以使用 Blackhole#consume 来保证方法被正常执行
        blackhole.consume(s1);
    }
}

以上代码测试的结果如下:


从以上结果可以看出,String 类型的 switch 判断,经过优化之后,性能提升了 2.4 倍,可谓效果显著。

注意事项

以上的 switch 优化是基于 String 类型的,同时我们需要注意 hashCode 重复的问题,例如对于字符串“Aa”和“BB”来说,他们的 hashCode 都是 2112,因此在优化是需要注意此类问题,也就是说我们使用 hashCode 时,必须保证判断添加的值是已知的,并且最好不要出现 hashCode 重复的问题,如果出现此类问题,我们的解决方案是在 case 中进行判断并赋值。

其他优化手段

我们本文重点讨论的是 switch 性能优化的方案,当然如果处于性能考虑,我们还可以使用更加高效的替代方案,例如集合或者是枚举,详见我的另一篇文章《9个小技巧让你的 if else看起来更优雅》

总结

通过本文我们知道 switch 本质上只支持 int 类型的条件判断,即使是 JDK 1.7 中的 String 类型,最终编译的时候还是会被转化为 hashCode(int)进行判断。但因为编译成字节码后会在 case 中使用 if equals 进行比较,所以性能并不算太高(只比 if 高一点点),因此我们可以直接把 String 转化成 int 类型进行比较,从而避免在 case 中进行 if equals 判断的性能消耗,这样就大大的提升 switch 的性能,但需要注意的是,有些 key 值的 hashCode 是相同的,因此在优化时需要提前规避。


最后的话

原创不易,如果觉得本文对你有用,请随手点击一个「」,这是对作者最大的支持与鼓励,谢谢你。

相关推荐

GitHub精选 | 基于go开发的定时任务管理系统

《GitHub精选》是我们分享Github中优质项目的栏目,包括技术、学习、实用与各种有趣的内容。本期推荐的是gocron-定时任务管理系统之前有过一期...

linux定时任务,让你更深入的了解系统

cron的软件包cronierpm-qlcronierpm–qlcrontabs最关键的是一个叫做crontab的命令,我们要书写的非系统定时任务就是通过此程序来编写的;要保证crond.se...

linux定时任务详解 linux定时任务执行

配置方式:(记忆口诀:分、时、日、月、星)#Fordetailsseeman4crontabs#Exampleofjobdefinition:#.---------------...

我终于会写 Java 的定时任务了 java定时任务指定时间执行

前言学过定时任务,但是我忘了,忘得一干二净,害怕,一直听别人说:...

如何用Windows计划任务设置:定时关机

定时关机其实是一个小功能,很多时候还真的需要它。你在网上找还真不是那么好找,下面我们用Windows计划任务设置:定时关机,而且,我还在计划任务中看到了几个不该有的计划任务。?打开:计划任务点击Co...

定时任务优化总结(从半个小时优化到秒级)

整体优化思路:1.按需查询、2.分小批次游标查询、3.JED场景下按数据库分片分组更新、4.精准定位要处理的数据、5.负载均衡业务背景:...

信创终端操作系统上定时任务crontab详解 | 统信 | 麒麟 | 中科方德

原文链接:信创终端操作系统上定时任务crontab详解|统信|麒麟|中科方德...

VIVO手机定时任务功能_定时开关机、定时振动模式

手机在生活中是必不可少的,有些功能我们想要他定时切换变更,本文就来分享一下Vivo手机的定时任务功能。Vivo手机如何进入定时任务界面...

下班总是忘记关电脑!添加任务自动定时关机,不用再跑一趟!

我们每个人都有过这样的经历,下班后、或者出差,单位里的电脑总是忘记关机!很是烦恼!经常忘关电脑的话,一是费电,二是不安全,会给网络攻击留有足够的时间,会导致信息泄露,存在安全隐患。其实只要我们在电脑...

为什么定时任务到时间不执行?带你深入源码找答案

前言早上研发经理给我分配了一个开发任务:每5秒统计一次APP在线人数,并问我啥时能上线?我心想这需求用Spring的定时任务完美解决啊!作为一个萌新正好借此机会在经理面前表现一番,于是我拍着胸脯跟经理...

我来总结下 几种定时任务的执行方式

首先带入我们的业务场景:我们买火车票或者叫外卖的时候,下完单之后会跳转到支付页面,页面会有一个计时器,要求在指定时间内完成支付,否则订单自动取消。这是延时任务的一个典型场景,分析这个场景,就是如何在订...

聊聊定时任务的六种模式 定时任务详解

这篇文章,我们聊聊实现定时任务的六种策略。1自定义单线程上图中,我们启动一个线程,该线程无限循环执行,每隔20毫秒执行业务代码。...

windows如何实现定时任务?配合脚本使用结局很满意

序言:作为一名程序员,通过定时任务去实现各种所需的功能是必须要掌握的,本文来讲解一下在windows服务器上如何创建定时任务。有需要的小伙伴赶紧收藏转发吧。第一步:打开控制面板-》系统和安全-》管理工...

电脑(计算机)如何定时执行任务 电脑定时器怎么定时

电脑想定时执行程序,任务计划步骤如下:1,系统服务里,确保"TaskScheduler"服务启动。2,在控制面板->系统和安全里,找到任务计划,并点击:  或在“管理工具”里点...

分布式定时任务最全详解(图文全面总结)

分布式定时任务是非常核心的分布式系统,下面我就全面来详解分布式定时任务以及分布式定时任务框架@mikechen本篇已收于mikechen原创超30万字《阿里架构师进阶专题合集》里面。什么是分布式定时任...

取消回复欢迎 发表评论: