需求:在一堆xml文件中查找需要的内容,如果每个xml都打开后查找,工作量巨大。然后就想到用golang写程序遍历每个xml。

实现思路:使用golang的ioutil.ReadDir()方法,得到一个fileinfo对象,再使用IsDir()方法判断是否为目录,如果是就继续ReadDir(),这样可以一层层遍历文件夹目录;如果否,就使用ioutil.ReadFile()方法,得到一个[]byte,利用string()方法转换为字符串后,再通过strings.Contains()方法匹配出,需要查找的内容。

备注:不一定是xml,也可以是txt或者其他类型的文件。本次代码中没做中文编码处理,故查找的内容不支持中文。使用时将编译好的exe放在目录下,双击运行,输入需要查找的内容。

golang代码:

package main

import (
	"fmt"
	"io/ioutil"
	"strings"
)

var x string

func main() {

	fmt.Println("请输入你要查找的内容,输入完成后按回车键(最大支持5级目录)")
	fmt.Scanln(&x)
	fileinf, err := ioutil.ReadDir("./")
	if err != nil {
		fmt.Println("读取目录失败,错误:", err)
		return
	}
	for _, b1 := range fileinf {
		if b1.IsDir() == true {
			fileinf2, err := ioutil.ReadDir("./" + b1.Name())
			if err != nil {
				fmt.Println("读取目录失败,错误:", err)
				return
			}
			for _, c := range fileinf2 {
				if c.IsDir() == true {
					fileinf3, err := ioutil.ReadDir("./" + b1.Name() + "/" + c.Name())
					if err != nil {
						fmt.Println("读取目录失败,错误:", err)
						return
					}
					for _, c2 := range fileinf3 {
						if c2.IsDir() == true {
							fileinf4, err := ioutil.ReadDir("./" + b1.Name() + "/" + c.Name() + "/" + c2.Name())
							if err != nil {
								fmt.Println("读取目录失败,错误:", err)
								return
							}
							for _, c3 := range fileinf4 {
								if c3.IsDir() == true {
									fileinf5, err := ioutil.ReadDir("./" + b1.Name() + "/" + c.Name() + "/" + c2.Name() + "/" + c3.Name())
									if err != nil {
										fmt.Println("读取目录失败,错误:", err)
										return
									}
									for _, c4 := range fileinf5 {
										if c4.IsDir() == true {
											fileinf6, err := ioutil.ReadDir("./" + b1.Name() + "/" + c.Name() + "/" + c2.Name() + "/" + c3.Name() + "/" + c4.Name())
											if err != nil {
												fmt.Println("读取目录失败,错误:", err)
												return
											}
											for _, c5 := range fileinf6 {
												findfile6(b1.Name(), c.Name(), c2.Name(), c3.Name(), c4.Name(), c5.Name())
											}
										} else {
											findfile5(b1.Name(), c.Name(), c2.Name(), c3.Name(), c4.Name())
										}

									}

								} else {
									findfile4(b1.Name(), c.Name(), c2.Name(), c3.Name())
								}

							}
						} else {
							findfile3(b1.Name(), c.Name(), c2.Name())
						}

					}
				} else {
					findfile2(b1.Name(), c.Name())
				}

			}

		} else {
			findfile(b1.Name())
		}

	}
	fmt.Println("查找完毕")
	for {

	}
}

func findfile(f string) {
	content, err := ioutil.ReadFile("./" + f)
	if err != nil {
		fmt.Println("读取文件失败,错误:", err)
		return
	}
	a := string(content)
	if strings.Contains(a, x) {
		fmt.Println("存在:" + f)
	}
}

func findfile2(d string, f string) {
	content, err := ioutil.ReadFile("./" + d + "/" + f)
	if err != nil {
		fmt.Println("读取文件失败,错误:", err)
		return
	}
	a := string(content)
	if strings.Contains(a, x) {
		fmt.Println("存在:" + d + "/" + f)
	}
}

func findfile3(d string, d2 string, f string) {
	content, err := ioutil.ReadFile("./" + d + "/" + d2 + "/" + f)
	if err != nil {
		fmt.Println("读取文件失败,错误:", err)
		return
	}
	a := string(content)
	if strings.Contains(a, x) {
		fmt.Println("存在:" + d + "/" + d2 + "/" + f)
	}
}

func findfile4(d string, d2 string, d3 string, f string) {
	content, err := ioutil.ReadFile("./" + d + "/" + d2 + "/" + d3 + "/" + f)
	if err != nil {
		fmt.Println("读取文件失败,错误:", err)
		return
	}
	a := string(content)
	if strings.Contains(a, x) {
		fmt.Println("存在:" + d + "/" + d2 + "/" + d3 + "/" + f)
	}
}
func findfile5(d string, d2 string, d3 string, d4 string, f string) {
	content, err := ioutil.ReadFile("./" + d + "/" + d2 + "/" + d3 + "/" + d4 + "/" + f)
	if err != nil {
		fmt.Println("读取文件失败,错误:", err)
		return
	}
	a := string(content)
	if strings.Contains(a, x) {
		fmt.Println("存在:" + d + "/" + d2 + "/" + d3 + "/" + d4 + "/" + f)
	}
}
func findfile6(d string, d2 string, d3 string, d4 string, d5 string, f string) {
	content, err := ioutil.ReadFile("./" + d + "/" + d2 + "/" + d3 + "/" + d4 + "/" + d5 + "/" + f)
	if err != nil {
		fmt.Println("读取文件失败,错误:", err)
		return
	}
	a := string(content)
	if strings.Contains(a, x) {
		fmt.Println("存在:" + d + "/" + d2 + "/" + d3 + "/" + d4 + "/" + d5 + "/" + f)
	}
}