From 80a1de60e61ddc8ef71ae89c3d4f51ce4918081c Mon Sep 17 00:00:00 2001
From: 9seconds <nineseconds@yandex.ru>
Date: Wed, 17 Mar 2021 21:24:42 +0300
Subject: [PATCH] Always close connection on context finished

---
 mtglib/proxy.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mtglib/proxy.go b/mtglib/proxy.go
index 6c68415..6cec3bc 100644
--- a/mtglib/proxy.go
+++ b/mtglib/proxy.go
@@ -29,6 +29,11 @@ func (p *Proxy) ServeConn(conn net.Conn) {
 	ctx := newStreamContext(p.ctx, p.logger, conn)
 	defer ctx.Close()
 
+	go func() {
+		<-ctx.Done()
+		ctx.Close()
+	}()
+
 	p.eventStream.Send(ctx, EventStart{
 		CreatedAt: time.Now(),
 		ConnID:    ctx.connID,