blob: a4236d8bc44ce3068708a2201d8d252d7b5c3dc4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
// Instead of using the default new.Listener this file will construct a custom
// one. The main purpose for this is to have more control over the settings
// (like keep-alive) and to retrieve the assigned port when using port 0.
package app
import (
"fmt"
"net"
"time"
)
func newListener(cfg *ServerConfig) (net.Listener, error) {
addr := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)
ln, err := net.Listen("tcp", addr)
if err != nil {
return nil, err
}
// set actual port on config object (in case original port was 0)
cfg.Port = ln.Addr().(*net.TCPAddr).Port
return tcpListener{ln.(*net.TCPListener)}, nil
}
// custom TCP listener with keep-alive timeout
type tcpListener struct {
*net.TCPListener
}
func (ln tcpListener) Accept() (net.Conn, error) {
tc, err := ln.AcceptTCP()
if err != nil {
return nil, err
}
tc.SetKeepAlive(true)
tc.SetKeepAlivePeriod(3 * time.Minute)
return tc, nil
}
|
