package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-11-03 03:03:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

accept io.Writer in Write() methods

Browse Source
This commit is contained in:
aler9
2022-05-08 14:33:29 +02:00
parent bdbce3c370
commit edeef85e9e
13 changed files with 183 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/base/response.go
View File

@@ -4,6 +4,7 @@ import (
"bufio"
"bytes"
"fmt"
"io"
"strconv"
)
@@ -203,16 +204,14 @@ func (res *Response) ReadIgnoreFrames(maxPayloadSize int, rb *bufio.Reader) erro
}
// Write writes a Response.
func (res Response) Write(bb *bytes.Buffer) {
bb.Reset()
func (res Response) Write(w io.Writer) {
if res.StatusMessage == "" {
if status, ok := statusMessages[res.StatusCode]; ok {
res.StatusMessage = status
}
}
bb.Write([]byte(rtspProtocol10 + " " +
w.Write([]byte(rtspProtocol10 + " " +
strconv.FormatInt(int64(res.StatusCode), 10) + " " +
res.StatusMessage + "\r\n"))
@@ -220,9 +219,9 @@ func (res Response) Write(bb *bytes.Buffer) {
res.Header["Content-Length"] = HeaderValue{strconv.FormatInt(int64(len(res.Body)), 10)}
}
res.Header.write(bb)
res.Header.write(w)
body(res.Body).write(bb)
body(res.Body).write(w)
}
// String implements fmt.Stringer.