Go实现查找目录下(包括子目录)替换文件内容

 

【功能】

    按指定的目录查找出文件，如果有子目录，子目录也将进行搜索，将其中的文件内容进行替换。

【缺陷】

    1. 没有过滤出文本文件

    2. 当文件过大时，效率不高

【代码】

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69

package main   import (     "flag"     "fmt"     "io/ioutil"     "os"     "path/filepath"     "strings" )   type ReplaceHelper struct {     Root    string //根目录     OldText string //需要替换的文本     NewText string //新的文本 }   func (h *ReplaceHelper) DoWrok() error {       return filepath.Walk(h.Root, h.walkCallback)   }   func (h ReplaceHelper) walkCallback(path string, f os.FileInfo, err error) error {       if err != nil {         return err     }     if f == nil {         return nil     }     if f.IsDir() {         //fmt.Pringln("DIR:",path)         return nil     }       //文件类型需要进行过滤       buf, err := ioutil.ReadFile(path)     if err != nil {         //err         return err     }     content := string(buf)       //替换     newContent := strings.Replace(content, h.OldText, h.NewText, -1)       //重新写入     ioutil.WriteFile(path, []byte(newContent), 0)       return err }   func main() {       flag.Parse()     helper := ReplaceHelper{         Root:    flag.Arg(0),         OldText: flag.Arg(1),         NewText: flag.Arg(2),     }     err := helper.DoWrok()     if err == nil {         fmt.Println("done!")     } else {         fmt.Println("error:", err.Error())     } }