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

最新推荐文章于 2025-10-22 08:20:47 发布

原创最新推荐文章于 2025-10-22 08:20:47 发布 · 1.9k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#Go语言 #go #golang

Golang 专栏收录该内容

165 篇文章

订阅专栏

本文介绍了一个利用Go语言实现的多路径并行文件搜索与目录遍历程序。该程序通过命令行参数接收多个路径，并在每个路径上并行执行文件搜索操作。它使用了Go的并发机制，如channel和goroutine，来提高搜索效率。程序首先定义了一个结构体jieguo用于记录搜索结果的数量，并在主函数中初始化了这个结构体。接着，通过循环遍历传入的路径列表，并对每个路径执行文件搜索。搜索过程中，程序不仅输出搜索结果，还通过递归调用walk函数进一步遍历目录下的文件。为了确保程序的健壮性，还包含了错误处理机制，如捕获执行命令时可能发生的错误，并在输出中显示。此程序展示了Go语言在处理复杂并发任务时的强大能力。

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
}