| | |
| | | package util |
| | | |
| | | import ( |
| | | "archive/zip" |
| | | "bytes" |
| | | "encoding/json" |
| | | "image" |
| | | "io" |
| | | "io/ioutil" |
| | | "net/http" |
| | | "os" |
| | | "path/filepath" |
| | | "strconv" |
| | | "time" |
| | | "webserver/extend/code" |
| | | |
| | | "basic.com/pubsub/protomsg.git" |
| | |
| | | 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 |
| | | } |