用Go语言写了一个电脑搜索文件的小东西

浏览数：18 / 时间：2015年06月08日

package main

import (
	"bytes"
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"sync"
)

type jieguo struct {
	go_num int
	Lock   sync.Mutex
}

var searchName string = os.Args[1]
var search_result chan string = make(chan string, 1)
var nums chan int = make(chan int, 1)
var x jieguo = jieguo{go_num: 0}

func main() {
	list := []string{"D:","E:","F"}
	for _, i := range list {
		fmt.Println(i, searchName)
		go pan(i)
	}
	for {
		select {
		case x := <-search_result:
			fmt.Println(x)
		case y := <-nums:
			if y == 0 {
				os.Exit(0)
			}
		}
	}
}

func pan(path string) {
	a := exec.Command("ls", path)
	result, e := a.Output()
	if e != nil {
		fmt.Println(e)
	}
	z := bytes.Split(result, []byte{10})
	for _, i := range z[:len(z)-1] {
		s := path + "\\" + string(i)
		fmt.Println(s)
		e = os.Chdir(s)
		if e == nil {
			x.Lock.Lock()
			x.go_num = x.go_num + 1
			x.Lock.Unlock()
			go func() {
				filepath.Walk(s, walk)
				x.Lock.Lock()
				x.go_num = x.go_num - 1
				nums <- x.go_num
				x.Lock.Unlock()
			}()
		} else {
			fmt.Println(e)
			if string(i) == searchName {
				fmt.Println(s)
			}
		}
	}
}

func walk(path string, info os.FileInfo, err error) error {
	if err != nil {
		return err
	}
	if !info.IsDir() && info.Name() == searchName {
		search_result <- path
		return nil
	}
	return nil
}

郑重声明：本站内容如果来自互联网及其他传播媒体，其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享，并不代表本站赞同其观点和对其真实性负责，也不构成任何其他建议。

用Go语言写了一个电脑搜索文件的小东西