|
@@ -20,7 +20,7 @@ func New() (*TCPListener, error) {
|
20
|
20
|
var err error
|
21
|
21
|
|
22
|
22
|
listener := TCPListener{}
|
23
|
|
- listener.socket, err = net.Listen("tcp", viper.GetString("tcp_listen"))
|
|
23
|
+ listener.socket, err = net.Listen("tcp", viper.GetString("tcp.listen"))
|
24
|
24
|
if err != nil {
|
25
|
25
|
return nil, err
|
26
|
26
|
}
|
|
@@ -57,6 +57,9 @@ func (l *TCPListener) handleConnection(conn net.Conn) {
|
57
|
57
|
log.Infof("[%s] message: %s", conn.RemoteAddr(), msg)
|
58
|
58
|
l.parseMessage(msg)
|
59
|
59
|
}
|
|
60
|
+ if viper.GetBool("tcp.close_after_message") {
|
|
61
|
+ conn.Close()
|
|
62
|
+ }
|
60
|
63
|
}
|
61
|
64
|
}
|
62
|
65
|
|