Add Context parameter to SendResponse

diff --git a/tracker/udp/server/server.go b/tracker/udp/server/server.go
index 9df2fc67caee6909adbd472a50f15499d1568102..cd9d9f0fdc86f566b0baa7241c78ca5b74daeb80 100644 (file)
--- 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
        }