[UMU 学 golang](1) Hello code!

UMU · · 1985 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

  一直玩 WSH,被说逼格跟不上时代,又据说 go 即将发布的 1.5 版解决了 GC stop-the-world 问题了,所以 UMU 打算以后用它来开发工具。

  最近想统计代码行数,以前用 VBS 写的一时间居然找不到,直接就用 golang 写了一个,基本从零开始用了大约 4 小时完成。下面谈一下心得。

  1、golang 的 runtime 居然没有 set,只能用 map 代替了,一开始觉得不优雅,不过想来也差不多,不计较那么多了。

  2、语法还确实挺简洁的,第一次练手就感觉学这个语言,其实是在学它的规范,语言本身太容易了。

  3、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
}

  4、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
}



有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:UMU

查看原文:[UMU 学 golang](1) Hello code!

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1985 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传