From 2fdef0f732a5d6549d2c42a116dfdd3dc75a0b48 Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期一, 06 一月 2020 16:25:34 +0800 Subject: [PATCH] merge ynPatch --- extend/util/util.go | 120 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 120 insertions(+), 0 deletions(-) diff --git a/extend/util/util.go b/extend/util/util.go index f3dc879..555bc37 100644 --- a/extend/util/util.go +++ b/extend/util/util.go @@ -1,12 +1,19 @@ package util import ( + "archive/zip" "bytes" + "crypto/md5" + "encoding/hex" "encoding/json" "image" + "io" "io/ioutil" "net/http" + "os" + "path/filepath" "strconv" + "time" "webserver/extend/code" "basic.com/pubsub/protomsg.git" @@ -206,3 +213,116 @@ out = out[:n] // compressed data return out, nil } + +// 鍒ゆ柇鎵�缁欒矾寰勬枃浠�/鏂囦欢澶规槸鍚﹀瓨鍦� +func Exists(path string) bool { + if path == "" { + return false + } + _, err := os.Stat(path) //os.Stat鑾峰彇鏂囦欢淇℃伅 + if err != nil { + if os.IsExist(err) { + return true + } + return false + } + return true +} +func CreateDirectory(path string) bool { + if path == "" { + return false + } + if Exists(path) { + return true + } + err := os.MkdirAll(path,os.ModePerm) + if err !=nil { + return false + } + return true +} + +//鏃ユ湡杞瓧绗︿覆 +func FormatDate(date time.Time, layout string) string { + layout = strings.Replace(layout, "yyyy", "2006", 1) + layout = strings.Replace(layout, "yy", "06", 1) + layout = strings.Replace(layout, "MM", "01", 1) + layout = strings.Replace(layout, "dd", "02", 1) + layout = strings.Replace(layout, "HH", "15", 1) + layout = strings.Replace(layout, "mm", "04", 1) + layout = strings.Replace(layout, "ss", "05", 1) + layout = strings.Replace(layout, "SSS", "000", -1) + return date.Format(layout) +} +func ZipCheck(zipPath string) bool { + f, err := os.Open(zipPath) + if err != nil { + return false + } + defer f.Close() + buf := make([]byte, 4) + if n, err := f.Read(buf); err != nil || n < 4 { + return false + } + return bytes.Equal(buf, []byte("PK\x03\x04")) +} +func UnZip(archive, target string) error { + reader, err := zip.OpenReader(archive) + if err != nil { + return err + } + if err := os.MkdirAll(target, 0755); err != nil { + return err + } + for _, file := range reader.File { + path := filepath.Join(target, file.Name) + if file.FileInfo().IsDir() { + os.MkdirAll(path, file.Mode()) + continue + } + fileReader, err := file.Open() + if err != nil { + return err + } + defer fileReader.Close() + targetFile, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode()) + if err != nil { + return err + } + defer targetFile.Close() + if _, err := io.Copy(targetFile, fileReader); err != nil { + return err + } + } + + return nil +} + +func FormatNum(oNum int,n int) string { + m := 0 + for { + oNum = oNum / 10 + m++ + if oNum == 0 { + break + } + } + fmtStr := "%0"+strconv.Itoa(m)+"d" + return fmt.Sprintf(fmtStr, n) +} + +func FileMd5(path string) (string,error){ + file, err := os.Open(path) + if err !=nil { + return "",err + } + + defer file.Close() + + _md5 := md5.New() + if _,err := io.Copy(_md5, file);err != nil { + return "",err + } + + return hex.EncodeToString(_md5.Sum(nil)),nil +} \ No newline at end of file -- Gitblit v1.8.0