重构代码
一网友需要对所写的代码进行重构,原代码如下:
/// <summary>
/// 检查上传文件类型
/// </summary>
/// <param name="type">文件类型</param>
/// <returns></returns>
public string strType(string type)
{
type = type.ToLower();
type = type.Substring(type.LastIndexOf("."));
if ((".swf,.flv,.doc,.docx,.jpg,.xls,.xlsx,.txt,.ppt,.pptx,.pub").Contains(type))
{
return type;
}
else
{
switch (type)
{
//case ".txt"://txt文本文件
// type = ".rtf";
// break;
case ".wps"://wps文件
type = ".doc";
break;
case ".et"://wps的表格文件
type = ".xls";
break;
case ".pps":
case ".dps":
type = ".ppt";
break;
}
}
return type;
}
}
Insus.NET试重构了一下,去if...else, 和switch ...case的,简单与明了:
public string strType(string type)
{
Dictionary<string, string> dict_type = new Dictionary<string, string>();
dict_type.Add(".swf", "swf");
dict_type.Add(".flv", "flv");
dict_type.Add(".doc", "doc");
dict_type.Add(".docx", "docx");
dict_type.Add(".jpg", "jpg");
dict_type.Add(".xls", "xls");
dict_type.Add(".xlsx", "xlsx");
dict_type.Add(".txt", "txt");
dict_type.Add(".ppt", "ppt");
dict_type.Add(".pptx", "pptx");
dict_type.Add(".pub", "pub");
dict_type.Add(".wps", "doc");
dict_type.Add(".et", "xls");
dict_type.Add(".pps", "ppt");
dict_type.Add(".dps", "ppt");
//如果以后还在有类型需要判断,请加在这里。
if (dict_type.ContainsKey(type))
return dict_type[type].ToString();
return string.Empty;
}
}
共同学习,写下你的评论
评论加载中...
作者其他优质文章