From 0bafd001e79a7b3985ec58ea926f71de2ef7cf9d Mon Sep 17 00:00:00 2001
From: Christian Muehlhaeuser <muesli@gmail.com>
Date: Fri, 4 Dec 2020 11:01:14 +0100
Subject: [PATCH] Simplify code

---
 main.go | 26 ++++++++++++--------------
 1 file changed, 12 insertions(+), 14 deletions(-)

diff --git a/main.go b/main.go
index 7115663..8ded7d7 100644
--- a/main.go
+++ b/main.go
@@ -23,13 +23,19 @@ type RTMPConnection struct {
 	packets chan av.Packet
 }
 
-func NewRTMPConnection(u string) (*RTMPConnection, error) {
+func NewRTMPConnection(u string) *RTMPConnection {
 	r := &RTMPConnection{
-		url:     u,
-		packets: make(chan av.Packet, 2),
+		url: u,
 	}
+	r.reset()
 
-	return r, nil // r.Dial()
+	return r
+}
+
+func (r *RTMPConnection) reset() {
+	r.packets = make(chan av.Packet, 2)
+	r.conn = nil
+	r.header = nil
 }
 
 func (r *RTMPConnection) Dial() error {
@@ -57,10 +63,8 @@ func (r *RTMPConnection) Disconnect() error {
 		return err
 	}
 
-	r.conn = nil
-	r.header = nil
 	close(r.packets)
-	r.packets = make(chan av.Packet, 2)
+	r.reset()
 
 	fmt.Println("connection closed:", r.url)
 	return nil
@@ -124,13 +128,7 @@ func main() {
 
 	conns := make([]*RTMPConnection, len(flag.Args()))
 	for i, u := range flag.Args() {
-		c, err := NewRTMPConnection(u)
-		if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
-		}
-
-		conns[i] = c
+		conns[i] = NewRTMPConnection(u)
 	}
 
 	server.HandlePublish = func(conn *rtmp.Conn) {