From c9fd11390ca85a7f3fd7e30ec5f1402d362f0c12 Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期二, 17 十二月 2019 19:52:36 +0800 Subject: [PATCH] add filePort --- extend/util/util.go | 113 +++++++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 103 insertions(+), 10 deletions(-) diff --git a/extend/util/util.go b/extend/util/util.go index 77cd755..f3dc879 100644 --- a/extend/util/util.go +++ b/extend/util/util.go @@ -1,13 +1,18 @@ package util import ( - "basic.com/pubsub/protomsg.git" + "bytes" "encoding/json" - "github.com/gin-gonic/gin" - "github.com/golang/glog" - "gocv.io/x/gocv" "image" + "io/ioutil" + "net/http" + "strconv" "webserver/extend/code" + + "basic.com/pubsub/protomsg.git" + "github.com/gin-gonic/gin" + "github.com/pierrec/lz4" + "gocv.io/x/gocv" "crypto/rand" "fmt" @@ -30,7 +35,6 @@ // ResponseFormat 杩斿洖鏁版嵁鏍煎紡鍖� func ResponseFormat(c *gin.Context, respStatus *code.Code, data interface{}) { if respStatus == nil { - glog.Error("response status param not found!") respStatus = code.RequestParamError } c.JSON(respStatus.Status, gin.H{ @@ -52,14 +56,14 @@ //struct杞琺ap func Struct2Map(obj interface{}) map[string]interface{} { - resultMap :=make(map[string]interface{},0) + resultMap := make(map[string]interface{}, 0) bytesData, err := json.Marshal(obj) - if err !=nil { + if err != nil { return resultMap } json.Unmarshal(bytesData, &resultMap) return resultMap - + // //t := reflect.TypeOf(obj) //v := reflect.ValueOf(obj) @@ -71,8 +75,21 @@ //return data } +func ReadImgData(url string) ([]byte, error) { + resp, err := http.Get(url) + if err != nil { + return nil, err + } + defer resp.Body.Close() + pix, err := ioutil.ReadAll(resp.Body) + if err != nil { + return nil, err + } + return pix, nil +} + // 鎸夊昂瀵稿幓鍒囧浘 -func SubImg(i protomsg.Image, x0, y0, x1, y1 int, ) []byte { +func SubImg(i protomsg.Image, x0, y0, x1, y1 int) []byte { img, _ := gocv.NewMatFromBytes(int(i.Height), int(i.Width), gocv.MatTypeCV8UC3, i.Data) rect := image.Rect(EnlargeSize(x0, y0, x1, y1, i)) region := img.Region(rect) @@ -112,4 +129,80 @@ y1_new = int(i.Height) } return -} \ No newline at end of file +} + +func StringIpToInt(ipstring string) int { + ipSegs := strings.Split(ipstring, ".") + var ipInt int = 0 + var pos uint = 24 + for _, ipSeg := range ipSegs { + tempInt, _ := strconv.Atoi(ipSeg) + tempInt = tempInt << pos + ipInt = ipInt | tempInt + pos -= 8 + } + return ipInt +} + +func IpIntToString(ipInt int) string { + ipSegs := make([]string, 4) + var len int = len(ipSegs) + buffer := bytes.NewBufferString("") + for i := 0; i < len; i++ { + tempInt := ipInt & 0xFF + ipSegs[len-i-1] = strconv.Itoa(tempInt) + + ipInt = ipInt >> 8 + } + for i := 0; i < len; i++ { + buffer.WriteString(ipSegs[i]) + if i < len-1 { + buffer.WriteString(".") + } + } + return buffer.String() +} + +func ParseScore(compareScore float32) float32 { + if compareScore < 1 { + compareScore = compareScore * 100 + } + f, _ := strconv.ParseFloat(fmt.Sprintf("%2.2f", compareScore), 32) + return float32(f) +} + +func ParseScore64(compareScore float64) float64 { + if compareScore < 1 { + compareScore = compareScore * 100 + } + f, _ := strconv.ParseFloat(fmt.Sprintf("%2.2f", compareScore), 64) + return f +} + +// UnCompress uncompress +func UnCompress(in []byte) ([]byte, error) { + out := make([]byte, 10*len(in)) + n, err := lz4.UncompressBlock(in, out) + if err != nil { + fmt.Println(err) + return nil, err + } + out = out[:n] // uncompressed data + return out, nil +} + +// Compress compress +func Compress(in []byte) ([]byte, error) { + out := make([]byte, len(in)) + ht := make([]int, 64<<10) // buffer for the compression table + n, err := lz4.CompressBlock(in, out, ht) + if err != nil { + fmt.Println(err) + return nil, err + } + if n >= len(in) { + fmt.Println("image is not compressible") + } + out = out[:n] // compressed data + return out, nil +} -- Gitblit v1.8.0