mirror of
https://github.com/bluenviron/mediamtx
synced 2024-12-15 03:04:45 +00:00
74 lines
1.3 KiB
Go
74 lines
1.3 KiB
Go
package conf
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/bluenviron/gortsplib/v4"
|
|
)
|
|
|
|
// SourceProtocol is the sourceProtocol parameter.
|
|
type SourceProtocol struct {
|
|
*gortsplib.Transport
|
|
}
|
|
|
|
// MarshalJSON implements json.Marshaler.
|
|
func (d SourceProtocol) MarshalJSON() ([]byte, error) {
|
|
var out string
|
|
|
|
if d.Transport == nil {
|
|
out = "automatic"
|
|
} else {
|
|
switch *d.Transport {
|
|
case gortsplib.TransportUDP:
|
|
out = "udp"
|
|
|
|
case gortsplib.TransportUDPMulticast:
|
|
out = "multicast"
|
|
|
|
case gortsplib.TransportTCP:
|
|
out = "tcp"
|
|
|
|
default:
|
|
return nil, fmt.Errorf("invalid protocol: %v", d.Transport)
|
|
}
|
|
}
|
|
|
|
return json.Marshal(out)
|
|
}
|
|
|
|
// UnmarshalJSON implements json.Unmarshaler.
|
|
func (d *SourceProtocol) UnmarshalJSON(b []byte) error {
|
|
var in string
|
|
if err := json.Unmarshal(b, &in); err != nil {
|
|
return err
|
|
}
|
|
|
|
switch in {
|
|
case "udp":
|
|
v := gortsplib.TransportUDP
|
|
d.Transport = &v
|
|
|
|
case "multicast":
|
|
v := gortsplib.TransportUDPMulticast
|
|
d.Transport = &v
|
|
|
|
case "tcp":
|
|
v := gortsplib.TransportTCP
|
|
d.Transport = &v
|
|
|
|
case "automatic":
|
|
d.Transport = nil
|
|
|
|
default:
|
|
return fmt.Errorf("invalid protocol '%s'", in)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UnmarshalEnv implements env.Unmarshaler.
|
|
func (d *SourceProtocol) UnmarshalEnv(_ string, v string) error {
|
|
return d.UnmarshalJSON([]byte(`"` + v + `"`))
|
|
}
|