package main import ( "bufio" "bytes" "flag" "fmt" "io" "io/IoUtil" "os" ) func main() { config := flag.String("f","replations.ini","-f replations.ini") dir := flag.String("d","config","-d configdir") flag.Parse() M,err := ParseConfig(*config) if err != nil { fmt.Println(err) return } err = AutoReplace(*dir,M) if err != nil { fmt.Println(err) return } } func ParseConfig(configpath string) (map[string]string,error) { file,err := os.Open(configpath) if err != nil { return nil,err } defer file.Close() M := make(map[string]string) buf := bufio.NewReader(file) var num int for { line,_,err := buf.ReadLine() if err != nil { if err.Error() == "EOF" { break } return nil,err } list := bytes.Split(line,[]byte("=")) if len(list) != 2 { return nil,fmt.Errorf("第%d行 出现多次'='",num) } M[string(bytes.Trimspace(list[0]))] = string(bytes.Trimspace(list[1])) num++ } return M,nil } func AutoReplace(dirpath string,variablesMap map[string]string) error { Files,err := getFiles(dirpath) if err != nil { return err } for fileName,body := range Files { fmt.Println(string(body)) for k,v := range variablesMap { body = bytes.Replace(body,[]byte(k),[]byte(fmt.Sprintf("{{.%s}}",v)),-1) } F,err := os.OpenFile(fileName,os.O_CREATE|os.O_RDWR|os.O_Trunc,0644) if err != nil { F.Close() return err } F.Write(body) F.Close() } return nil } func getFiles(path string) (map[string][]byte,error) { if !bytes.HasSuffix([]byte(path),[]byte("/")) { path = path + "/" } files,err := IoUtil.ReadDir(path) if err != nil || len(files) <= 0 { return nil,err } var fileInfo map[string][]byte = make(map[string][]byte) for _,file := range files { if file.IsDir() { continue } F,err := os.Open(path + file.Name()) if err != nil { return nil,err } buf := make([]byte,file.Size()) n,err := io.ReadFull(F,buf) if err != nil { return nil,err } if bytes.Contains(buf[:n],[]byte{0}) { continue } fileInfo[path+file.Name()] = buf[:n] } return fileInfo,nil }
原文地址:https://www.jb51.cc/go/189907.html
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。