C语言switch/case圈复杂度优化重构

软件重构是改善代码可读性、可扩展性、可维护性等目的的常见技术手段。圈复杂度作为一项软件质量度量指标，能从一定程度上反映这些内部质量需求（当然并不是全部），所以圈复杂度往往被很多项目采用作为软件质量的度量指标之一。

C语言开发的项目中，switch/case代码块是一个很容易造成圈复杂度超标的语言特性，所以本文主要介绍下降低switch/case圈复杂度的重构方法（如下图）。switch圈复杂度优化重构可分为两部分：程序块的重构和case的重构。程序块重构是对代码的局部优化，而case重构是对代码的整体设计，所涉及的重构手段也各不相同。

程序块重构

程序块重构指的是每个case内的代码段重构。Martin Fowler 的《重构——改善既有代码的设计》（电子版）书中总结了80多种重构方法。书中针对每种技术都给出了示例说明，另外这里、这里还提供了其他语言的示例和进一步介绍。因为存在大量示例，所以本文针对这些方法不再给出示例，有兴趣的同学可以通过上面几种途径了解学习。不过这些技术中有些是改善代码的可读性，有些是改善代码的可扩展性，并不是每项技术都能有效减低圈复杂度。其中可以降低圈复杂度的方法有如下几种：

• 提炼函数（Extract Method）。你有一段代码可以被组织在一起并独立出来。将这段代码放进一个独立函数中，并将函数名称解释该函数的用途。
• 分解条件表达式（Decompose Conditional）。你有一个复杂的条件（if-then-else）语句。从if、then、else三分段落中分别提炼出独立函数。
• 合并条件表达式（Consolidate Conditional Expression）。你有一系列条件测试，都得到相同结果。将这些测试合并为一个条件表达式，并将这个条件表达式提炼成为一个独立函数。
• 合并重复的条件片段（Consolidate Duplicate Conditional Fragments）。在条件表达式的每个分支上有着相同的一段代码。将这段重复的代码搬移到条件表达式之外。
• 移除控制标记（Remove Control Flag）。在一系列布尔表达式中，某个变量带有“控制标记”的作用。以break语句或return语句取代控制标记。

这些重构方法除了降低圈复杂度外，还有如下好处：

• 满足单一职责设计原则，提高代码可读性。
• 去除重复冗余代码。你可以删除大量相同的条件语句。
• 满足“Tell, Dont Ask”原则，告诉对象需要做什么，而不是怎么做。

case重构

对于一个switch有几十个case的情况，其圈复杂度往往上百，程序块重构显然已不能解决其本质复杂度。如果要降低其圈复杂度，必然需要对代码进行重新设计。

C语言的switch/case语言特性本质是描述一种查表逻辑，其中表结构和表的控制（即查表）都通过软件来表达。表通过代码来描述，这显然不是一种最佳的实现方式。我们需要做的就是，避免控制中的复杂性，将精力集中在数据的组织上，以反映所模拟世界的真实结构，并将数据与控制进行分离。

表的设计由两部分组成：对象（表项）的抽象和表的构建。对象如何抽象，对象粒度如何划分，对象间的关系如何设计？这些问题涉及抽象思维能力的训练，而且也与具体业务逻辑强相关，不是本文重点。读者可阅读《计算机程序的构造和解释》来进一步了解软件抽象等相关技术细节。

表的构建方法是本文的重点，其可分为编译期构建、链接期构建和运行时构建。3种方法各有所长和不足，可根据自身需要进行选择。

编译期表构建

问题背景

boot启动支持3种启动方式，每种启动方式的用户菜单流程也不尽相同。启动菜单支持输入检查、存储、菜单回退等功能。原有设计中函数设计臃肿，菜单项通过switch/case来进行选择处理，有十几个函数圈复杂度超过40，最大的圈复杂度为147，代码维护困难。

重构方法

boot启动用户菜单本质是一个优先状态机，每个菜单项是其中一个状态。抽象菜单项对象T_PROMT，其包含提示打印、输入检查、存储、状态跳转等成员。构建T_PROMT aPromtArray[]菜单表描述所有菜单项对象，通过MenuFsm实现状态机的控制：通过对象T_PROMT的jumpto接口实现状态的跳转，通过check接口实现输入检查，通过setvalue接口实现存储，通过parent实现菜单回退到上级菜单（因为上级菜单是动态变化的，无法静态初始化，所以在jumpto中进行动态赋值）。示例代码如下：

typedef struct prompt
{
    WORD32 type;
    CHAR *name;/*env name*/
    CHAR *prompt;/*prompt info to user*/
    WORD32  (*check)(CHAR *src);/*check func for user's input*/
    struct prompt* (*jumpto)(struct prompt*, WORD32);
    struct prompt *parent;
    VOID (*setvalue)(CHAR *name);
}T_PROMT;

static T_PROMT aPromtArray[] =
{
    /*            env name          prompt string   check func      jump func   parent      set func */
    {TYPE_NORMAL, ENV_LOCAL_IP,     "Local IP:",   CheckIpAddr,    LocalIpJump    ,NULL,    SetCltIpAddr   },
    {TYPE_NORMAL, ENV_SERVER_IP,    "Server IP:",  CheckIpAddr,    ServeripJump   ,NULL,    SetSerIpAddr   },
 /* 共 22 个表项，以下略 */
};

static SWORD32 MenuFsm(struct prompt *menu)
{
    SWORD32 dwRet = BSP_OK;
    WORD32 dwIndex;

    while(menu != NULL) {
        if (menu == GetPrompt(ENV_NULL)) {
            dwRet = MODE_MENU_BACK;
            break;
        }

        dwIndex= PrintPromptAndGetUserInput(menu);
        if (dwIndex  != NORMAL_MENU_BACK ) {
            menu = menu->jumpto(menu, dwIndex);
        } else {
            menu = menu->parent;
        }
    }

    return dwRet;
}

static struct prompt* GetPrompt(char *name)
{
    WORD32 i = 0;
    struct prompt *pt = NULL;

    WORD32 dwSize = sizeof(aPromtArray)/sizeof(aPromtArray[0]);

    for (i = 0; i < dwSize; i++) {
        if (strcmp(name, aPromtArray[i].name) == 0) {
            pt = &aPromtArray[i];
            break;
        }
    }

    return pt;
}

　　

运行时表构建

问题背景

内核模块通过ioctl对外部提供接口，而此模块ioctl控制码有84个，原ioctl函数通过switch/case完成ioctl的分发和处理，此实现方案导致函数代码长度达767行，圈复杂度达124，难以维护，不满足项目软件质量要求（函数圈复杂度在12以下）。

重构方法

抽象ioctl接口对象ctrl_operations并实例化；通过bsp_iocmds_init构建字典（哈希表），实现ioctl控制码到ioctl接口的映射；在board_dev_init模块初始化中完成哈希表的初始化；在boardctrl_do_ioctl中通过哈希查表接口bsp_dict_get获取ioctl控制码的处理接口。

示例代码

struct ctrl_operations {
    SWORD32 (*board_init)(struct board *bd);
    SWORD32 (*board_exit)(struct board *bd);
/* 共 92 个表项，以下略 */
};

struct ctrl_operations ioctl_ops = {
    .inherits           = &extern_ops,
    .epld_op            = bsp_epld_op,
    .epldrw             = bsp_epld_rw,
/* 共 84 个字段，以下略 */
};

void bsp_iocmds_init(struct board *bd, pt_bsp_dict pdict)
{
    bsp_dict_add(pdict, BSP_IOCMD_ROV_WR, bd->ops->rov_wr);
    bsp_dict_add(pdict, BSP_IOCMD_TCAM_INFO, bd->ops->tcam_info);
/* 共 84 个key，以下略 */

}

static SWORD32 __init board_dev_init(void)
{
    struct board *bd = get_board();
/* 删除无关代码 */
    bd->iocmds = bsp_dict_new(DICT_HINT, bsp_cmp, bsp_hash);
    bsp_iocmds_init(bd, bd->iocmds);

    return BSP_OK;
}

WORD32 boardctrl_do_ioctl(unsigned int cmd, void *pParam)
{
    WORD32 dwIoNum  = _IOC_NR(cmd);
    struct board *bd = get_board();
    WORD32 dwRet = BSP_E_BRDCTRL_NOTSUPPORT;
    PT_OPS_FUNC ops;

    ops = bsp_dict_get(bd->iocmds, dwIoNum);

    if(likely(ops)) {
        dwRet = ops(bd, pParam);
    }

    return dwRet;

}

当然除了使用哈希表，也可以使用链表等数据结构来组织数据。　　

链接期表构建

问题背景

编译期表构建和运行时表构建2种方法，能优化设计，降低圈复杂度，但有一件事情没有做完美：新增一个表项时，必须修改公共的静态表（编译期表构建，如需要修改aPromtArray）或注册函数（运行时表构建，如需要修改bsp_iocmds_init），无法做到完全满足“开发封闭原则”。

链接期表构建方法则可以解决这个问题。

重构方法

通过gcc的section属性，把所有(ioctl控制码，接口)数据对（即元组）定义在同一个section数据段中。在链接阶段，链接器会构建初始化此section数据段，话句话说，连接器帮助我们完成了这个对象数组的初始化和构建。然后利用gcc导出的__start_ctrl_op_section和__stop_ctrl_op_section符号，boardctrl_do_ioctl即可完成对section数据表的查表操作。

此项技术在u-boot、Linux kernel中大量使用。当添加一个新表项时，只需要添加一句ctrl_op_init，不需要修改任何公共代码或数据。

示例代码：

typedef void (*ctrl_op)(struct board *bd);

#define _init __attribute__((section("ctrl_op_section")))
#define ctrl_op_init(num, func) ctrl_op __no_##func _init = (ctrl_op)num; \
                                ctrl_op __fn_##func _init = func

extern ctrl_op __start_ctrl_op_section;
extern ctrl_op __stop_ctrl_op_section;

ctrl_op_init(BSP_IOCMD_ROV_WR, bsp_rov_wr);
ctrl_op_init(BSP_IOCMD_TCAM_INFO, bsp_tcam_info);
/* 共 84 个ctrl_op_init，以下略 */

WORD32 boardctrl_do_ioctl(unsigned int cmd, void *pParam)
{
    WORD32 dwIoNum  = _IOC_NR(cmd);
    struct board *bd = get_board();
    WORD32 dwRet = BSP_E_BRDCTRL_NOTSUPPORT;
    ctrl_op * ptr = &__start_ctrl_op_section;

    do {
        if((WORD32)*ptr == dwIoNum) {
            ptr++;
            if(likely(ptr))
                return (*ptr)(bd, pParam);
        }
        ptr += 2;
    } while (ptr < &__stop_ctrl_op_section);

    return dwRet;

}