Golang 计算目录中代码的行数

程序名字

countline

os 包来获取运行参数

运行的时候要提供两个参数，一个是目录，一个是扩展名
如：
countline . go

没有运行参数的就退出程序

    if len(os.Args) != 3 {
        fmt.Println("countline [Dir] [extention]")
        return
    }

计算文件有多少个 ‘\n’

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
}

遍历目录

    filepath.Walk(os.Args[1], func(path string, info os.FileInfo, err error) error {
        if getExt(path) == os.Args[2] {
            fmt.Println(path)
            n, e := countFileLine(path)
            if e == nil {
                fmt.Println("lines: ", n)
                sum = sum + n
            }
        }
        return nil
    })

filepath.Walk会第二个成员是一个函数，其中的path就是遍历得到的文件路径，经过判断它是一个文件，就直接读出来。

完整代码:

package main

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

var (
    sum int64
)

func main() {
    if len(os.Args) != 3 {
        fmt.Println("countline [Dir] [extention]")
        return
    }
    filepath.Walk(os.Args[1], func(path string, info os.FileInfo, err error) error {
        if getExt(path) == os.Args[2] {
            fmt.Println(path)
            n, e := countFileLine(path)
            if e == nil {
                fmt.Println("lines: ", n)
                sum = sum + n
            }
        }
        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
}