engine/common/index.go

package common

import (
	"sync/atomic"
	"time"

	"github.com/pion/rtp"
	"go.uber.org/zap"
	"m7s.live/engine/v4/log"
	"m7s.live/engine/v4/util"
)

type TimelineData[T any] struct {
	Timestamp time.Time
	Value     T
}
type TrackState byte

const (
	TrackStateOnline  TrackState = iota // 上线
	TrackStateOffline                   // 下线
)

// Base 基础Track类
type Base struct {
	Name     string
	log.Zap  `json:"-"`
	Stream   IStream     `json:"-"`
	Attached atomic.Bool `json:"-"`
	State    TrackState
	ts       time.Time
	bytes    int
	frames   int
	BPS      int
	FPS      int
	RawSize  int                 // 裸数据长度
	RawPart  []int               // 裸数据片段用于UI上显示
	BPSs     []TimelineData[int] // 10s码率统计
	FPSs     []TimelineData[int] // 10s帧率统计
}

func (bt *Base) ComputeBPS(bytes int) {
	bt.bytes += bytes
	bt.frames++
	if elapse := time.Since(bt.ts).Seconds(); elapse > 1 {
		bt.BPS = bt.bytes / int(elapse)
		bt.FPS = bt.frames / int(elapse)
		bt.bytes = 0
		bt.frames = 0
		bt.ts = time.Now()
		bt.BPSs = append(bt.BPSs, TimelineData[int]{Timestamp: bt.ts, Value: bt.BPS})
		if len(bt.BPSs) > 9 {
			copy(bt.BPSs, bt.BPSs[1:])
			bt.BPSs = bt.BPSs[:10]
		}
		bt.FPSs = append(bt.FPSs, TimelineData[int]{Timestamp: bt.ts, Value: bt.FPS})
		if len(bt.FPSs) > 10 {
			copy(bt.FPSs, bt.FPSs[1:])
			bt.FPSs = bt.FPSs[:10]
		}
	}
}

func (bt *Base) GetBase() *Base {
	return bt
}
func (bt *Base) SnapForJson() {
}
func (bt *Base) Flush(bf *BaseFrame) {
	bt.ComputeBPS(bf.BytesIn)
	bf.Timestamp = time.Now()
}
func (bt *Base) SetStuff(stuff ...any) {
	for _, s := range stuff {
		switch v := s.(type) {
		case IStream:
			bt.Stream = v
			bt.Zap = v.With(zap.String("track", bt.Name))
		case TrackState:
			bt.State = v
		case string:
			bt.Name = v
		}
	}
}

type Track interface {
	GetBase() *Base
	LastWriteTime() time.Time
	SnapForJson()
	SetStuff(stuff ...any)
}

type AVTrack interface {
	Track
	PreFrame() *AVFrame
	CurrentFrame() *AVFrame
	Attach()
	Detach()
	WriteAVCC(ts uint32, frame *util.BLL) error //写入AVCC格式的数据
	WriteRTP(*util.ListItem[RTPFrame])
	WriteRTPPack(*rtp.Packet)
	Flush()
	SetSpeedLimit(time.Duration)
	GetRTPFromPool() *util.ListItem[RTPFrame]
}
type VideoTrack interface {
	AVTrack
	WriteSliceBytes(slice []byte)
	WriteAnnexB(uint32, uint32, []byte)
	SetLostFlag()
}

type AudioTrack interface {
	AVTrack
	WriteADTS(uint32, []byte)
	WriteRaw(uint32, []byte)
}