mirror of
https://github.com/Monibuca/plugin-record.git
synced 2025-10-07 09:41:07 +08:00
309 lines
9.9 KiB
Go
309 lines
9.9 KiB
Go
package record
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"io/ioutil"
|
||
"log"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"m7s.live/engine/v4/util"
|
||
)
|
||
|
||
var mu sync.Mutex
|
||
|
||
func errorJsonString(args map[string]interface{}) string {
|
||
resultJsonData := make(map[string]interface{})
|
||
for field, value := range args {
|
||
resultJsonData[field] = value
|
||
}
|
||
jsonString, _ := json.Marshal(resultJsonData)
|
||
return string(jsonString)
|
||
}
|
||
|
||
// 根据事件id拉取录像流
|
||
func (conf *RecordConfig) API_event_pull(w http.ResponseWriter, r *http.Request) {
|
||
token := r.Header.Get("token")
|
||
resultJsonData := make(map[string]interface{})
|
||
resultJsonData["code"] = -1
|
||
if token == "" || token != "m7s" {
|
||
resultJsonData["msg"] = "token错误"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
var postData map[string]interface{}
|
||
body, err := io.ReadAll(r.Body)
|
||
if err != nil {
|
||
resultJsonData["msg"] = "Unable to read request body "
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
err = json.Unmarshal(body, &postData)
|
||
id := int(postData["id"].(float64))
|
||
if id > 0 {
|
||
var eventRecord EventRecord
|
||
result := mysqldb.First(&eventRecord, id) // 根据主键查询
|
||
if result.Error != nil {
|
||
log.Println("Error finding eventrecord:", result.Error)
|
||
}
|
||
fileUrlPath := eventRecord.Urlpath
|
||
if fileUrlPath != "" {
|
||
// 判断协议类型
|
||
scheme := "http"
|
||
if r.TLS != nil {
|
||
scheme = "https" // 如果 r.TLS 不为空，则表示是 HTTPS
|
||
}
|
||
// 获取完整请求的地址：包含协议、主机和端口
|
||
requestHost := fmt.Sprintf("%s://%s", scheme, r.Host)
|
||
fileUrlPath = requestHost + "/" + fileUrlPath
|
||
newStreamPath := eventRecord.StreamPath + "/" + strings.TrimSuffix(eventRecord.Filename, ".flv")
|
||
requestFullPath := fmt.Sprintf(requestHost+"/hdl/api/pull?streamPath=%s&target=%s", newStreamPath, fileUrlPath)
|
||
resp, err := http.Get(requestFullPath)
|
||
if err != nil {
|
||
log.Fatal("Error while making GET request:", err)
|
||
}
|
||
defer resp.Body.Close() // 确保函数结束后关闭响应体
|
||
|
||
// 读取响应体
|
||
body, err := ioutil.ReadAll(resp.Body)
|
||
if err != nil {
|
||
log.Fatal("Error reading response body:", err)
|
||
}
|
||
|
||
// 打印响应内容
|
||
fmt.Println("Response Status:", resp.Status)
|
||
fmt.Println("Response Body:", string(body))
|
||
resultJsonData["streamPath"] = newStreamPath
|
||
resultJsonData["code"] = 0
|
||
resultJsonData["msg"] = ""
|
||
util.ReturnError(util.APIErrorNone, errorJsonString(resultJsonData), w, r)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 录像报警列表
|
||
func (conf *RecordConfig) API_alarm_list(w http.ResponseWriter, r *http.Request) {
|
||
token := r.Header.Get("token")
|
||
resultJsonData := make(map[string]interface{})
|
||
resultJsonData["code"] = -1
|
||
if token == "" || token != "m7s" {
|
||
resultJsonData["msg"] = "token错误"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
//TODO 用token验证用户信息是否有效，并获取用户信息换取userid
|
||
var postData map[string]interface{}
|
||
body, err := io.ReadAll(r.Body)
|
||
if err != nil {
|
||
resultJsonData["msg"] = "Unable to read request body "
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
err = json.Unmarshal(body, &postData)
|
||
pageNum := postData["pageNum"].(float64)
|
||
if pageNum <= 0 {
|
||
resultJsonData["msg"] = "pageNum error"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
pageSize := postData["pageSize"].(float64)
|
||
if pageSize <= 0 {
|
||
resultJsonData["msg"] = "pageSize error"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
exceptions, totalCount, err := paginate(Exception{}, int(pageNum), int(pageSize), postData)
|
||
if err != nil {
|
||
resultJsonData["msg"] = err.Error()
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
resultJsonData["totalCount"] = totalCount
|
||
resultJsonData["pageNum"] = pageNum
|
||
resultJsonData["pageSize"] = pageSize
|
||
resultJsonData["list"] = exceptions
|
||
resultJsonData["code"] = 0
|
||
resultJsonData["msg"] = ""
|
||
util.ReturnError(util.APIErrorNone, errorJsonString(resultJsonData), w, r)
|
||
}
|
||
|
||
// 事件录像列表
|
||
func (conf *RecordConfig) API_event_list(w http.ResponseWriter, r *http.Request) {
|
||
token := r.Header.Get("token")
|
||
resultJsonData := make(map[string]interface{})
|
||
resultJsonData["code"] = -1
|
||
if token == "" || token != "m7s" {
|
||
resultJsonData["msg"] = "token错误"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
//TODO 用token验证用户信息是否有效，并获取用户信息换取userid
|
||
var postData map[string]interface{}
|
||
body, err := io.ReadAll(r.Body)
|
||
if err != nil {
|
||
resultJsonData["msg"] = "Unable to read request body "
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
err = json.Unmarshal(body, &postData)
|
||
pageNum := postData["pageNum"].(float64)
|
||
if pageNum <= 0 {
|
||
resultJsonData["msg"] = "pageNum error"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
pageSize := postData["pageSize"].(float64)
|
||
if pageSize <= 0 {
|
||
resultJsonData["msg"] = "pageSize error"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
eventRecords, totalCount, err := paginate(EventRecord{}, int(pageNum), int(pageSize), postData)
|
||
if err != nil {
|
||
resultJsonData["msg"] = err.Error()
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
resultJsonData["totalCount"] = totalCount
|
||
resultJsonData["pageNum"] = pageNum
|
||
resultJsonData["pageSize"] = pageSize
|
||
resultJsonData["list"] = eventRecords
|
||
resultJsonData["code"] = 0
|
||
resultJsonData["msg"] = ""
|
||
util.ReturnError(util.APIErrorNone, errorJsonString(resultJsonData), w, r)
|
||
}
|
||
|
||
// 事件录像
|
||
func (conf *RecordConfig) API_event_start(w http.ResponseWriter, r *http.Request) {
|
||
mu.Lock()
|
||
defer mu.Unlock()
|
||
token := r.Header.Get("token")
|
||
resultJsonData := make(map[string]interface{})
|
||
resultJsonData["code"] = -1
|
||
if token == "" || token != "m7s" {
|
||
resultJsonData["msg"] = "token错误"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
//TODO 用token验证用户信息是否有效，并获取用户信息换取userid
|
||
var eventRecordModel EventRecord
|
||
// 读取请求体
|
||
body, err := io.ReadAll(r.Body)
|
||
if err != nil {
|
||
resultJsonData["msg"] = "Unable to read request body "
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
// 解析JSON数据到map
|
||
err = json.Unmarshal(body, &eventRecordModel)
|
||
if err != nil {
|
||
resultJsonData["msg"] = "Invalid JSON format "
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
streamPath := eventRecordModel.StreamPath
|
||
if streamPath == "" {
|
||
resultJsonData["msg"] = "no streamPath"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
//TODO 获取到磁盘容量低，磁盘报错的情况下需要报异常，并且根据事件类型做出处理
|
||
if getDisckException(streamPath) {
|
||
resultJsonData["msg"] = "disk is full"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
eventId := eventRecordModel.EventId
|
||
if eventId == "" {
|
||
resultJsonData["msg"] = "no eventId"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
//recordMode := eventRecordModel.RecordMode
|
||
//if recordMode == "" {
|
||
// resultJsonData["msg"] = "no recordMode"
|
||
// util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
// return
|
||
//}
|
||
eventName := eventRecordModel.EventName
|
||
if eventName == "" {
|
||
resultJsonData["msg"] = "no eventName"
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
beforeDuration := eventRecordModel.BeforeDuration
|
||
if beforeDuration == "" {
|
||
beforeDuration = strconv.Itoa(conf.beforeDuration)
|
||
}
|
||
afterDuration := eventRecordModel.AfterDuration
|
||
if afterDuration == "" {
|
||
afterDuration = strconv.Itoa(conf.afterDuration)
|
||
}
|
||
recordTime := time.Now().Format("2006-01-02 15:04:05")
|
||
fileName := time.Now().Format("20060102150405")
|
||
startTime := time.Now().Add(-30 * time.Second).Format("2006-01-02 15:04:05")
|
||
endTime := time.Now().Add(30 * time.Second).Format("2006-01-02 15:04:05")
|
||
//切片大小
|
||
fragment := eventRecordModel.Fragment
|
||
//var id string
|
||
irecorder := NewFLVRecorder()
|
||
found := false
|
||
conf.recordings.Range(func(key, value any) bool {
|
||
tmpIRecorder := value.(*FLVRecorder)
|
||
existStreamPath := tmpIRecorder.GetSubscriber().Stream.Path
|
||
if existStreamPath == streamPath {
|
||
irecorder = tmpIRecorder
|
||
found = true
|
||
}
|
||
return found
|
||
})
|
||
recorder := irecorder.GetRecorder()
|
||
recorder.FileName = fileName
|
||
recorder.append = false
|
||
filepath := conf.Flv.Path + "/" + streamPath + "/" + fileName + recorder.Ext
|
||
urlpath := "record/" + streamPath + "/" + fileName + recorder.Ext
|
||
if fragment != "" {
|
||
if f, err := time.ParseDuration(fragment); err == nil {
|
||
recorder.Fragment = f
|
||
}
|
||
}
|
||
|
||
if found {
|
||
irecorder.UpdateTimeout(30 * time.Second)
|
||
} else {
|
||
err = irecorder.StartWithDynamicTimeout(streamPath, fileName, 30*time.Second)
|
||
}
|
||
//go func() {
|
||
// timer := time.NewTimer(30 * time.Second)
|
||
//
|
||
// // 等待计时器到期
|
||
// <-timer.C
|
||
// irecorder.Stop(zap.String("reason", "api"))
|
||
//}()
|
||
//id = recorder.ID
|
||
if err != nil {
|
||
exceptionChannel <- &Exception{AlarmType: "record", AlarmDesc: "录像失败", StreamPath: streamPath}
|
||
resultJsonData["msg"] = err.Error()
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
eventRecord := EventRecord{StreamPath: streamPath, EventId: eventId, RecordMode: "1", EventName: eventName, BeforeDuration: beforeDuration,
|
||
AfterDuration: afterDuration, CreateTime: recordTime, StartTime: startTime, EndTime: endTime, Filepath: filepath, Filename: fileName + recorder.Ext, EventDesc: eventRecordModel.EventDesc, Urlpath: urlpath}
|
||
err = mysqldb.Omit("id", "fragment", "isDelete").Create(&eventRecord).Error
|
||
if err != nil {
|
||
resultJsonData["msg"] = err.Error()
|
||
util.ReturnError(-1, errorJsonString(resultJsonData), w, r)
|
||
return
|
||
}
|
||
outid := eventRecord.Id
|
||
resultJsonData["id"] = outid
|
||
resultJsonData["code"] = 0
|
||
resultJsonData["msg"] = ""
|
||
util.ReturnError(util.APIErrorNone, errorJsonString(resultJsonData), w, r)
|
||
}
|