package utils
|
|
import (
|
"archive/zip"
|
"basic.com/valib/logger.git"
|
"bufio"
|
"bytes"
|
"crypto/md5"
|
"errors"
|
"fmt"
|
"io"
|
"io/ioutil"
|
"os"
|
"os/exec"
|
"path"
|
"path/filepath"
|
"runtime"
|
)
|
|
func GetExePath() string {
|
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
if nil != err {
|
dir = "./"
|
}
|
|
return dir
|
}
|
|
func PathExists(path string) bool {
|
_, err := os.Stat(path)
|
if nil == err {
|
return true
|
}
|
|
if os.IsNotExist(err) {
|
return false
|
}
|
|
return false
|
}
|
|
func ShellExecute(dir, exe string, args ...string) (string, error) {
|
cmd := exec.Command(exe, args...)
|
stdout, err := cmd.StdoutPipe()
|
if nil != err {
|
msg := fmt.Sprintf("ShellExecute cmd.StdoutPipe err:%v", err)
|
return "", errors.New(msg)
|
}
|
|
cmd.Stderr = os.Stderr
|
if "" != dir {
|
cmd.Dir = dir
|
}
|
|
err = cmd.Start()
|
if nil != err {
|
msg := fmt.Sprintf("ShellExecute cmd.Start err:%v", err)
|
return "", errors.New(msg)
|
}
|
|
var buffer bytes.Buffer
|
reader := bufio.NewReader(stdout)
|
for {
|
line, err2 := reader.ReadString('\n')
|
if io.EOF == err2 {
|
break
|
}
|
|
if nil != err2 {
|
msg := fmt.Sprintf("ShellExecute reader.ReadString err:%v", err2)
|
return "", errors.New(msg)
|
}
|
|
buffer.WriteString(line)
|
}
|
|
err = cmd.Wait()
|
if nil != err {
|
msg := fmt.Sprintf("ShellExecute cmd.Wait err:%v", err)
|
return "", errors.New(msg)
|
}
|
|
return buffer.String(), nil
|
}
|
|
func DeCompressZip(zipFile, dest string) error {
|
reader, err := zip.OpenReader(zipFile)
|
if err != nil {
|
return err
|
}
|
defer reader.Close()
|
|
for _, f := range reader.Reader.File {
|
if f.FileInfo().IsDir() {
|
fullPath := path.Join(dest, f.Name)
|
err = os.MkdirAll(fullPath, os.ModePerm)
|
if err != nil {
|
return err
|
}
|
continue
|
}
|
|
r, err := f.Open()
|
if err != nil {
|
return err
|
}
|
defer r.Close()
|
|
fullPath := path.Join(dest, f.Name)
|
w, err := os.Create(fullPath)
|
if err != nil {
|
return err
|
}
|
|
defer w.Close()
|
_, err = io.Copy(w, r)
|
if err != nil {
|
return err
|
}
|
}
|
|
return nil
|
}
|
|
func Md5sum(filepath string) (string, error) {
|
f, err := os.Open(filepath)
|
if nil != err {
|
return "", err
|
}
|
defer f.Close()
|
|
payload, err := ioutil.ReadAll(f)
|
if nil != err {
|
return "", err
|
}
|
|
md5sum := fmt.Sprintf("%x", md5.Sum(payload))
|
runtime.GC()
|
|
return md5sum, nil
|
}
|
|
//强制更新文件内容，（与copy命令有区别）
|
func CopyFileContents(src, dest string) (err error) {
|
in, err := os.Open(src)
|
if nil != err {
|
return
|
}
|
defer in.Close()
|
|
if PathExists(dest) { //替换前重命名备份原来文件
|
tmpDest := dest + "1"
|
os.Rename(dest, tmpDest)
|
}
|
out, err := os.Create(dest)
|
if nil != err {
|
return
|
}
|
defer func() {
|
cErr := out.Close()
|
if nil != cErr {
|
err = cErr
|
}
|
}()
|
|
if _, err = io.Copy(out, in); err != nil {
|
return
|
}
|
|
err = out.Sync()
|
|
return
|
}
|
|
func DeCompressTGZ(gzFile, dest string) (err error) {
|
logger.Info("开始解压" + gzFile + "到" + dest + "\n")
|
command := "tar -zxf " + gzFile + " -C " + dest
|
result, err := execCommand(command)
|
logger.Info("执行命令：", command)
|
logger.Info(string(result) + "\n")
|
logger.Info("解压" + gzFile + "到" + dest + "完成！")
|
return
|
}
|
|
func CompressTGZ(gzFile, dest string) (err error) {
|
logger.Info("开始压缩" + gzFile + " " + dest + "\n")
|
command := "tar -zcf " + gzFile + " -C " + dest + " ."
|
result, err := execCommand(command)
|
if err != nil {
|
logger.Error("压缩失败"+gzFile+" "+dest+",error:", err)
|
return
|
}
|
logger.Info(string(result) + "\n")
|
logger.Info("压缩" + gzFile + " " + dest + "完成！")
|
return
|
}
|
|
//执行命令
|
func execCommand(cmd string) ([]byte, error) {
|
return exec.Command("/bin/bash", "-c", cmd).Output()
|
}
|
