golanggohlsrtmpwebrtcmedia-serverobs-studiortcprtmp-proxyrtmp-serverrtprtsprtsp-proxyrtsp-relayrtsp-serversrtstreamingwebrtc-proxy
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.2 KiB
81 lines
2.2 KiB
package core |
|
|
|
import ( |
|
"io/ioutil" |
|
"net/http" |
|
"os" |
|
"strings" |
|
"testing" |
|
|
|
"github.com/aler9/gortsplib" |
|
"github.com/stretchr/testify/require" |
|
) |
|
|
|
func TestMetrics(t *testing.T) { |
|
serverCertFpath, err := writeTempFile(serverCert) |
|
require.NoError(t, err) |
|
defer os.Remove(serverCertFpath) |
|
|
|
serverKeyFpath, err := writeTempFile(serverKey) |
|
require.NoError(t, err) |
|
defer os.Remove(serverKeyFpath) |
|
|
|
p, ok := newInstance("metrics: yes\n" + |
|
"encryption: optional\n" + |
|
"serverCert: " + serverCertFpath + "\n" + |
|
"serverKey: " + serverKeyFpath + "\n") |
|
require.Equal(t, true, ok) |
|
defer p.close() |
|
|
|
track, err := gortsplib.NewTrackH264(96, |
|
&gortsplib.TrackConfigH264{SPS: []byte{0x01, 0x02, 0x03, 0x04}, PPS: []byte{0x01, 0x02, 0x03, 0x04}}) |
|
require.NoError(t, err) |
|
|
|
source, err := gortsplib.DialPublish("rtsp://localhost:8554/rtsp_path", |
|
gortsplib.Tracks{track}) |
|
require.NoError(t, err) |
|
defer source.Close() |
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{ |
|
"-re", |
|
"-stream_loop", "-1", |
|
"-i", "emptyvideo.mkv", |
|
"-c", "copy", |
|
"-f", "flv", |
|
"rtmp://localhost:1935/rtmp_path", |
|
}) |
|
require.NoError(t, err) |
|
defer cnt1.close() |
|
|
|
req, err := http.NewRequest(http.MethodGet, "http://localhost:9998/metrics", nil) |
|
require.NoError(t, err) |
|
|
|
res, err := http.DefaultClient.Do(req) |
|
require.NoError(t, err) |
|
defer res.Body.Close() |
|
require.Equal(t, http.StatusOK, res.StatusCode) |
|
|
|
bo, err := ioutil.ReadAll(res.Body) |
|
require.NoError(t, err) |
|
|
|
vals := make(map[string]string) |
|
lines := strings.Split(string(bo), "\n") |
|
for _, l := range lines[:len(lines)-1] { |
|
fields := strings.Split(l, " ") |
|
vals[fields[0]] = fields[1] |
|
} |
|
|
|
require.Equal(t, map[string]string{ |
|
"paths{name=\"rtsp_path\",state=\"ready\"}": "1", |
|
"paths{name=\"rtmp_path\",state=\"ready\"}": "1", |
|
"rtmp_conns{state=\"idle\"}": "0", |
|
"rtmp_conns{state=\"publish\"}": "1", |
|
"rtmp_conns{state=\"read\"}": "0", |
|
"rtsp_sessions{state=\"idle\"}": "0", |
|
"rtsp_sessions{state=\"publish\"}": "1", |
|
"rtsp_sessions{state=\"read\"}": "0", |
|
"rtsps_sessions{state=\"idle\"}": "0", |
|
"rtsps_sessions{state=\"publish\"}": "0", |
|
"rtsps_sessions{state=\"read\"}": "0", |
|
}, vals) |
|
}
|
|
|