replace ListenTCP with Listen

This commit is contained in:
aler9
2020-12-13 15:13:43 +01:00
parent 42877b4d52
commit c11f056530
3 changed files with 7 additions and 12 deletions

View File

@@ -8,7 +8,7 @@ import (
// Server is a RTSP server.
type Server struct {
conf ServerConf
listener *net.TCPListener
listener net.Listener
}
// Close closes the server.

View File

@@ -31,8 +31,8 @@ type ServerConf struct {
ReadBufferCount int
// function used to initialize the TCP listener.
// It defaults to net.ListenTCP.
ListenTCP func(network string, address *net.TCPAddr) (*net.TCPListener, error)
// It defaults to net.Listen
Listen func(network string, address string) (net.Listener, error)
}
// Serve starts a server on the given address.
@@ -46,16 +46,11 @@ func (c ServerConf) Serve(address string) (*Server, error) {
if c.ReadBufferCount == 0 {
c.ReadBufferCount = 1
}
if c.ListenTCP == nil {
c.ListenTCP = net.ListenTCP
if c.Listen == nil {
c.Listen = net.Listen
}
addr, err := net.ResolveTCPAddr("tcp", address)
if err != nil {
return nil, err
}
listener, err := c.ListenTCP("tcp", addr)
listener, err := c.Listen("tcp", address)
if err != nil {
return nil, err
}

View File

@@ -214,7 +214,7 @@ func TestServerPublishReadTCP(t *testing.T) {
require.Equal(t, 0, cnt2.wait())
}
func TestServerTeardown(t *testing.T) {
func TestServerTeardownResponse(t *testing.T) {
ts, err := newTestServ()
require.NoError(t, err)
defer ts.close()