Add option to close TCP connection after message

irccloud · Jan 17, 2017 · 058f0f4 · 058f0f4
1 parent 5dd5adb
commit 058f0f4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/examples/irccat.json b/examples/irccat.json
@@ -1,5 +1,8 @@
 {
-  "tcp_listen": ":12345",
+  "tcp": {
+    "listen": ":12345",
+    "close_after_message": false
+  },
   "http": {
     "listen": ":8045",
     "tls": false,

diff --git a/tcplistener/tcplistener.go b/tcplistener/tcplistener.go
@@ -20,7 +20,7 @@ func New() (*TCPListener, error) {
 	var err error
 
 	listener := TCPListener{}
-	listener.socket, err = net.Listen("tcp", viper.GetString("tcp_listen"))
+	listener.socket, err = net.Listen("tcp", viper.GetString("tcp.listen"))
 	if err != nil {
 		return nil, err
 	}
@@ -57,6 +57,9 @@ func (l *TCPListener) handleConnection(conn net.Conn) {
 			log.Infof("[%s] message: %s", conn.RemoteAddr(), msg)
 			l.parseMessage(msg)
 		}
+		if viper.GetBool("tcp.close_after_message") {
+			conn.Close()
+		}
 	}
 }