变量替换代码片段

代码片段分享

package main

import (
"bufio"
"bytes"
"io/ioutil"
"os"
"path/filepath"
"strings"
"sync"
"time"
)

func main() {
M := ReadStaticAndVariable("staticTovariable")
if len(os.Args) != 2 {
println("args error.")
return
}
list, err := GetConfigFiles(os.Args[1])
if err != nil {
println(err.Error())
return
}
StaticToVariable(list, M)
}

func GetConfigFiles(path string) ([]string, error) {
if !strings.HasSuffix(path, "/") {
path = path + "/"
}
var list []string
files, err := ioutil.ReadDir(path)
if err != nil {
return list, err
}
for _, v := range files {
if v.IsDir() {
continue
}
list = append(list, path+v.Name())
}
return list, nil
}

func ReadStaticAndVariable(path string) map[string]string {
File, err := os.Open(path)
if err != nil {
return nil
}
defer File.Close()
buf := bufio.NewReader(File)
M := make(map[string]string)
for {
line, _, err := buf.ReadLine()
if err != nil {
if err.Error() == "EOF" {
break
}
return nil
}
list := bytes.Split(line, []byte("="))
if len(list) != 2 {
continue
}
k := bytes.TrimSpace(list[0])
v := bytes.TrimSpace(list[1])
M[string(k)] = string(v)
}
return M
}

//设置并发处理的个数.
const number int = 5

func StaticToVariable(pathList []string, M map[string]string) {
c := make(chan int, number)
for _, name := range pathList {
c <- 1
go staticToVariable(c, name, M)
}
for {
if len(c) != 0 {
time.Sleep(1e9)
continue
}
break
}
}

func staticToVariable(c chan int, path string, M map[string]string) {
defer func(c chan int) {
<-c
}(c)
path = filepath.ToSlash(path)
File, err := os.Open(path)
if err != nil {
return
}
tmpName := filepath.Dir(path) + "/." + filepath.Base(path)
tmp, err := os.OpenFile(tmpName, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
if err != nil {
File.Close()
return
}
buf := getBuf()
defer putBuf(buf)
for {
n, err := File.Read(buf)
if err != nil {
if err.Error() == "EOF" {
break
}
File.Close()
return
}
for k, v := range M {
buf = bytes.Replace(buf[:n], []byte(k), []byte(v), -1)
}
tmp.Write(bytes.TrimSpace(buf))
tmp.Sync()
}
File.Close()
tmp.Close()
os.Remove(path)
os.Rename(tmpName, path)
return
}

var BufPool *sync.Pool = new(sync.Pool)

func getBuf() []byte {
b := BufPool.Get()
if b != nil {
return b.([]byte)
}
return make([]byte, 512)
}

func putBuf(buf []byte) {
BufPool.Put(buf)
}

func checkIsBinary(path string) bool {
File, err := os.Open(path)
if err != nil {
return true
}
buf := make([]byte, 1024)
n, err := File.Read(buf)
if err != nil {
return true
}
if bytes.Contains(buf[:n], []byte{0x0}) {
return true
}
return false
}