Fix listen example error handling

Don't call hub.Register if Accept returned error.
pion · Mar 23, 2020 · 3f3108d · 3f3108d
1 parent 8d11fee
commit 3f3108d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/examples/listen-psk/main.go b/examples/listen-psk/main.go
@@ -61,7 +61,9 @@ func main() {
 			// functions like `RemoteCertificate` etc.
 
 			// Register the connection with the chat hub
-			hub.Register(conn)
+			if err == nil {
+				hub.Register(conn)
+			}
 		}
 	}()
 

diff --git a/examples/listen/main.go b/examples/listen/main.go
@@ -62,7 +62,9 @@ func main() {
 			// functions like `RemoteCertificate` etc.
 
 			// Register the connection with the chat hub
-			hub.Register(conn)
+			if err == nil {
+				hub.Register(conn)
+			}
 		}
 	}()