From 19445d14e73f4bb96e218a65e126ae526f89537d Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 08 五月 2019 14:14:59 +0800
Subject: [PATCH] context使用
---
decoder/main.go | 61 ++++++++++++++++++++----------
1 files changed, 40 insertions(+), 21 deletions(-)
diff --git a/decoder/main.go b/decoder/main.go
index 97b3496..eb39bae 100644
--- a/decoder/main.go
+++ b/decoder/main.go
@@ -2,6 +2,7 @@
import (
"context"
+ "decoder/valib/goffmpeg"
"decoder/valib/ipc"
srv "decoder/work/service"
"encoding/json"
@@ -48,9 +49,6 @@
var (
mapCameraInfo = make(map[string]CameraInfo)
-
- tcp = `tcp://192.168.1.124`
- port = 7000
)
func recvCameraInfoFromIPC(ctx context.Context, url string, ch chan<- CameraInfo) {
@@ -59,18 +57,24 @@
for {
msg := ipc.Recv()
if msg != nil {
- fmt.Println(string(msg))
+
var c CameraInfo
if err := json.Unmarshal(msg, &c); err == nil {
+
+ if _, ok := mapCameraInfo[c.ID]; ok {
+ continue
+ }
+
ch <- c
- fmt.Printf("recv camere info %+v\n", c)
+
+ msgIpc := MsgIPC{"new decoder", 0}
+ if b, err := json.Marshal(msgIpc); err == nil {
+ ipc.Send(b)
+ }
} else {
fmt.Println(err)
}
- msgIpc := MsgIPC{"new decoder", port}
- if b, err := json.Marshal(msgIpc); err == nil {
- ipc.Send(b)
- }
+
}
}
}
@@ -78,16 +82,29 @@
func main() {
flag.Parse()
- if testIt {
- test()
- }
+ goffmpeg.InitFFmpeg()
ctx, cancel := context.WithCancel(context.Background())
+ if testIt {
+ test(ctx)
+ }
+
ch := make(chan CameraInfo)
- tcpURL := tcp + strconv.Itoa(port)
- port++
- go recvCameraInfoFromIPC(ctx, tcpURL, ch)
+
+ tcp := ``
+ port := 7001
+
+ if strings.Index(ipcURL, "tcp://") == 0 {
+ i := strings.LastIndex(ipcURL, ":")
+ tcp = ipcURL[0 : i+1]
+
+ strPort := ipcURL[i+1:]
+ port, _ = strconv.Atoi(strPort)
+ port++
+ }
+
+ go recvCameraInfoFromIPC(ctx, ipcURL, ch)
for {
select {
@@ -98,29 +115,31 @@
mapCameraInfo[c.ID] = c
ipcAddr := tcp + strconv.Itoa(port)
port++
- fmt.Printf("create ipc %s for decode : %s, on camera id %s\n", ipcAddr, c.URL, c.ID)
url := strings.TrimSpace(c.URL)
id := strings.TrimSpace(c.ID)
addr := strings.TrimSpace(ipcAddr)
- go runSender(id, url, addr)
+ go runSender(ctx, id, url, addr)
}
}
}
+
+ goffmpeg.FreeFFmpeg()
cancel()
}
-func runSender(cameraID, rtspURL, ipcLabel string) {
- d := srv.NewSender(cameraID, rtspURL, ipcLabel)
+func runSender(ctx context.Context, cameraID, rtspURL, ipcLabel string) {
+ d := srv.NewSender(ctx, cameraID, rtspURL, ipcLabel)
if asServer {
d.RunAsServer()
}
d.RunAsClient()
}
-func test() {
+
+func test(ctx context.Context) {
fmt.Println("start test")
fmt.Println(picFolder)
- runSender("cameraid", streamURL, ipcURL)
+ runSender(ctx, "cameraid", streamURL, ipcURL)
}
--
Gitblit v1.8.0