[UMU 学 golang](2) 升级 1.5 + 优化

安装 1.5 时,直接覆盖 1.4.2,结果不能用了,报错:imports runtime: C source files not allowed whennotusing cgo or SWIG

解决方法是删掉 go 1.5......然后修复安装一遍。

优化:1、加了计时功能,纯属蛋疼。2、学到一个不占空间的 struct{},map[string]bool 改为 map[string]struct{}。

// UMU @ 2015年08月17日 11:30
// Last update: 2015年08月21日 17:40
package main
import (
	"bufio"
	"fmt"
	"os"
	"path"
	"path/filepath"
	"time"
)
func isSourceFile(ext string) bool {
	var kSourceFileExts = []string{".c", ".cc", ".cpp", ".h", ".hpp",
		".go",
		".java"}
	for _, r := range kSourceFileExts {
		if r == ext {
			return true
		}
	}
	return false
}
func main() {
	if len(os.Args) < 2 {
		fmt.Println("Usage: ", os.Args[0], "[file or folder]...")
		return
	}
	start := time.Now()
	files := make(map[string]struct{})
	lines := 0
	for _, r := range os.Args {
		fi, err := os.Stat(r)
		if err == nil || os.IsExist(err) {
			if fi.IsDir() {
				AddDirectory(r, files)
			} else {
				AddFile(r, files)
			}
		}
	}
	for file, _ := range files {
		l := CountLine(file)
		lines += l
		fmt.Println(file, l)
	}
	fmt.Printf("Total lines: %d, cost %fs\n", lines, timeElapsed(start))
}
func AddDirectory(name string, files map[string]struct{}) {
	filepath.Walk(name, func(path string, fi os.FileInfo, err error) error {
		if nil == fi {
			return err
		}
		if fi.IsDir() {
			return nil
		}
		AddFile(path, files)
		return nil
	})
}
func AddFile(name string, files map[string]struct{}) {
	ext := path.Ext(name)
	if isSourceFile(ext) {
		path, err := filepath.Abs(name)
		if err == nil {
			_, exists := files[path]
			if exists {
				fmt.Println("Duplicated", path)
			} else {
				files[path] = struct{}{}
				fmt.Println("Add", path)
			}
		}
	}
}
func CountLine(path string) (num int) {
	f, err := os.Open(path)
	if nil != err {
		return
	}
	defer f.Close()
	s := bufio.NewScanner(f)
	for s.Scan() {
		num += 1
	}
	return
}
func timeElapsed(start time.Time) float64 {
	dis := time.Since(start).Seconds()
	return dis
}

本文来自:开源中国博客

感谢作者:UMU

查看原文:[UMU 学 golang](2) 升级 1.5 + 优化

• 编辑
• 预览