重构代码 --- 计算长函数工具的实现（附源代码）

长函数代表着腐化的代码，难以修改和扩展。在网上搜了竞没找到相关工具，于是就自己手动写了一个。

计算函数行数，找出长函数，这样就可以开始重构代码了。

 

工具：fl.py

说明：计算函数长度，目前只支持.c文件和.cpp文件，支持其他文件类型，只需将代码中.cpp改为对应的文件类型即可

环境要求：安装Python 2.7

使用方法： ./fl.py max_lengthrelative_dir

Max_length：大于该长度的函数则输出

Relative_dir：目录相对路径

 

使用例子：./fl.py 300101_StarZoneServer/

@IED_40_125_sles10sp1:~> ./fl.py 300 101_ZoneServer/
Lines > 300 , Search in dir: 101_StarZoneServer/
101_ZoneServer/DropEngine/DropFramework.cpp : line 295 total : 563 namespace DroppedThingPrivateFunction
101_ZoneServer/GameMaster/PetAttributeCommand.cpp : line 41 total : 341 int CPetAttributeCommand::SecurityCheck()
101_ZoneServer/GameMaster/RoleAttributeCommand.cpp : line 186 total : 320 int CRoleAttributeCommand::SecurityCheck()
101_ZoneServer/GameMaster/PetEnhanceCommand.cpp : line 36 total : 403 int CPetEnhanceCommand::SecurityCheck()

输出第一行为参数，函数长度大于300，搜索目录名。

后面输出为：

文件名 函数起始行号 函数总长度函数名称

 

如果有什么不足或BUG，欢迎交流反馈和完善。

#!/usr/bin/python
# Create by douglaschen, 2011.12.12. All right Reserved.
import os
import string
import sys
import linecache

__DEBUG = 0
def fl(FILENAME, LENGTH):
	sysSed="sed -e '/^{\|^}/=' -n "
	sysSed += str(FILENAME)

	result = os.popen(sysSed).read()

	resultlist = result.split('\n')

	# print len(resultlist)
	for i in range(0, len(resultlist)-1, 2):
		start = int(resultlist[i])
		end   = int(resultlist[i+1])
		if ((end - start) < LENGTH):
			continue

		print FILENAME, ": line", start - 1, "total :", (end - start), 
		print linecache.getline(FILENAME, start - 1), 


def walkdir(directory):
	if __DEBUG != 0:
		import pdb 
		pdb.set_trace()
	# print directory
	for root, dirs, files in os.walk(directory):
		# print "root : ", root
		# print "dirs : ", dirs
		# print "files :", files
		for file in files:
			# print file
			string = str(file)
			if not (string.endswith(".cpp") or string.endswith(".c")):
				continue
		
			file = os.path.join(root, file)
			# print "file is : ", file
			fl(file, LENGTH)
	
		for dir in dirs:
			# print dir
			walkdir(dir)

if __name__ == '__main__':
	# fl("UseThing.cpp", 50)
	LENGTH=150
	DIR=os.path.abspath('CandyAT/Bin/Server_proj/')
	
	argc = len(sys.argv)
	if 3 != argc:
		print "argc :", argc
		print "Usage : ./fl.py max_length relative_dir"
		sys.exit(0)
	if (argc > 1):
		LENGTH = int(sys.argv[1])
	if (argc > 2):
		DIR = sys.argv[2]
	print "Lines >", LENGTH, ", Search in dir:", DIR
	walkdir(DIR)

如果有什么不足或BUG，欢迎交流反馈和完善。