apps/plugin-hdl
1
0
Fork 0
mirror of https://github.com/Monibuca/plugin-hdl.git synced 2025-10-04 16:32:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复传入WriteTimeout的单位

Browse Source
This commit is contained in:
dexter
2023-02-08 22:34:57 +08:00
parent 14fd31f7c0
commit a1425f7b97
5 changed files with 106 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.go
View File

@@ -79,7 +79,7 @@ func (sub *HDLSubscriber) OnEvent(event any) {
// s := util.SizeOfBuffers(v)
if hdlConfig.WriteTimeout > 0 {
if conn, ok := sub.Writer.(net.Conn); ok {
conn.SetWriteDeadline(time.Now().Add(time.Second * time.Duration(hdlConfig.WriteTimeout)))
conn.SetWriteDeadline(time.Now().Add(hdlConfig.WriteTimeout))
}
}
if _, err := v.WriteTo(sub); err != nil {
@@ -147,7 +147,7 @@ func (c *HDLConfig) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
if hijacker, ok := w.(http.Hijacker); ok && c.WriteTimeout > 0 {
conn, _, _ := hijacker.Hijack()
conn.SetWriteDeadline(time.Now().Add(time.Second * time.Duration(c.WriteTimeout)))
conn.SetWriteDeadline(time.Now().Add(c.WriteTimeout))
sub.SetIO(conn)
} else {
w.(http.Flusher).Flush()