121 lines
2.1 KiB
Go
121 lines
2.1 KiB
Go
|
package tools
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bufio"
|
|||
|
|
"bytes"
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"log"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func PathCreate(dir string) error {
|
|||
|
|
return os.MkdirAll(dir, os.ModePerm)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func FileCreate(content bytes.Buffer, name string) {
|
|||
|
|
file, err := os.Create(name)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println(err)
|
|||
|
|
}
|
|||
|
|
_, err = file.WriteString(content.String())
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println(err)
|
|||
|
|
}
|
|||
|
|
file.Close()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
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() {
|
|||
|
|
log.Println("DIR:", path)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//文件类型需要进行过滤
|
|||
|
|
|
|||
|
|
buf, err := ioutil.ReadFile(path)
|
|||
|
|
if err != nil {
|
|||
|
|
//err
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
content := string(buf)
|
|||
|
|
log.Printf("h.OldText: %s \n", h.OldText)
|
|||
|
|
log.Printf("h.NewText: %s \n", h.NewText)
|
|||
|
|
|
|||
|
|
//替换
|
|||
|
|
newContent := strings.Replace(content, h.OldText, h.NewText, -1)
|
|||
|
|
|
|||
|
|
//重新写入
|
|||
|
|
ioutil.WriteFile(path, []byte(newContent), 0)
|
|||
|
|
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func FileMonitoringById(ctx context.Context, filePth string, id string, group string, hookfn func(context.Context, string, string, []byte)) {
|
|||
|
|
f, err := os.Open(filePth)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalln(err)
|
|||
|
|
}
|
|||
|
|
defer f.Close()
|
|||
|
|
|
|||
|
|
rd := bufio.NewReader(f)
|
|||
|
|
f.Seek(0, 2)
|
|||
|
|
for {
|
|||
|
|
if ctx.Err() != nil {
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
line, err := rd.ReadBytes('\n')
|
|||
|
|
// 如果是文件末尾不返回
|
|||
|
|
if err == io.EOF {
|
|||
|
|
time.Sleep(500 * time.Millisecond)
|
|||
|
|
continue
|
|||
|
|
} else if err != nil {
|
|||
|
|
log.Fatalln(err)
|
|||
|
|
}
|
|||
|
|
go hookfn(ctx, id, group, line)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取文件大小
|
|||
|
|
func GetFileSize(filename string) int64 {
|
|||
|
|
var result int64
|
|||
|
|
filepath.Walk(filename, func(path string, f os.FileInfo, err error) error {
|
|||
|
|
result = f.Size()
|
|||
|
|
return nil
|
|||
|
|
})
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//获取当前路径，比如：E:/abc/data/test
|
|||
|
|
func GetCurrentPath() string {
|
|||
|
|
dir, err := os.Getwd()
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
}
|
|||
|
|
return strings.Replace(dir, "\\", "/", -1)
|
|||
|
|
}
|