完善人脸照片图片切图

zhangzengfei

2024-06-06 0b932b8af021e1f97a27c85d823f828fbf854a39

 util/http.go

@@ -3,8 +3,12 @@
import (
   "bytes"
   "encoding/json"
   "errors"
   "fmt"
   "io"
   "net/http"
   "strconv"
   "time"

   "gat1400Exchange/pkg/logger"
   "gat1400Exchange/vo"
@@ -33,6 +37,33 @@
      return nil, err
   }

   if header != nil {
      for k, v := range header {
         req.Header.Set(k, v)
      }
   }

   cli := &http.Client{}
   resp, err := cli.Do(req)
   if err != nil {
      return nil, err
   }

   defer resp.Body.Close()
   body, err := io.ReadAll(resp.Body)
   if err != nil {
      return nil, err
   }

   return body, nil
}

func HttpPut(url string, header map[string]string, data []byte) ([]byte, error) {
   req, err := http.NewRequest(http.MethodPut, url, bytes.NewBuffer(data))
   if err != nil {
      return nil, err
   }

   //req.Header.Set("Content-Type", "applicaiton/json; charset=UTF-8")
   if header != nil {
      for k, v := range header {
@@ -55,11 +86,109 @@
   return body, nil
}

func HttpGet(url string) error {
   _, err := http.Get(url)
func HttpGet(url string, header map[string]string) ([]byte, error) {
   req, err := http.NewRequest(http.MethodGet, url, nil)
   if err != nil {
      return err
      return nil, err
   }

   return nil
   //req.Header.Set("Content-Type", "applicaiton/json; charset=UTF-8")
   if header != nil {
      for k, v := range header {
         req.Header.Set(k, v)
      }
   }
   cli := &http.Client{}
   resp, err := cli.Do(req)
   if err != nil {
      return nil, err
   }

   defer resp.Body.Close()
   body, err := io.ReadAll(resp.Body)
   if err != nil {
      return nil, err
   }

   return body, nil
}

func ImageDownload(downloadUrl string, fb func(length, downLen int64)) ([]byte, error) {
   var (
      fSize   int64
      buf     = make([]byte, 10*1024)
      buffer  bytes.Buffer
      written int64
   )

   request, err := http.NewRequest("GET", downloadUrl, nil)
   if err != nil {
      return nil, err
   }

   //创建一个http client
   client := new(http.Client)
   client.Timeout = time.Second * 600 //设置超时时间

   // get方法获取资源
   resp, err := client.Do(request)
   if err != nil {
      return nil, err
   }

   if resp.StatusCode != 200 {
      return nil, errors.New(fmt.Sprintf("请求失败, 状态码:%d", resp.StatusCode))
   }

   //读取服务器返回的文件大小
   fSize, err = strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 32)
   if err != nil {
      return nil, err
   }

   if resp.Body == nil {
      return nil, errors.New("body is null")
   }

   defer resp.Body.Close()

   //下面是 io.copyBuffer() 的简化版本
   for {
      //读取bytes
      nr, er := resp.Body.Read(buf)
      if nr > 0 {
         //写入bytes
         nw, ew := buffer.Write(buf[0:nr])
         //数据长度大于0
         if nw > 0 {
            written += int64(nw)
         }

         //写入出错
         if ew != nil {
            err = ew
            break
         }

         //读取是数据长度不等于写入的数据长度
         if nr != nw {
            err = io.ErrShortWrite
            break
         }
      }

      if er != nil {
         if er != io.EOF {
            err = er
         }
         break
      }

      //没有错误了快使用 callback
      if fb != nil {
         fb(fSize, written)
      }
   }

   return buffer.Bytes(), err
}