chain.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. package interceptor
  2. // Chain is an interceptor that runs all child interceptors in order.
  3. type Chain struct {
  4. interceptors []Interceptor
  5. }
  6. // NewChain returns a new Chain interceptor.
  7. func NewChain(interceptors []Interceptor) *Chain {
  8. return &Chain{interceptors: interceptors}
  9. }
  10. // BindRTCPReader lets you modify any incoming RTCP packets. It is called once per sender/receiver, however this might
  11. // change in the future. The returned method will be called once per packet batch.
  12. func (i *Chain) BindRTCPReader(reader RTCPReader) RTCPReader {
  13. for _, interceptor := range i.interceptors {
  14. reader = interceptor.BindRTCPReader(reader)
  15. }
  16. return reader
  17. }
  18. // BindRTCPWriter lets you modify any outgoing RTCP packets. It is called once per PeerConnection. The returned method
  19. // will be called once per packet batch.
  20. func (i *Chain) BindRTCPWriter(writer RTCPWriter) RTCPWriter {
  21. for _, interceptor := range i.interceptors {
  22. writer = interceptor.BindRTCPWriter(writer)
  23. }
  24. return writer
  25. }
  26. // BindLocalStream lets you modify any outgoing RTP packets. It is called once for per LocalStream. The returned method
  27. // will be called once per rtp packet.
  28. func (i *Chain) BindLocalStream(ctx *StreamInfo, writer RTPWriter) RTPWriter {
  29. for _, interceptor := range i.interceptors {
  30. writer = interceptor.BindLocalStream(ctx, writer)
  31. }
  32. return writer
  33. }
  34. // UnbindLocalStream is called when the Stream is removed. It can be used to clean up any data related to that track.
  35. func (i *Chain) UnbindLocalStream(ctx *StreamInfo) {
  36. for _, interceptor := range i.interceptors {
  37. interceptor.UnbindLocalStream(ctx)
  38. }
  39. }
  40. // BindRemoteStream lets you modify any incoming RTP packets. It is called once for per RemoteStream. The returned method
  41. // will be called once per rtp packet.
  42. func (i *Chain) BindRemoteStream(ctx *StreamInfo, reader RTPReader) RTPReader {
  43. for _, interceptor := range i.interceptors {
  44. reader = interceptor.BindRemoteStream(ctx, reader)
  45. }
  46. return reader
  47. }
  48. // UnbindRemoteStream is called when the Stream is removed. It can be used to clean up any data related to that track.
  49. func (i *Chain) UnbindRemoteStream(ctx *StreamInfo) {
  50. for _, interceptor := range i.interceptors {
  51. interceptor.UnbindRemoteStream(ctx)
  52. }
  53. }
  54. // Close closes the Interceptor, cleaning up any data if necessary.
  55. func (i *Chain) Close() error {
  56. var errs []error
  57. for _, interceptor := range i.interceptors {
  58. errs = append(errs, interceptor.Close())
  59. }
  60. return flattenErrs(errs)
  61. }