Fixes with RTPRelay and actioncam
Endless loop in RTPRelay and exiting from whole app using Ctrl+C
This commit is contained in:
parent
e82f5a97c9
commit
205c9896dc
2 changed files with 61 additions and 55 deletions
|
|
@ -71,6 +71,7 @@ func main() {
|
||||||
case sig := <-signalChannel:
|
case sig := <-signalChannel:
|
||||||
log.Printf("Got signal %s, exiting...\n", sig)
|
log.Printf("Got signal %s, exiting...\n", sig)
|
||||||
cancel()
|
cancel()
|
||||||
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
}(cancel)
|
}(cancel)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -20,6 +20,8 @@ type RTPRelay struct {
|
||||||
context context.Context
|
context context.Context
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var close bool
|
||||||
|
|
||||||
// CreateRTPRelay creates a UDP listener that handles live data
|
// CreateRTPRelay creates a UDP listener that handles live data
|
||||||
// from the camera and forwards it as an RTP stream
|
// from the camera and forwards it as an RTP stream
|
||||||
func CreateRTPRelay(ctx context.Context, targetAddress net.IP, targetPort int) *RTPRelay {
|
func CreateRTPRelay(ctx context.Context, targetAddress net.IP, targetPort int) *RTPRelay {
|
||||||
|
|
@ -29,7 +31,9 @@ func CreateRTPRelay(ctx context.Context, targetAddress net.IP, targetPort int) *
|
||||||
log.Printf("ERROR: %s\n", err)
|
log.Printf("ERROR: %s\n", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
close = false
|
||||||
relay := RTPRelay{
|
relay := RTPRelay{
|
||||||
|
close: false,
|
||||||
targetIP: targetAddress,
|
targetIP: targetAddress,
|
||||||
targetPort: targetPort,
|
targetPort: targetPort,
|
||||||
listener: conn,
|
listener: conn,
|
||||||
|
|
@ -66,7 +70,7 @@ func handleCameraStream(relay RTPRelay, conn net.PacketConn) {
|
||||||
|
|
||||||
frameBuffer := bytes.Buffer{}
|
frameBuffer := bytes.Buffer{}
|
||||||
packetBuffer := bytes.Buffer{}
|
packetBuffer := bytes.Buffer{}
|
||||||
|
T:
|
||||||
for {
|
for {
|
||||||
conn.SetReadDeadline(time.Now().Add(10 * time.Second))
|
conn.SetReadDeadline(time.Now().Add(10 * time.Second))
|
||||||
|
|
||||||
|
|
@ -75,12 +79,12 @@ func handleCameraStream(relay RTPRelay, conn net.PacketConn) {
|
||||||
log.Println("Context Done")
|
log.Println("Context Done")
|
||||||
rtpConn.Close()
|
rtpConn.Close()
|
||||||
relay.listener.Close()
|
relay.listener.Close()
|
||||||
break
|
break T
|
||||||
default:
|
default:
|
||||||
if relay.close {
|
if close {
|
||||||
rtpConn.Close()
|
rtpConn.Close()
|
||||||
relay.listener.Close()
|
relay.listener.Close()
|
||||||
break
|
break T
|
||||||
}
|
}
|
||||||
|
|
||||||
conn.ReadFrom(buffer)
|
conn.ReadFrom(buffer)
|
||||||
|
|
@ -136,6 +140,7 @@ func handleCameraStream(relay RTPRelay, conn net.PacketConn) {
|
||||||
|
|
||||||
// Stop stops listening for packets
|
// Stop stops listening for packets
|
||||||
func (r *RTPRelay) Stop() {
|
func (r *RTPRelay) Stop() {
|
||||||
r.listener.Close()
|
close = true
|
||||||
r.close = true
|
r.close = true
|
||||||
|
r.listener.Close()
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue