Go實現尋找目錄下(包括子目錄)替換檔案內容

標籤：blog   os   檔案   io   re   c   

 

【功能】

    按指定的目錄尋找出檔案，如果有子目錄，子目錄也將進行搜尋，將其中的檔案內容進行替換。

【缺陷】

    1. 沒有過濾出文字檔

    2. 當檔案過大時，效率不高

【代碼】

package mainimport ("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 {//errreturn 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())}}