update token

liuxiaolong

2020-06-06 aa26c2c62693dcf7e8484fbc831e67b040db2fa0

 service/CamraUpdatesnashot.go

@@ -1,98 +1,15 @@
package service

import (
   "flag"
   "fmt"
   "os"
   "os/exec"
   "strconv"
   "time"
   "webserver/extend/config"

   "bytes"
   "encoding/json"
   "errors"
   "gocv.io/x/gocv"
   "io"
   "log"
   "mime/multipart"
   "net/http"
   "path/filepath"

   "basic.com/dbapi.git"
   "basic.com/fileServer/WeedFSClient.git"
   "basic.com/valib/capture.git"
   "basic.com/valib/logger.git"
   "errors"
   "strconv"
   "webserver/cache"
)

func PostFormBufferData(uri string, filepath string, fileName string) (maps map[string]interface{}, err0 error) {
   // 要指定转byte的格式
   picMat := gocv.IMRead(filepath, gocv.IMReadColor)
   defer picMat.Close()
   if picMat.Empty() {
      return nil, errors.New("file not exist")
   }

   pheight := picMat.Rows()
   pwidth := picMat.Cols()
   pdata := picMat.ToBytes()

   imgs := gocv.NewMat()
   imgs, _ = gocv.NewMatFromBytes(pheight, pwidth, gocv.MatTypeCV8UC3, pdata)
   fdata, _ := gocv.IMEncode(".jpg", imgs)
   body := &bytes.Buffer{}
   writer := multipart.NewWriter(body)
   _, err := writer.CreateFormFile("file", fileName)
   if err != nil {
      return nil, err
   }
   boundary := writer.Boundary()
   //close_string := fmt.Sprintf("\r\n--%s--\r\n", boundary)
   close_buf := bytes.NewBufferString(fmt.Sprintf("\r\n--%s--\r\n", boundary))
   file := bytes.NewBuffer(fdata)
   request_reader := io.MultiReader(body, file, close_buf)
   //_, err = io.Copy(part, file)
   //writer.WriteField(key, val)
   request, err := http.NewRequest("POST", uri, request_reader)
   request.Header.Add("Content-Type", writer.FormDataContentType())
   timeout := time.Duration(5 * time.Second) //超时时间50ms
   client := &http.Client{Timeout: timeout}
   resp, err := client.Do(request)
   if err != nil {
      log.Fatal(err)
      return nil, err
   }
   defer func() {
      if r := recover(); r != nil {
         fmt.Printf("panic的内容%v\n", r)
         msg := "上传图片服务器异常"
         if _, ok := r.(error); ok {
            msg = r.(error).Error()
            fmt.Println("panic--recover()得到的是error类型")
         }
         if _, ok := r.(string); ok {
            msg = r.(string)
            fmt.Println("panic--recover()得到的是string类型")
         }
         err0 = errors.New(msg)
      }
   }()
   defer resp.Body.Close()
   {
      body := &bytes.Buffer{}
      _, err := body.ReadFrom(resp.Body)
      if err != nil {
         log.Fatal(err)
      }
      fmt.Println(resp.StatusCode)
      //fmt.Println(resp.Header)
      fmt.Println(body)
      //decoder := json.NewDecoder(strings.NewReader(body.String()))
      decoder := make(map[string]interface{})
      if err := json.Unmarshal([]byte(body.String()), &decoder); err != nil {
         return nil, err
      }
      return decoder, nil
   }
}

/*以下属于旧版本地摄像机获取截图（国标摄像机底图无法刷新）
func processphoto(cid string) (local string, err error) {
   var camApi dbapi.CameraApi
   caminfo, err := camApi.GetCameraById(cid)
@@ -142,4 +59,39 @@
      return "", errors.New("update filelocal to camera fail")
   }
   return filename, nil
}*/

//调用统一接口，刷新底图（集成国标底图刷新）
func UpdateCapture(cid string) (fileName string,err error){
   var cameraApi dbapi.CameraApi
   camera, err := cameraApi.GetCameraById(cid)
   if err !=nil{
      return "",errors.New("camera not exist")
   }

   m := capture.Rtsp
   if camera.Type == 1{//国标摄像机
      m = capture.GB28181
   }
   b,err := capture.Capture("libcffmpeg.so",m,camera.Rtsp,capture.JPEGFileExt,1280,720,10)
   if err == nil{
      localConf, err2 := cache.GetServerInfo()
      if err2 !=nil || localConf.WebPicIp == "" {
         logger.Debug("localConfig is wrong!!!")
         return "",err2
      }
      var weedfsUri = "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection=persistent"
      weedFilePath, err := WeedFSClient.UploadFile(weedfsUri, camera.Name+".jpg", b)
      if err != nil {
         return "",err
      } else {
         ok := cameraApi.UpdateSnapshotUrl(cid, weedFilePath)
         if !ok {
            return "", errors.New("update camera's snapshot fail")
         }
         return weedFilePath, nil
      }
   }else{
      return "",err
   }
}