分享
golang 多线程查找文件内容
modprobe · · 1540 次点击 · · 开始浏览这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。
1 package main 2
3 import ( 4 "fmt"
5 "io/ioutil"
6 "os"
7 "path/filepath"
8 "strings"
9 ) 10
11 var matchCount = 0
12 var ch = make(chan int, 512) 13
14 func findFile(path string, text string) { 15 var pathArray [100]string
16 var count = 0
17 filepath.Walk(path, func(path string, info os.FileInfo, err error) error { 18 if err != nil { 19 } 20 //find text
21 if !info.IsDir() { 22 if info.Size() < 1024*1024 { 23 pathArray[count] = path 24 count++
25 if count >= 100 { 26 count = 0
27 go findText(pathArray[0:100], text) 28 <-ch 29 } 30 } 31 } 32 return nil 33 }) 34 go findText(pathArray[0:count], text) 35 <-ch 36
37 fmt.Printf("一共发现了 %d 个匹配的文件", matchCount) 38
39 } 40
41 func findText(paths []string, text string) { 42 for _, path := range paths { 43 fi, err := os.Open(path) 44 if err != nil { 45 panic(err) 46 } 47 defer fi.Close() 48 fd, err := ioutil.ReadAll(fi) 49 if err != nil { 50 panic(err) 51 } 52
53 if strings.Index(string(fd), text) > -1 { 54 matchCount++
55 fmt.Println(path) 56 } 57 } 58 ch <- 1
59 } 60
61 func main() { 62 args := os.Args 63 if len(args) < 3 { 64 fmt.Println("需要两个参数 path text") 65 return
66 } 67 path := args[1] 68 text := args[2] 69
70 _, err := os.Stat(path) 71
72 if err != nil { 73 fmt.Println("path不存在") 74 } else { 75 findFile(path, text) 76 } 77 }计数器增加的时候应该加锁
有疑问加站长微信联系(非本文作者)
入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889
关注微信1540 次点击
上一篇:linux的go语言安装
下一篇:golang substring
添加一条新回复
(您需要 后才能回复 没有账号 ?)
- 请尽量让自己的回复能够对别人有帮助
- 支持 Markdown 格式, **粗体**、~~删除线~~、
`单行代码` - 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
- 图片支持拖拽、截图粘贴等方式上传
收入到我管理的专栏 新建专栏
1 package main 2
3 import ( 4 "fmt"
5 "io/ioutil"
6 "os"
7 "path/filepath"
8 "strings"
9 ) 10
11 var matchCount = 0
12 var ch = make(chan int, 512) 13
14 func findFile(path string, text string) { 15 var pathArray [100]string
16 var count = 0
17 filepath.Walk(path, func(path string, info os.FileInfo, err error) error { 18 if err != nil { 19 } 20 //find text
21 if !info.IsDir() { 22 if info.Size() < 1024*1024 { 23 pathArray[count] = path 24 count++
25 if count >= 100 { 26 count = 0
27 go findText(pathArray[0:100], text) 28 <-ch 29 } 30 } 31 } 32 return nil 33 }) 34 go findText(pathArray[0:count], text) 35 <-ch 36
37 fmt.Printf("一共发现了 %d 个匹配的文件", matchCount) 38
39 } 40
41 func findText(paths []string, text string) { 42 for _, path := range paths { 43 fi, err := os.Open(path) 44 if err != nil { 45 panic(err) 46 } 47 defer fi.Close() 48 fd, err := ioutil.ReadAll(fi) 49 if err != nil { 50 panic(err) 51 } 52
53 if strings.Index(string(fd), text) > -1 { 54 matchCount++
55 fmt.Println(path) 56 } 57 } 58 ch <- 1
59 } 60
61 func main() { 62 args := os.Args 63 if len(args) < 3 { 64 fmt.Println("需要两个参数 path text") 65 return
66 } 67 path := args[1] 68 text := args[2] 69
70 _, err := os.Stat(path) 71
72 if err != nil { 73 fmt.Println("path不存在") 74 } else { 75 findFile(path, text) 76 } 77 }计数器增加的时候应该加锁