Go语言初尝

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


        对于语言设计之争, 唯一需要牢记的一句话是: 如果把 C 变成 C++, 那么 C 就消失了。


        Go 是一个轻量级的简洁的支持并发的语言,  可以用于探索性个人项目, 这是我想学这门语言的主要原因。 对于有一定编程经验的人来说, 学习一种新语言的方式是, 先概览下语言特性, 然后编写一个中等规模的程序, 尽可能地运用到大部分重要特性。 


        下面的程序用于计算一个目录下所有文件或目录的大小。  

       

package main

import (
  "fmt"
  "time"
  "os" 
  "log"
)

type ShortFileInfo struct {
	fileName string
	size     int64
}

func (fileInfo *ShortFileInfo) Desc() string {
   return fmt.Sprintf("{%s:%d}", fileInfo.fileName, fileInfo.size)
}

func produceFiles(dirName string) []os.FileInfo {
    path, err := os.Open(dirName)
    if err != nil {
        log.Fatal(err)
    }
    defer path.Close()
	fileInfos, err := path.Readdir(0);
	if err != nil {
	    log.Fatal(err)
	}
    return fileInfos;
}

func procFile(fileInfo os.FileInfo, baseDirName string, channelBuffer chan ShortFileInfo) {
	var filesize int64
	fileName := fileInfo.Name()
	if fileInfo.IsDir() {
		filesize = totalFilesizeInDir(fileInfo, baseDirName)
	} else {
		filesize = fileInfo.Size()    
	}
    shortFileInfo := ShortFileInfo{fileName, filesize};
	fmt.Println(time.Now().String() + " store: " + shortFileInfo.Desc())
    channelBuffer <- shortFileInfo 
}

func totalFilesizeInDir(fileInfo os.FileInfo, baseDirName string) int64 {
	var filesize int64 = 0
	fileInfos := produceFiles(baseDirName + "\\" + fileInfo.Name())
	for _, subfileInfo := range fileInfos {
		if subfileInfo.IsDir() {
			filesize += totalFilesizeInDir(subfileInfo, baseDirName + "\\" + fileInfo.Name())
		} else {
			filesize += subfileInfo.Size()
		}
	}	
	return filesize
}

func sleep(ns int) {
	time.Sleep(time.Duration(time.Second)*time.Duration(ns))
}

const (
	B int64 = 1
	KB int64 = 1024
	MB int64 = 1024*1024
	GB int64 = 1024*1024*1024
	TB int64 = 1024*1024*1024*1024
)

const formatF string = "%8.4f" 

func readableSize(sizeInBytes int64) string {
	switch {
		case B <= sizeInBytes && sizeInBytes < KB:
			return fmt.Sprintf("%dB", sizeInBytes)
		case KB <= sizeInBytes && sizeInBytes < MB:
			return fmt.Sprintf(formatF+"KB", float64(sizeInBytes)/float64(KB))
		case MB <= sizeInBytes && sizeInBytes < GB:
			return fmt.Sprintf(formatF+"MB", float64(sizeInBytes)/float64(MB))
		case GB <= sizeInBytes && sizeInBytes < TB:
			return fmt.Sprintf(formatF+"GB", float64(sizeInBytes)/float64(GB))
		case TB <= sizeInBytes:
			return fmt.Sprintf(formatF+"TB", float64(sizeInBytes)/float64(TB))
		default:
			return "0"
	}
}

func main() {

    baseDirName := "C:\\Users\\qin.shuq\\Desktop"
    
    fileList := produceFiles(baseDirName)
    fileNumber := len(fileList)
    channelBuffer := make(chan ShortFileInfo, fileNumber)
    fileInfoMap := make(map[string] int64, fileNumber)
	
    for _, fileInfo := range fileList {
        go procFile(fileInfo, baseDirName, channelBuffer)   
    }  
	
	for count := 0 ; count <= fileNumber ; {
		select {
			case fileInfo := <- channelBuffer: 
				fmt.Println(time.Now().String() + " fetch: " + fileInfo.Desc())
				fileInfoMap[fileInfo.fileName] = fileInfo.size
				count++
			default:
				if count == fileNumber {
					close(channelBuffer)
				}
				fmt.Println("Waiting for data ...")	
				sleep(2)	
		}
	}
	
	var totalSize int64 
	totalSize = 0
	for _ , filesize := range fileInfoMap {
		totalSize += filesize
	}
	fmt.Printf("Total size in %s:%dB %s", baseDirName , totalSize, readableSize(totalSize));

} 


               


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

本文来自:CSDN博客

感谢作者:shuqin1984

查看原文:Go语言初尝

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

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