mirror of
https://github.com/livekit/livekit.git
synced 2026-04-05 08:05:41 +00:00
54 lines
1.7 KiB
Go
54 lines
1.7 KiB
Go
package routing
|
|
|
|
import (
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
"github.com/livekit/livekit-server/proto/livekit"
|
|
)
|
|
|
|
//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 -generate
|
|
|
|
// MessageSink is an abstraction for writing protobuf messages and having them read by a MessageSource,
|
|
// potentially on a different node via a transport
|
|
//counterfeiter:generate . MessageSink
|
|
type MessageSink interface {
|
|
WriteMessage(msg proto.Message) error
|
|
Close()
|
|
OnClose(f func())
|
|
}
|
|
|
|
//counterfeiter:generate . MessageSource
|
|
type MessageSource interface {
|
|
// source exposes a one way channel to make it easier to use with select
|
|
ReadChan() <-chan proto.Message
|
|
}
|
|
|
|
type ParticipantCallback func(roomName, identity string, reconnect bool, requestSource MessageSource, responseSink MessageSink)
|
|
|
|
// Router allows multiple nodes to coordinate the participant session
|
|
//counterfeiter:generate . Router
|
|
type Router interface {
|
|
GetNodeForRoom(roomName string) (string, error)
|
|
SetNodeForRoom(roomName string, nodeId string) error
|
|
ClearRoomState(roomName string) error
|
|
RegisterNode() error
|
|
UnregisterNode() error
|
|
RemoveDeadNodes() error
|
|
GetNode(nodeId string) (*livekit.Node, error)
|
|
ListNodes() ([]*livekit.Node, error)
|
|
|
|
// participant signal connection is ready to start
|
|
StartParticipantSignal(roomName, identity string, reconnect bool) (reqSink MessageSink, resSource MessageSource, err error)
|
|
|
|
// when a new participant's RTC connection is ready to start
|
|
OnNewParticipantRTC(callback ParticipantCallback)
|
|
Start() error
|
|
Stop()
|
|
}
|
|
|
|
// NodeSelector selects an appropriate node to run the current session
|
|
//counterfeiter:generate . NodeSelector
|
|
type NodeSelector interface {
|
|
SelectNode(nodes []*livekit.Node, room *livekit.Room) (*livekit.Node, error)
|
|
}
|