golang file list

filelist.go

package main

import (
    "flag"
    "fmt"
    "os"
    "path/filepath"
    "strings"
)

var (
    total int
)

func getFilelist(path string) {
    err := filepath.Walk(path, func(path string, f os.FileInfo, err error) error {
        if f == nil {
            return err
        }
        if f.IsDir() {
            return nil
        }
        endsWith := strings.HasSuffix(path, ".go") // true
        test := strings.HasSuffix(path, "_test.go")
        if endsWith && !test {
            total++
            println(path)
        }
        return nil
    })
    if err != nil {
        fmt.Printf("filepath.Walk() returned %v\n", err)
    }
}

func main() {
    total = 0
    flag.Parse()
    root := flag.Arg(0)
    getFilelist(root)
    println("total = ", total)
}