瀏覽代碼

Add option to close TCP connection after message

tags/v0.2.0
Russ Garrett 7 年之前
父節點
當前提交
058f0f4eee
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 8 行新增2 行删除
  1. 4
    1
      examples/irccat.json
  2. 4
    1
      tcplistener/tcplistener.go

+ 4
- 1
examples/irccat.json 查看文件

1
 {
1
 {
2
-  "tcp_listen": ":12345",
2
+  "tcp": {
3
+    "listen": ":12345",
4
+    "close_after_message": false
5
+  },
3
   "http": {
6
   "http": {
4
     "listen": ":8045",
7
     "listen": ":8045",
5
     "tls": false,
8
     "tls": false,

+ 4
- 1
tcplistener/tcplistener.go 查看文件

20
 	var err error
20
 	var err error
21
 
21
 
22
 	listener := TCPListener{}
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
 	if err != nil {
24
 	if err != nil {
25
 		return nil, err
25
 		return nil, err
26
 	}
26
 	}
57
 			log.Infof("[%s] message: %s", conn.RemoteAddr(), msg)
57
 			log.Infof("[%s] message: %s", conn.RemoteAddr(), msg)
58
 			l.parseMessage(msg)
58
 			l.parseMessage(msg)
59
 		}
59
 		}
60
+		if viper.GetBool("tcp.close_after_message") {
61
+			conn.Close()
62
+		}
60
 	}
63
 	}
61
 }
64
 }
62
 
65
 

Loading…
取消
儲存