Merge pull request #4 from ourfor-pp/master

增加跨域处理&&修复UDP占用持续增长的bug
This commit is contained in:
dexter
2020-08-31 11:31:11 +08:00
committed by GitHub
2 changed files with 12 additions and 1 deletions

3
.gitignore vendored
View File

@@ -1,3 +1,4 @@
.idea .idea
.vscode .vscode
node_modules node_modules
/.history

10
main.go
View File

@@ -140,6 +140,7 @@ func (rtc *WebRTC) Play(streamPath string) bool {
switch connectionState { switch connectionState {
case ICEConnectionStateDisconnected: case ICEConnectionStateDisconnected:
sub.Close() sub.Close()
rtc.Close()
case ICEConnectionStateConnected: case ICEConnectionStateConnected:
//rtc.videoTrack = rtc.GetSenders()[0].Track() //rtc.videoTrack = rtc.GetSenders()[0].Track()
@@ -240,6 +241,15 @@ func (rtc *WebRTC) GetAnswer() ([]byte, error) {
func run() { func run() {
http.HandleFunc("/webrtc/play", func(w http.ResponseWriter, r *http.Request) { http.HandleFunc("/webrtc/play", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Access-Control-Allow-Credentials", "true")
origin := r.Header["Origin"]
if len(origin) == 0 {
w.Header().Set("Access-Control-Allow-Origin", "*")
} else {
w.Header().Set("Access-Control-Allow-Origin", origin[0])
}
w.Header().Set("Content-Type", "application/json")
streamPath := r.URL.Query().Get("streamPath") streamPath := r.URL.Query().Get("streamPath")
var offer SessionDescription var offer SessionDescription
var rtc WebRTC var rtc WebRTC