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

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

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

C语言switch/case圈复杂度优化重构_第1张图片

程序块重构

程序块重构指的是每个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;

}

  

 

转载于:https://www.cnblogs.com/wahaha02/p/7435603.html

你可能感兴趣的:(c/c++,数据结构与算法)