Code refactoring

This commit is contained in:
Alexey Khit
2023-02-24 12:28:03 +03:00
parent f5892e4cfc
commit ad3c5440fe
+17 -15
View File
@@ -90,9 +90,6 @@ func asyncHandler(tr *api.Transport, msg *api.Message) error {
return err return err
} }
// apiV2 - json/object exchange, V2 - raw SDP and raw Candidates exchange
apiV2 := msg.Type == "webrtc"
cons := webrtc.NewServer(pc) cons := webrtc.NewServer(pc)
cons.UserAgent = tr.Request.UserAgent() cons.UserAgent = tr.Request.UserAgent()
cons.Listen(func(msg any) { cons.Listen(func(msg any) {
@@ -110,6 +107,9 @@ func asyncHandler(tr *api.Transport, msg *api.Message) error {
} }
}) })
// V2 - json/object exchange, V1 - raw SDP exchange
apiV2 := msg.Type == "webrtc"
// 1. SetOffer, so we can get remote client codecs // 1. SetOffer, so we can get remote client codecs
var offer string var offer string
if apiV2 { if apiV2 {
@@ -141,14 +141,18 @@ func asyncHandler(tr *api.Transport, msg *api.Message) error {
return err return err
} }
tr.Write(&api.Message{Type: "webrtc/answer", Value: answer}) if apiV2 {
desc := pion.SessionDescription{Type: pion.SDPTypeAnswer, SDP: answer}
tr.Write(&api.Message{Type: "webrtc", Value: desc})
} else {
tr.Write(&api.Message{Type: "webrtc/answer", Value: answer})
}
asyncCandidates(tr, cons) asyncCandidates(tr, cons)
return nil return nil
} }
// syncHandler
func syncHandler(w http.ResponseWriter, r *http.Request) { func syncHandler(w http.ResponseWriter, r *http.Request) {
url := r.URL.Query().Get("src") url := r.URL.Query().Get("src")
stream := streams.Get(url) stream := streams.Get(url)
@@ -161,18 +165,18 @@ func syncHandler(w http.ResponseWriter, r *http.Request) {
ct, _, _ = mime.ParseMediaType(ct) ct, _, _ = mime.ParseMediaType(ct)
} }
// apiV2 - json/object exchange, V1 - raw SDP exchange // V2 - json/object exchange, V1 - raw SDP exchange
apiV2 := ct == "application/json" apiV2 := ct == "application/json"
var offer string var offer string
if apiV2 { if apiV2 {
var sd pion.SessionDescription var desc pion.SessionDescription
if err := json.NewDecoder(r.Body).Decode(&sd); err != nil { if err := json.NewDecoder(r.Body).Decode(&desc); err != nil {
log.Error().Err(err).Caller().Send() log.Error().Err(err).Caller().Send()
http.Error(w, err.Error(), http.StatusBadRequest) http.Error(w, err.Error(), http.StatusBadRequest)
return return
} }
offer = sd.SDP offer = desc.SDP
} else { } else {
body, err := io.ReadAll(r.Body) body, err := io.ReadAll(r.Body)
if err != nil { if err != nil {
@@ -211,7 +215,7 @@ func syncHandler(w http.ResponseWriter, r *http.Request) {
func ExchangeSDP(stream *streams.Stream, offer string, userAgent string) (answer string, err error) { func ExchangeSDP(stream *streams.Stream, offer string, userAgent string) (answer string, err error) {
pc, err := newPeerConnection() pc, err := newPeerConnection()
if err != nil { if err != nil {
log.Error().Err(err).Caller().Msg("NewPConn") log.Error().Err(err).Caller().Send()
return return
} }
@@ -231,19 +235,17 @@ func ExchangeSDP(stream *streams.Stream, offer string, userAgent string) (answer
log.Trace().Msgf("[webrtc] offer:\n%s", offer) log.Trace().Msgf("[webrtc] offer:\n%s", offer)
if err = conn.SetOffer(offer); err != nil { if err = conn.SetOffer(offer); err != nil {
log.Warn().Err(err).Caller().Msg("conn.SetOffer") log.Warn().Err(err).Caller().Send()
return return
} }
// 2. AddConsumer, so we get new tracks // 2. AddConsumer, so we get new tracks
if err = stream.AddConsumer(conn); err != nil { if err = stream.AddConsumer(conn); err != nil {
log.Warn().Err(err).Caller().Msg("stream.AddConsumer") log.Warn().Err(err).Caller().Send()
_ = conn.Close() _ = conn.Close()
return return
} }
// exchange sdp without waiting all candidates
//answer, err := conn.ExchangeSDP(offer, false)
answer, err = conn.GetCompleteAnswer() answer, err = conn.GetCompleteAnswer()
if err == nil { if err == nil {
answer, err = syncCanditates(answer) answer, err = syncCanditates(answer)
@@ -251,7 +253,7 @@ func ExchangeSDP(stream *streams.Stream, offer string, userAgent string) (answer
log.Trace().Msgf("[webrtc] answer\n%s", answer) log.Trace().Msgf("[webrtc] answer\n%s", answer)
if err != nil { if err != nil {
log.Error().Err(err).Caller().Msg("conn.GetCompleteAnswer") log.Error().Err(err).Caller().Send()
} }
return return