[UMU 學 golang](1) Hello code!

　　一直玩 WSH，被說逼格跟不上時代，又聽說 go 即將發佈的 1.5 版解決了 GC stop-the-world 問題了，因此 UMU 打算之後用它來開發工具。

　　最近想統計代碼行數，之前用 VBS 寫的一時間竟然找不到，直接就用 golang 寫了一個，基本從零開始用了大約 4 小時完成。下面談一下心得。

　　一、golang 的 runtime 竟然沒有 set，只能用 map 代替了，一開始以爲不優雅，不過想來也差很少，不計較那麼多了。

　　二、語法還確實挺簡潔的，第一次練手就感受學這個語言，實際上是在學它的規範，語言自己太容易了。

　　三、defer 挺好用的，簡潔、省心，好比這個核心函數：

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
}

　　四、LiteIDE X 還比較好用，寫 import 時，會自動按字母順序排列庫名，這樣免得糾結順序了……

剛剛接觸，就說到這裏，最後附代碼。

// UMU @ 2015-08-17 11:30
// Last update: 2015-08-17 17:01
package main

import (
	"bufio"
	"fmt"
	"os"
	"path"
	"path/filepath"
)

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
	}

	files := make(map[string]bool)
	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.Println("Total lines:", lines)
}

func AddDirectory(name string, files map[string]bool) {
	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
	})
	return
}

func AddFile(name string, files map[string]bool) {
	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] = true
				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
}