package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-05 15:16:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

add OnRequest, OnResponse to ServerConn.Read

Browse Source
This commit is contained in:
aler9
2020-12-13 13:41:15 +01:00
parent 2a1af5a409
commit a23ccb1e8b
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
serverconn.go
View File

@@ -59,7 +59,14 @@ func (sc *ServerConn) EnableReadTimeout(v bool) {
}
// ServerConnReadHandlers allows to set the handlers required by ServerConn.Read.
// all fields are optional.
type ServerConnReadHandlers struct {
// called after receiving any request.
OnRequest func(req *base.Request)
// called before sending any response.
OnResponse func(res *base.Response)
// called after receiving a OPTIONS request.
// if nil, it is generated automatically.
OnOptions func(req *base.Request) (*base.Response, error)
@@ -96,6 +103,10 @@ type ServerConnReadHandlers struct {
func (sc *ServerConn) backgroundRead(handlers ServerConnReadHandlers, done chan error) {
handleRequest := func(req *base.Request) (*base.Response, error) {
if handlers.OnRequest != nil {
handlers.OnRequest(req)
}
switch req.Method {
case base.Options:
if handlers.OnOptions != nil {
@@ -250,6 +261,10 @@ func (sc *ServerConn) backgroundRead(handlers ServerConnReadHandlers, done chan
// add server
res.Header["Server"] = base.HeaderValue{"gortsplib"}
if handlers.OnResponse != nil {
handlers.OnResponse(res)
}
sc.nconn.SetWriteDeadline(time.Now().Add(sc.s.conf.WriteTimeout))
res.Write(sc.bw)