如何统计文件夹或者文件代码行数

francis0072011

已于 2022-06-28 00:28:35 修改

阅读量961

点赞数

CC 4.0 BY-SA版权

分类专栏：环境文章标签： golang 工具-统计代码行数

于 2022-06-13 01:17:01 首次发布

原文链接：https://editor.youkuaiyun.com/md?not_checkout=1&articleId=125252823

环境专栏收录该内容

5 篇文章

订阅专栏

该博客介绍了一个用Go编写的程序，用于统计指定目录下特定扩展名文件的代码行数。通过`filepath.Walk`遍历目录，读取文件并计算换行符来实现计数。程序提供命令行参数，接收目录路径和文件扩展名作为输入。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

如何统计文件夹或者文件代码行数

代码

package main

import (
    "bufio"
    "fmt"
    "io/ioutil"
    "os"
    "path/filepath"
    "strings"
)

var (
	sum int64
)

// 示例 
// （1）交叉编译：
// （2）使用：./app.exe  countline /d/projects/xxx go
func main() {
	if len(os.Args) < 3 {
	   fmt.Println("countline [Dir] [extention]")
	   return
	}
	filepath.Walk(os.Args[2], func(path string, info os.FileInfo, err error) error {
	   if getExt(path) == os.Args[3] {
	       fmt.Println(path)
	       count := countLine(path)
	       fmt.Println("lines: ", count)
	       sum = sum + int64(count)
	   }
	   return nil
	})
	fmt.Println("total lines: ", sum)
}

//获取得文件的扩展名，最后一个.后面的内容
func getExt(f string) (ext string) {
	//  fmt.Println("ext:", f)
	index := strings.LastIndex(f, ".")
	data := []byte(f)
	for i := index + 1; i < len(data); i++ {
		ext = ext + string([]byte{data[i]})
	}
	return
}

func countFileLine(name string) (count int64, err error) {
	fmt.Println("count file:", name)
	data, err := ioutil.ReadFile(name)
	if err != nil {
		return
	}
	count = 0
	for i := 0; i < len(data); i++ {
		if data[i] == '\n' {
			count++
		}
	}
	fmt.Println("line:", count)
	return
}

func countLine(fileName string) int {
	file, err := os.Open(fileName)
	if err != nil {
		return 0
	}
	defer file.Close()
	fd := bufio.NewReader(file)
	count := 0
	for {
		_, err := fd.ReadString('\n')
		if err != nil {
			break
		}
		count++

	}
	fmt.Println(count)
	return count
}