需求:在一堆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)
}
}