zishuo/webrtc
1
0
Fork 0
mirror of https://github.com/pion/webrtc.git synced 2025-10-07 08:01:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Simplify examples/ice-tcp

Browse Source 
Example was copied from play-from-disk-renegotation and contained
a global PeerConnection. Instead create a PeerConnection on each inbound
request
This commit is contained in:
Sean DuBois
2022-02-18 22:40:26 -05:00
parent 94262c1b2b
commit b458528350
1 changed files with 49 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

97
examples/ice-tcp/main.go
View File

@@ -5,7 +5,9 @@ package main
import ( import (
"encoding/json" "encoding/json"
"errors"
"fmt" "fmt"
"io"
"net" "net"
"net/http" "net/http"
"time" "time"
@@ -13,58 +15,34 @@ import (
"github.com/pion/webrtc/v3" "github.com/pion/webrtc/v3"
) )
var peerConnection *webrtc.PeerConnection //nolint var api *webrtc.API //nolint
func doSignaling(w http.ResponseWriter, r *http.Request) { func doSignaling(w http.ResponseWriter, r *http.Request) {
var err error peerConnection, err := api.NewPeerConnection(webrtc.Configuration{})
if err != nil {
if peerConnection == nil { panic(err)
settingEngine := webrtc.SettingEngine{}
// Enable support only for TCP ICE candidates.
settingEngine.SetNetworkTypes([]webrtc.NetworkType{
webrtc.NetworkTypeTCP4,
webrtc.NetworkTypeTCP6,
})
var tcpListener net.Listener
tcpListener, err = net.ListenTCP("tcp", &net.TCPAddr{
IP: net.IP{0, 0, 0, 0},
Port: 8443,
})
if err != nil {
panic(err)
}
fmt.Printf("Listening for ICE TCP at %s\n", tcpListener.Addr())
tcpMux := webrtc.NewICETCPMux(nil, tcpListener, 8)
settingEngine.SetICETCPMux(tcpMux)
api := webrtc.NewAPI(webrtc.WithSettingEngine(settingEngine))
if peerConnection, err = api.NewPeerConnection(webrtc.Configuration{}); err != nil {
panic(err)
}
// Set the handler for ICE connection state
// This will notify you when the peer has connected/disconnected
peerConnection.OnICEConnectionStateChange(func(connectionState webrtc.ICEConnectionState) {
fmt.Printf("ICE Connection State has changed: %s\n", connectionState.String())
})
// Send the current time via a DataChannel to the remote peer every 3 seconds
peerConnection.OnDataChannel(func(d *webrtc.DataChannel) {
d.OnOpen(func() {
for range time.Tick(time.Second * 3) {
if err = d.SendText(time.Now().String()); err != nil {
panic(err)
}
}
})
})
} }
// Set the handler for ICE connection state
// This will notify you when the peer has connected/disconnected
peerConnection.OnICEConnectionStateChange(func(connectionState webrtc.ICEConnectionState) {
fmt.Printf("ICE Connection State has changed: %s\n", connectionState.String())
})
// Send the current time via a DataChannel to the remote peer every 3 seconds
peerConnection.OnDataChannel(func(d *webrtc.DataChannel) {
d.OnOpen(func() {
for range time.Tick(time.Second * 3) {
if err = d.SendText(time.Now().String()); err != nil {
if errors.Is(io.ErrClosedPipe, err) {
return
}
panic(err)
}
}
})
})
var offer webrtc.SessionDescription var offer webrtc.SessionDescription
if err = json.NewDecoder(r.Body).Decode(&offer); err != nil { if err = json.NewDecoder(r.Body).Decode(&offer); err != nil {
panic(err) panic(err)
@@ -101,6 +79,29 @@ func doSignaling(w http.ResponseWriter, r *http.Request) {
} }
func main() { func main() {
settingEngine := webrtc.SettingEngine{}
// Enable support only for TCP ICE candidates.
settingEngine.SetNetworkTypes([]webrtc.NetworkType{
webrtc.NetworkTypeTCP4,
webrtc.NetworkTypeTCP6,
})
tcpListener, err := net.ListenTCP("tcp", &net.TCPAddr{
IP: net.IP{0, 0, 0, 0},
Port: 8443,
})
if err != nil {
panic(err)
}
fmt.Printf("Listening for ICE TCP at %s\n", tcpListener.Addr())
tcpMux := webrtc.NewICETCPMux(nil, tcpListener, 8)
settingEngine.SetICETCPMux(tcpMux)
api = webrtc.NewAPI(webrtc.WithSettingEngine(settingEngine))
http.Handle("/", http.FileServer(http.Dir("."))) http.Handle("/", http.FileServer(http.Dir(".")))
http.HandleFunc("/doSignaling", doSignaling) http.HandleFunc("/doSignaling", doSignaling)