server: add ServerConn to OnResponse

This commit is contained in:
aler9
2021-05-05 18:54:37 +02:00
parent 178fe919ea
commit 353382175b
2 changed files with 4 additions and 4 deletions

View File

@@ -476,7 +476,7 @@ func (sc *ServerConn) handleRequest(req *base.Request) (*base.Response, error) {
func (sc *ServerConn) handleRequestOuter(req *base.Request) error { func (sc *ServerConn) handleRequestOuter(req *base.Request) error {
if h, ok := sc.s.Handler.(ServerHandlerOnRequest); ok { if h, ok := sc.s.Handler.(ServerHandlerOnRequest); ok {
h.OnRequest(req) h.OnRequest(sc, req)
} }
res, err := sc.handleRequest(req) res, err := sc.handleRequest(req)
@@ -494,7 +494,7 @@ func (sc *ServerConn) handleRequestOuter(req *base.Request) error {
res.Header["Server"] = base.HeaderValue{"gortsplib"} res.Header["Server"] = base.HeaderValue{"gortsplib"}
if h, ok := sc.s.Handler.(ServerHandlerOnResponse); ok { if h, ok := sc.s.Handler.(ServerHandlerOnResponse); ok {
h.OnResponse(res) h.OnResponse(sc, res)
} }
switch { switch {

View File

@@ -31,12 +31,12 @@ type ServerHandlerOnSessionClose interface {
// ServerHandlerOnRequest can be implemented by a ServerHandler. // ServerHandlerOnRequest can be implemented by a ServerHandler.
type ServerHandlerOnRequest interface { type ServerHandlerOnRequest interface {
OnRequest(*base.Request) OnRequest(*ServerConn, *base.Request)
} }
// ServerHandlerOnResponse can be implemented by a ServerHandler. // ServerHandlerOnResponse can be implemented by a ServerHandler.
type ServerHandlerOnResponse interface { type ServerHandlerOnResponse interface {
OnResponse(*base.Response) OnResponse(*ServerConn, *base.Response)
} }
// ServerHandlerOnDescribeCtx is the context of a DESCRIBE request. // ServerHandlerOnDescribeCtx is the context of a DESCRIBE request.