79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
|
package handler
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"VideoStream/proto"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"gocv.io/x/gocv"
|
|||
|
|
"io"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func SetUpToolGroup(router *gin.Engine) {
|
|||
|
|
toolGroup := router.Group("/tool")
|
|||
|
|
toolGroup.GET("/video_stream", GetVideoStream)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type videoStreamReq struct {
|
|||
|
|
ID int `json:"id" form:"id"`
|
|||
|
|
Key string `json:"key" form:"key"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func GetVideoStream(c *gin.Context) {
|
|||
|
|
var req videoStreamReq
|
|||
|
|
if err := c.ShouldBind(&req); err != nil {
|
|||
|
|
c.JSON(400, gin.H{"error": err.Error()})
|
|||
|
|
return
|
|||
|
|
} else {
|
|||
|
|
//查看id是否存在
|
|||
|
|
index := -1
|
|||
|
|
for _, device := range proto.Config.DeviceInfo {
|
|||
|
|
if device.ID == req.ID {
|
|||
|
|
index = req.ID
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if index == -1 {
|
|||
|
|
c.JSON(400, gin.H{"error": "id not exist"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
//查看key是否正确
|
|||
|
|
if req.Key != "123456" {
|
|||
|
|
c.JSON(400, gin.H{"error": "key error"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
//设备流
|
|||
|
|
c.Stream(func(w io.Writer) bool {
|
|||
|
|
var count int
|
|||
|
|
for {
|
|||
|
|
frame, cnt := GetDeviceCurrentFrame(req.ID)
|
|||
|
|
if cnt == count {
|
|||
|
|
time.Sleep(50 * time.Millisecond)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
//gocv.Matrix转为jpeg
|
|||
|
|
img, err := gocv.IMEncode(".jpg", frame)
|
|||
|
|
frame_ := img.GetBytes()
|
|||
|
|
|
|||
|
|
_, err = w.Write([]byte("--frame\r\nContent-Type: image/jpeg\r\n\r\n"))
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Printf("写入头部信息错误: %v\n", err)
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
_, err = w.Write(frame_)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Printf("写入帧数据错误: %v\n", err)
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
_, err = w.Write([]byte("\r\n"))
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Printf("写入帧结束标记错误: %v\n", err)
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
time.Sleep(50 * time.Millisecond) // 控制帧率，模拟每秒约20帧，可按实际调整
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|