Rewrite stream info API

pkg/mjpeg/client.go

@@ -2,6 +2,7 @@ package mjpeg
 
 import (
 	"bufio"
+	"encoding/json"
 	"errors"
 	"github.com/AlexxIT/go2rtc/pkg/streamer"
 	"github.com/AlexxIT/go2rtc/pkg/tcp"
@@ -11,6 +12,7 @@ import (
 	"net/textproto"
 	"strconv"
 	"strings"
+	"sync/atomic"
 	"time"
 )
 
@@ -24,6 +26,7 @@ type Client struct {
 	res    *http.Response
 
 	track *streamer.Track
+	recv  uint32
 }
 
 func NewClient(res *http.Response) *Client {
@@ -70,6 +73,17 @@ func (c *Client) Stop() error {
 	return nil
 }
 
+func (c *Client) MarshalJSON() ([]byte, error) {
+	info := &streamer.Info{
+		Type:       "MJPEG source",
+		URL:        c.res.Request.URL.String(),
+		RemoteAddr: c.RemoteAddr,
+		UserAgent:  c.UserAgent,
+		Recv:       atomic.LoadUint32(&c.recv),
+	}
+	return json.Marshal(info)
+}
+
 func (c *Client) startJPEG() error {
 	buf, err := io.ReadAll(c.res.Body)
 	if err != nil {
@@ -79,6 +93,8 @@ func (c *Client) startJPEG() error {
 	packet := &rtp.Packet{Header: rtp.Header{Timestamp: now()}, Payload: buf}
 	_ = c.track.WriteRTP(packet)
 
+	atomic.AddUint32(&c.recv, uint32(len(buf)))
+
 	req := c.res.Request
 
 	for !c.closed {
@@ -98,6 +114,8 @@ func (c *Client) startJPEG() error {
 
 		packet = &rtp.Packet{Header: rtp.Header{Timestamp: now()}, Payload: buf}
 		_ = c.track.WriteRTP(packet)
+
+		atomic.AddUint32(&c.recv, uint32(len(buf)))
 	}
 
 	return nil
@@ -141,6 +159,8 @@ func (c *Client) startMJPEG(boundary string) error {
 		packet := &rtp.Packet{Header: rtp.Header{Timestamp: now()}, Payload: buf}
 		_ = c.track.WriteRTP(packet)
 
+		atomic.AddUint32(&c.recv, uint32(len(buf)))
+
 		if _, err = r.Discard(2); err != nil {
 			return err
 		}

pkg/mjpeg/consumer.go

@@ -1,8 +1,10 @@
 package mjpeg
 
 import (
+	"encoding/json"
 	"github.com/AlexxIT/go2rtc/pkg/streamer"
 	"github.com/pion/rtp"
+	"sync/atomic"
 )
 
 type Consumer struct {
@@ -14,7 +16,7 @@ type Consumer struct {
 	codecs []*streamer.Codec
 	start  bool
 
-	send int
+	send uint32
 }
 
 func (c *Consumer) GetMedias() []*streamer.Media {
@@ -28,6 +30,7 @@ func (c *Consumer) GetMedias() []*streamer.Media {
 func (c *Consumer) AddTrack(media *streamer.Media, track *streamer.Track) *streamer.Track {
 	push := func(packet *rtp.Packet) error {
 		c.Fire(packet.Payload)
+		atomic.AddUint32(&c.send, uint32(len(packet.Payload)))
 		return nil
 	}
 
@@ -38,3 +41,13 @@ func (c *Consumer) AddTrack(media *streamer.Media, track *streamer.Track) *strea
 
 	return track.Bind(push)
 }
+
+func (c *Consumer) MarshalJSON() ([]byte, error) {
+	info := &streamer.Info{
+		Type:       "MJPEG client",
+		RemoteAddr: c.RemoteAddr,
+		UserAgent:  c.UserAgent,
+		Send:       atomic.LoadUint32(&c.send),
+	}
+	return json.Marshal(info)
+}