mirror of
https://github.com/hanzoai/livekit.git
synced 2026-10-30 18:29:34 +00:00
zeekay
e9292da479
This fork lived at github.com/hanzoai/livekit but still declared
`module github.com/livekit/livekit-server`, so all 505 self-references pointed
at upstream. A fork that does not own its module path is the exact thing the
owned-module-paths rule exists to prevent: every import line in every consumer
carries someone else's brand, and `go get` resolves to their code, not ours.
module + 505 self-imports github.com/livekit/livekit-server -> github.com/hanzoai/livekit
KV client github.com/redis/go-redis/v9 -> github.com/hanzokv/go/v9
message bus github.com/livekit/psrpc -> github.com/hanzoai/psrpc
psrpc had to be forked (hanzoai/psrpc, Apache-2.0) because its bus is typed
against the KV client — the same shape redsync had for hanzoai/git.
DELIBERATELY NOT FORKED: github.com/livekit/{protocol,mediatransportutil,
mageutil} keep their addresses. Only `protocol` still reaches go-redis, through
a single file (redis/redis.go) out of 305. Forking an 11.5MB core wire-protocol
library that this server must track in lockstep — and that upstream pushed
yesterday — to change one file trades a tidy dependency graph for permanent
divergence on the fastest-moving dependency here. So one indirect
github.com/redis/go-redis entry remains, via livekit/protocol/redis, and that
is the correct place for it to live.
110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
// Copyright 2023 LiveKit, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package rtc
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/hanzoai/livekit/pkg/rtc/datatrack"
|
|
"github.com/hanzoai/livekit/pkg/rtc/types"
|
|
"github.com/livekit/protocol/livekit"
|
|
"github.com/livekit/protocol/logger"
|
|
)
|
|
|
|
var _ types.DataDownTrack = (*DataDownTrack)(nil)
|
|
var _ types.DataTrackSender = (*DataDownTrack)(nil)
|
|
|
|
type DataDownTrackParams struct {
|
|
Logger logger.Logger
|
|
SubscriberID livekit.ParticipantID
|
|
PublishDataTrack types.DataTrack
|
|
Handle uint16
|
|
Transport types.DataTrackTransport
|
|
BytesTrackStats *BytesTrackStats
|
|
}
|
|
|
|
type DataDownTrack struct {
|
|
params DataDownTrackParams
|
|
logger logger.Logger
|
|
createdAt int64
|
|
}
|
|
|
|
func NewDataDownTrack(params DataDownTrackParams) (*DataDownTrack, error) {
|
|
d := &DataDownTrack{
|
|
params: params,
|
|
createdAt: time.Now().UnixNano(),
|
|
}
|
|
d.logger = params.Logger.WithValues("name", d.Name(), "handle", d.Handle())
|
|
|
|
if err := d.params.PublishDataTrack.AddDataDownTrack(d); err != nil {
|
|
d.logger.Warnw("could not add data down track", err)
|
|
return nil, err
|
|
}
|
|
|
|
d.logger.Infow("created data down track")
|
|
return d, nil
|
|
}
|
|
|
|
func (d *DataDownTrack) Close() {
|
|
d.logger.Infow("closing data down track")
|
|
if d.params.BytesTrackStats != nil {
|
|
d.params.BytesTrackStats.Stop()
|
|
}
|
|
d.params.PublishDataTrack.DeleteDataDownTrack(d.SubscriberID())
|
|
}
|
|
|
|
func (d *DataDownTrack) Handle() uint16 {
|
|
return d.params.Handle
|
|
}
|
|
|
|
func (d *DataDownTrack) PublishDataTrack() types.DataTrack {
|
|
return d.params.PublishDataTrack
|
|
}
|
|
|
|
func (d *DataDownTrack) ID() livekit.TrackID {
|
|
return d.params.PublishDataTrack.ID()
|
|
}
|
|
|
|
func (d *DataDownTrack) Name() string {
|
|
return d.params.PublishDataTrack.Name()
|
|
}
|
|
|
|
func (d *DataDownTrack) SubscriberID() livekit.ParticipantID {
|
|
// add `createdAt` to ensure repeated subscriptions from same subscriber to same publisher does not collide
|
|
return livekit.ParticipantID(fmt.Sprintf("%s:%d", d.params.SubscriberID, d.createdAt))
|
|
}
|
|
|
|
func (d *DataDownTrack) WritePacket(data []byte, packet *datatrack.Packet, _arrivalTime int64) {
|
|
forwardedPacket := *packet
|
|
forwardedPacket.Handle = d.params.Handle
|
|
buf, err := forwardedPacket.Marshal()
|
|
if err != nil {
|
|
d.logger.Warnw("could not marshal data track message", err)
|
|
return
|
|
}
|
|
if err := d.params.Transport.SendDataTrackMessage(buf); err != nil {
|
|
d.logger.Warnw("could not send data track message", err)
|
|
return
|
|
}
|
|
if d.params.BytesTrackStats != nil {
|
|
d.params.BytesTrackStats.AddBytes(uint64(len(buf)), true)
|
|
}
|
|
}
|
|
|
|
func (d *DataDownTrack) UpdateSubscriptionOptions(subscriptionOptions *livekit.DataTrackSubscriptionOptions) {
|
|
// DT-TODO
|
|
}
|