2019-03-10发表2024-09-06更新代码世界 / 重构6 分钟读完 (大约834个字)

序

最近在经手一个Winform项目,里面有个需求：TreeView控件选择不同的TreeNode时，需要展示不同的右键菜单。记录一下重构过程。

需求

现将公司内部需求初始需求转换如下：
如果TreeView选中的TreeNode是国家，则右键菜单展示”添加省”,一个选项。
如果TreeView选中的TreeNode是省份，则右键菜单展示”添加市”，”删除”两个选项。
如果TreeView选中的TreeNode是城市，则右键菜单展示”添加区”，”删除”两个选项。
如果TreeView选中的TreeNode是区，则右键菜单展示，”删除”一个选项。
这部分的伪代码如下：

var area = TreeView.SelectedNode.Tag as Area; //
if(area!=null)
{
    var areaType=area.type;
    ContextMenuStrip menuStrip = new ContextMenuStrip();
    if(areaType==AreaType.国家)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("添加省"));
    }
    if(areaType==AreaType.省份)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("添加市"));
        menuStrip.Items.Add(new ToolStripMenuItem("删除"));
    }
    if(areaType==AreaType.城市)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("添加区"));
        menuStrip.Items.Add(new ToolStripMenuItem("删除"));
    }
    if(areaType==AreaType.区)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("删除"));
    }

    menuStrip.Show(e.Location);
}

后来业务发展了,除了AreaType.国家这个选项，右键菜单都需要添加一个”打开地图”的右键菜单，且位于右键菜单的最上方。
重构之后的示例代码变成了如下：

var area = TreeView.SelectedNode.Tag as Area; //
if(area!=null)
{
    var areaType=area.type;
    ContextMenuStrip menuStrip = new ContextMenuStrip();
    if(areaType==AreaType.国家)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("添加省"));
    }
    if(areaType==AreaType.省份)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("打开地图"));
        menuStrip.Items.Add(new ToolStripMenuItem("添加市"));
        menuStrip.Items.Add(new ToolStripMenuItem("删除"));
    }
    if(areaType==AreaType.城市)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("打开地图"));
        menuStrip.Items.Add(new ToolStripMenuItem("添加区"));
        menuStrip.Items.Add(new ToolStripMenuItem("删除"));

    }
    if(areaType==AreaType.区)
    {
        menuStrip.Items.Add(new ToolStripMenuItem("打开地图"));
        menuStrip.Items.Add(new ToolStripMenuItem("删除"));
    }

    menuStrip.Show(e.Location);
}

这次需求变动，已经让我闻到了代码中的坏味道。增删改一个右键菜单需要修改好几处地方，删除右键菜单也是同理。
故需要对代码进行重构。

重构

提取函数，伪代码如下

private ContextMenuStrip menuStrip = new ContextMenuStrip(); //菜单转为私有属性
private void AddToolStripMenuItem(string controlName, bool isAdd)
{
    if (isAdd)
    {
        this.menuStrip.Items.Add(new ToolStripMenuItem(controlName););
    }
}

故调用处的实现代码变成了如下

var area = TreeView.SelectedNode.Tag as Area; //
if(area!=null)
{
    var areaType=area.type;
    this.menuStrip.Items.Clear();
    AddToolStripMenuItem("添加省",areaType==AreaType.国家);
    AddToolStripMenuItem("打开地图",!areaType==AreaType.国家);
    AddToolStripMenuItem("添加市",areaType==AreaType.省份);
    AddToolStripMenuItem("添加区",areaType==AreaType.城市);
    AddToolStripMenuItem("删除",!areaType==AreaType.国家);
    menuStrip.Show(e.Location);
}

若以后需求变更,打开地图需要改成打开百度地图，然后新增打开高德地图也只需要修改一行代码，并新增一行代码即可。

总结

该重构过程运用的是提取函数和函数添加参数。
AddToolStripMenuItem函数只关心菜单项是否添加，而不关心菜单项添加的逻辑是什么。
重构之后，右键菜单的可能的菜单项也一目了然，每一个菜单项对应的添加条件也一目了然。若isAdd参数对应的表达式过长，则可以将该表达式封装成函数进行传递。
通过该重构提高了项目的可维护性，也算是一种成就。

解决方案

ProjectAClass添加一个委托事件,CallMethod函数添加一个Dosometing dosometing的参数。

public class ProjectAClass
{
    public delegate void Dosometing();

    public static void CallMethod(bool isTrue, Dosometing dosometing)
    {
        if (isTrue)
        {
            dosometing?.Invoke();
        }
    }
}

具体的调用代码处就变成了

1	ProjectAClass.CallMethod(true,()=>{ProjectC_StaticClassC.DoSomething();});

扩展

若ProjectAClass.CallMethod方法多地方出现，然而没有办法一下修改到位。可以通过参数默认值的方式来实现修改。

public static void CallMethod(bool isTrue, Dosometing dosometing=null)
{
    if (isTrue)
    {
        if(dosometing == null)
        {
            ProjectB_StaticClass.DoSomething();
        }
        else
        {
            dosometing.Invoke(); // or dosometing()
        }         
    }
}

该方法可以避免原来的调用方法进行大面积的修改。

序

需求

重构

总结

序

场景

需求

解决方案

扩展

总结

链接

分类

最新文章

归档

标签

订阅更新

广告

follow.it