Fix leaking ticker channels, fixes #2v0.1.2

2 files changed, 6 insertions, 4 deletions

diff --git a/http.go b/http.go
index 6abac6d..b5982e8 100644
--- a/http.go
+++ b/http.go
@@ -37,12 +37,13 @@ func httpHandler(conn net.Conn, delay time.Duration) {
 		io.Copy(ioutil.Discard, conn)
 	}()
 
-	tick := time.Tick(delay)
+	ticker := time.NewTicker(delay)
+	defer ticker.Stop()
 	for {
 		select {
 		case <-eof:
 			return
-		case <-tick:
+		case <-ticker.C:
 			_, err := fmt.Fprintf(conn, "X-%0x: %0x\r\n", rand.Uint32(), rand.Uint32())
 			if err != nil {
 				return
diff --git a/ssh.go b/ssh.go
index cced23e..1a24498 100644
--- a/ssh.go
+++ b/ssh.go
@@ -16,12 +16,13 @@ func sshHandler(conn net.Conn, delay time.Duration) {
 		io.Copy(ioutil.Discard, conn)
 	}()
 
-	tick := time.Tick(delay)
+	ticker := time.NewTicker(delay)
+	defer ticker.Stop()
 	for {
 		select {
 		case <-eof:
 			return
-		case <-tick:
+		case <-ticker.C:
 			_, err := fmt.Fprintf(conn, "%x\r\n", rand.Uint32())
 			if err != nil {
 				return