Go 1.5关键字搜索目录、文件、文件内容_修复一个小BUG

最新推荐文章于 2025-06-09 19:45:00 发布

JieLinDee

最新推荐文章于 2025-06-09 19:45:00 发布

阅读量1.8k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Golang 文章标签： golang go 搜索

本文链接：https://blog.youkuaiyun.com/fyxichen/article/details/47418523

Golang 专栏收录该内容

165 篇文章

订阅专栏

本文介绍了一个使用Go语言编写的实用工具，该工具能够帮助开发者在指定目录下搜索包含特定关键字的文件或文件夹，并提供了多种搜索选项。通过简单的命令行参数配置，可以指定搜索范围、目标类型及搜索模式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package main

import (
	"bufio"
	"bytes"
	"flag"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"sync/atomic"
)

var re []byte
var count int32
var c, f, d bool
var i, p string

func main() {
	flag.StringVar(&p, "p", "./", "-p=finddir")   //搜索的目录，默认是当前目录，path
	flag.StringVar(&i, "r", "", "-r=matchstring") //关键字，index
	flag.BoolVar(&c, "c", false, "-c=true")       //是否在文件内容搜索关键字，content
	flag.BoolVar(&f, "f", false, "-f=true")       //是否根据关键字查找文件，file
	flag.BoolVar(&d, "d", false, "-d=true")       //是否根据关键字查找文件夹，directory
	flag.Parse()
	if len(i) == 0 {
		fmt.Println("关键字不能为空")
		return
	}
	re = []byte(i)
	if err := filepath.Walk(p, walk); err != nil {
		fmt.Println("Walk:", err)
	}
	if count == 0 {
		fmt.Println("搜索路径：", p)
		fmt.Println("关键字：", i)
		if d {
			fmt.Println("是否搜索目录：", d)
		}
		if f {
			fmt.Println("是否搜索文件：", f)
		}
		if c {
			fmt.Println("是否搜索内容：", c)
		}
	}
	fmt.Printf("共发现 : %d\n", count)
}
func walk(path string, info os.FileInfo, err error) error {
	if err != nil {
		return err
	}
	if info.IsDir() {
		if d && bytes.Contains([]byte(info.Name()), re) {
			atomic.AddInt32(&count, 1)
			fmt.Println("匹配到目录：", path)
		}
		return nil
	}

	if f {
		if bytes.Contains([]byte(info.Name()), re) {
			atomic.AddInt32(&count, 1)
			fmt.Println("匹配到文件：", path)
			return nil
		}
	}
	if c {
		return read(path)
	}
	return nil
}

//在文件中查找关键字
func read(path string) error {
	File, err := os.Open(path)
	defer File.Close()
	if err != nil {
		fmt.Println(path, ":", err)
		return err
	}
	r := bufio.NewReader(File)
	i := 0
	for {
		i++
		b, _, err := r.ReadLine()
		if bytes.Contains(b, []byte{0}) { //排除二进制文件
			return nil
		}
		if err != nil {
			if err == io.EOF {
				if bytes.Contains(b, re) {
					atomic.AddInt32(&count, 1)
					fmt.Printf("在文件：%s第 %d行\n", path, i)
					fmt.Println(string(bytes.TrimSpace(b)))
				}
				return nil
			}
			return err
		}
		if bytes.Contains(b, re) {
			atomic.AddInt32(&count, 1)
			fmt.Printf("在文件：%s第 %d行\n", path, i)
			fmt.Println(string(bytes.TrimSpace(b)))
		}
	}
	return nil
}