From: Matt Joiner <anacrolix@gmail.com>
Date: Wed, 28 Dec 2022 23:22:03 +0000 (+1100)
Subject: Add Context parameter to SendResponse
X-Git-Tag: v1.49.0~7^2~8
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=7d3d4bc08834b406e97c53cf240f16a19004c49c;p=btrtrc.git

Add Context parameter to SendResponse
---

diff --git a/tracker/udp/server/server.go b/tracker/udp/server/server.go
index 9df2fc67..cd9d9f0f 100644
--- a/tracker/udp/server/server.go
+++ b/tracker/udp/server/server.go
@@ -35,7 +35,7 @@ type AnnounceTracker = trackerServer.AnnounceTracker
 
 type Server struct {
 	ConnTracker  ConnectionTracker
-	SendResponse func(data []byte, addr net.Addr) (int, error)
+	SendResponse func(ctx context.Context, data []byte, addr net.Addr) (int, error)
 	Announce     *trackerServer.AnnounceHandler
 }
 
@@ -160,7 +160,7 @@ func (me *Server) handleAnnounce(
 		return err
 	}
 	buf.Write(b)
-	n, err := me.SendResponse(buf.Bytes(), source)
+	n, err := me.SendResponse(ctx, buf.Bytes(), source)
 	if err != nil {
 		return err
 	}
@@ -183,7 +183,7 @@ func (me *Server) handleConnect(ctx context.Context, source RequestSourceAddr, t
 		TransactionId: tid,
 	})
 	udp.Write(&buf, udp.ConnectionResponse{connId})
-	n, err := me.SendResponse(buf.Bytes(), source)
+	n, err := me.SendResponse(ctx, buf.Bytes(), source)
 	if err != nil {
 		return err
 	}