- Refactor the Go proxy for dependency injection: every proxy server, the bootstrap, the signal handler, the load balancers, and AMF0 now accept functional-option seams (factories/closures) so tests can inject fakes without binding real sockets, talking to real Redis, or racing on package globals. - Drop the package-global `lb.SrsLoadBalancer`. The bootstrap creates the LB locally and threads it through every proxy server constructor. Two old global indirections in `internal/signal` and `internal/rtmp/amf0` are likewise replaced by per-instance fields. - Rename `internal/server` → `internal/proxy` and rename the `lb` public surface for clarity: `SRSLoadBalancer` is split into `OriginService` / `HLSService` / `RTCService` and recomposed as `OriginLoadBalancer`; `SRSServer` → `OriginServer`; all proxy server types gain a `Proxy` qualifier (e.g. `RTMPServer` → `RTMPProxyServer`). - Extract the Redis client behind a new `internal/redisclient` package with a minimal `RedisClient` interface and a counterfeiter fake. - Add counterfeiter fakes (`proxyfakes`, `lbfakes`, `redisclientfakes`) and ~7.5k lines of unit tests covering bootstrap, memory + Redis LBs, all five proxy servers, the signal handler, and AMF0. - Add two new E2E flows — `proxy-e2e-srt-test.sh` (SRT publish through proxy, verify SRT/RTMP/HTTP-FLV/HLS playback) and `proxy-e2e-whip-test.sh` (WHIP publish, verify RTMP/HTTP-FLV/HLS via origin `rtc_to_rtmp`) — plus `setup-ffmpeg-with-whip.sh`, a macOS builder for an ffmpeg with openssl-DTLS WHIP and SRT support that the two scripts auto-invoke when needed. - Workspace reorg: move `memory/` and `skills/` to the repo root so all agent tools (Claude / Codex / Kiro / OpenClaw) share one source of truth via symlinks. Sync `docs/proxy/proxy-load-balancer.md` and `memory/srs-codebase-map.md` with the new names. No protocol, log, HTTP API, or wire-format changes. Refactor only — all externally observable proxy behavior is unchanged. --------- Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com> Co-authored-by: chatgpt-codex-connector[bot] <199175422+chatgpt-codex-connector[bot]@users.noreply.github.com>
198 lines
6.1 KiB
Go
198 lines
6.1 KiB
Go
// Code generated by counterfeiter. DO NOT EDIT.
|
|
package lbfakes
|
|
|
|
import (
|
|
"context"
|
|
"srsx/internal/lb"
|
|
"sync"
|
|
)
|
|
|
|
type FakeHLSService struct {
|
|
LoadHLSBySPBHIDStub func(context.Context, string) (lb.HLSPlayStream, error)
|
|
loadHLSBySPBHIDMutex sync.RWMutex
|
|
loadHLSBySPBHIDArgsForCall []struct {
|
|
arg1 context.Context
|
|
arg2 string
|
|
}
|
|
loadHLSBySPBHIDReturns struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}
|
|
loadHLSBySPBHIDReturnsOnCall map[int]struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}
|
|
LoadOrStoreHLSStub func(context.Context, string, lb.HLSPlayStream) (lb.HLSPlayStream, error)
|
|
loadOrStoreHLSMutex sync.RWMutex
|
|
loadOrStoreHLSArgsForCall []struct {
|
|
arg1 context.Context
|
|
arg2 string
|
|
arg3 lb.HLSPlayStream
|
|
}
|
|
loadOrStoreHLSReturns struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}
|
|
loadOrStoreHLSReturnsOnCall map[int]struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}
|
|
invocations map[string][][]interface{}
|
|
invocationsMutex sync.RWMutex
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadHLSBySPBHID(arg1 context.Context, arg2 string) (lb.HLSPlayStream, error) {
|
|
fake.loadHLSBySPBHIDMutex.Lock()
|
|
ret, specificReturn := fake.loadHLSBySPBHIDReturnsOnCall[len(fake.loadHLSBySPBHIDArgsForCall)]
|
|
fake.loadHLSBySPBHIDArgsForCall = append(fake.loadHLSBySPBHIDArgsForCall, struct {
|
|
arg1 context.Context
|
|
arg2 string
|
|
}{arg1, arg2})
|
|
stub := fake.LoadHLSBySPBHIDStub
|
|
fakeReturns := fake.loadHLSBySPBHIDReturns
|
|
fake.recordInvocation("LoadHLSBySPBHID", []interface{}{arg1, arg2})
|
|
fake.loadHLSBySPBHIDMutex.Unlock()
|
|
if stub != nil {
|
|
return stub(arg1, arg2)
|
|
}
|
|
if specificReturn {
|
|
return ret.result1, ret.result2
|
|
}
|
|
return fakeReturns.result1, fakeReturns.result2
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadHLSBySPBHIDCallCount() int {
|
|
fake.loadHLSBySPBHIDMutex.RLock()
|
|
defer fake.loadHLSBySPBHIDMutex.RUnlock()
|
|
return len(fake.loadHLSBySPBHIDArgsForCall)
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadHLSBySPBHIDCalls(stub func(context.Context, string) (lb.HLSPlayStream, error)) {
|
|
fake.loadHLSBySPBHIDMutex.Lock()
|
|
defer fake.loadHLSBySPBHIDMutex.Unlock()
|
|
fake.LoadHLSBySPBHIDStub = stub
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadHLSBySPBHIDArgsForCall(i int) (context.Context, string) {
|
|
fake.loadHLSBySPBHIDMutex.RLock()
|
|
defer fake.loadHLSBySPBHIDMutex.RUnlock()
|
|
argsForCall := fake.loadHLSBySPBHIDArgsForCall[i]
|
|
return argsForCall.arg1, argsForCall.arg2
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadHLSBySPBHIDReturns(result1 lb.HLSPlayStream, result2 error) {
|
|
fake.loadHLSBySPBHIDMutex.Lock()
|
|
defer fake.loadHLSBySPBHIDMutex.Unlock()
|
|
fake.LoadHLSBySPBHIDStub = nil
|
|
fake.loadHLSBySPBHIDReturns = struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}{result1, result2}
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadHLSBySPBHIDReturnsOnCall(i int, result1 lb.HLSPlayStream, result2 error) {
|
|
fake.loadHLSBySPBHIDMutex.Lock()
|
|
defer fake.loadHLSBySPBHIDMutex.Unlock()
|
|
fake.LoadHLSBySPBHIDStub = nil
|
|
if fake.loadHLSBySPBHIDReturnsOnCall == nil {
|
|
fake.loadHLSBySPBHIDReturnsOnCall = make(map[int]struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
})
|
|
}
|
|
fake.loadHLSBySPBHIDReturnsOnCall[i] = struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}{result1, result2}
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadOrStoreHLS(arg1 context.Context, arg2 string, arg3 lb.HLSPlayStream) (lb.HLSPlayStream, error) {
|
|
fake.loadOrStoreHLSMutex.Lock()
|
|
ret, specificReturn := fake.loadOrStoreHLSReturnsOnCall[len(fake.loadOrStoreHLSArgsForCall)]
|
|
fake.loadOrStoreHLSArgsForCall = append(fake.loadOrStoreHLSArgsForCall, struct {
|
|
arg1 context.Context
|
|
arg2 string
|
|
arg3 lb.HLSPlayStream
|
|
}{arg1, arg2, arg3})
|
|
stub := fake.LoadOrStoreHLSStub
|
|
fakeReturns := fake.loadOrStoreHLSReturns
|
|
fake.recordInvocation("LoadOrStoreHLS", []interface{}{arg1, arg2, arg3})
|
|
fake.loadOrStoreHLSMutex.Unlock()
|
|
if stub != nil {
|
|
return stub(arg1, arg2, arg3)
|
|
}
|
|
if specificReturn {
|
|
return ret.result1, ret.result2
|
|
}
|
|
return fakeReturns.result1, fakeReturns.result2
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadOrStoreHLSCallCount() int {
|
|
fake.loadOrStoreHLSMutex.RLock()
|
|
defer fake.loadOrStoreHLSMutex.RUnlock()
|
|
return len(fake.loadOrStoreHLSArgsForCall)
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadOrStoreHLSCalls(stub func(context.Context, string, lb.HLSPlayStream) (lb.HLSPlayStream, error)) {
|
|
fake.loadOrStoreHLSMutex.Lock()
|
|
defer fake.loadOrStoreHLSMutex.Unlock()
|
|
fake.LoadOrStoreHLSStub = stub
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadOrStoreHLSArgsForCall(i int) (context.Context, string, lb.HLSPlayStream) {
|
|
fake.loadOrStoreHLSMutex.RLock()
|
|
defer fake.loadOrStoreHLSMutex.RUnlock()
|
|
argsForCall := fake.loadOrStoreHLSArgsForCall[i]
|
|
return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadOrStoreHLSReturns(result1 lb.HLSPlayStream, result2 error) {
|
|
fake.loadOrStoreHLSMutex.Lock()
|
|
defer fake.loadOrStoreHLSMutex.Unlock()
|
|
fake.LoadOrStoreHLSStub = nil
|
|
fake.loadOrStoreHLSReturns = struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}{result1, result2}
|
|
}
|
|
|
|
func (fake *FakeHLSService) LoadOrStoreHLSReturnsOnCall(i int, result1 lb.HLSPlayStream, result2 error) {
|
|
fake.loadOrStoreHLSMutex.Lock()
|
|
defer fake.loadOrStoreHLSMutex.Unlock()
|
|
fake.LoadOrStoreHLSStub = nil
|
|
if fake.loadOrStoreHLSReturnsOnCall == nil {
|
|
fake.loadOrStoreHLSReturnsOnCall = make(map[int]struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
})
|
|
}
|
|
fake.loadOrStoreHLSReturnsOnCall[i] = struct {
|
|
result1 lb.HLSPlayStream
|
|
result2 error
|
|
}{result1, result2}
|
|
}
|
|
|
|
func (fake *FakeHLSService) Invocations() map[string][][]interface{} {
|
|
fake.invocationsMutex.RLock()
|
|
defer fake.invocationsMutex.RUnlock()
|
|
copiedInvocations := map[string][][]interface{}{}
|
|
for key, value := range fake.invocations {
|
|
copiedInvocations[key] = value
|
|
}
|
|
return copiedInvocations
|
|
}
|
|
|
|
func (fake *FakeHLSService) recordInvocation(key string, args []interface{}) {
|
|
fake.invocationsMutex.Lock()
|
|
defer fake.invocationsMutex.Unlock()
|
|
if fake.invocations == nil {
|
|
fake.invocations = map[string][][]interface{}{}
|
|
}
|
|
if fake.invocations[key] == nil {
|
|
fake.invocations[key] = [][]interface{}{}
|
|
}
|
|
fake.invocations[key] = append(fake.invocations[key], args)
|
|
}
|
|
|
|
var _ lb.HLSService = new(FakeHLSService)
|