批量替换模版文件代码分享

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
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!