Remove context from SendDatagram

connection.go

@@ -2343,7 +2343,7 @@ func (s *connection) onStreamCompleted(id protocol.StreamID) {
 	}
 }
 
-func (s *connection) SendDatagram(ctx context.Context, p []byte) error {
+func (s *connection) SendDatagram(p []byte) error {
 	if !s.supportsDatagrams() {
 		return errors.New("datagram support disabled")
 	}
@@ -2359,7 +2359,7 @@ func (s *connection) SendDatagram(ctx context.Context, p []byte) error {
 	}
 	f.Data = make([]byte, len(p))
 	copy(f.Data, p)
-	return s.datagramQueue.AddAndWait(ctx, f)
+	return s.datagramQueue.AddAndWait(f)
 }
 
 func (s *connection) ReceiveDatagram(ctx context.Context) ([]byte, error) {

datagram_queue.go

@@ -37,9 +37,8 @@ type datagramQueue struct {
 }
 
 type queuedDatagramFrame struct {
-	cancelChan <-chan struct{}
-	frame      *wire.DatagramFrame
-	peekCount  uint8
+	frame     *wire.DatagramFrame
+	peekCount uint8
 }
 
 func newDatagramQueue(hasData func(), logger utils.Logger) *datagramQueue {
@@ -57,13 +56,10 @@ func newDatagramQueue(hasData func(), logger utils.Logger) *datagramQueue {
 // It blocks until the frame has been dequeued.
 func (h *datagramQueue) AddAndWait(f *wire.DatagramFrame) error {
 	frame := &queuedDatagramFrame{
-		cancelChan: ctx.Done(),
-		frame:      f,
+		frame: f,
 	}
 
 	select {
-	case <-ctx.Done():
-		return dropDatagramCtxCancelledErr
 	case h.sendQueue <- frame:
 		h.hasData()
 	case <-h.closed:

datagram_queue_test.go

@@ -70,7 +70,6 @@ var _ = Describe("Datagram Queue", func() {
 		It("drop after peeking too many times", func() {
 			f := &wire.DatagramFrame{Data: []byte("foobar")}
 			errChan := make(chan error, 1)
-			ctx1, cancel1 := context.WithCancel(context.Background())
 
 			go func() {
 				defer GinkgoRecover()
@@ -84,23 +83,6 @@ var _ = Describe("Datagram Queue", func() {
 			Eventually(errChan).Should(Receive(Equal(&DatagramQueuedTooLong{})))
 		})
 
-		It("drop after peeking too many times", func() {
-			f := &wire.DatagramFrame{Data: []byte("foobar")}
-			errChan := make(chan error, 1)
-
-			go func() {
-				defer GinkgoRecover()
-				mtuDiscoverer.EXPECT().CurrentSize().Return(1234)
-				errChan <- queue.AddAndWait(context.Background(), f)
-			}()
-			Eventually(queued).Should(HaveLen(1))
-			for i := 0; i < int(maxPeekAttempt); i++ {
-				Expect(queue.Peek()).To(Equal(f))
-			}
-			Expect(queue.Peek()).To(BeNil())
-			Eventually(errChan).Should(Receive(Equal(dropDatagramCtxCancelledErr)))
-		})
-
 		It("closes", func() {
 			errChan := make(chan error, 1)
 			go func() {

interface.go

@@ -188,7 +188,7 @@ type Connection interface {
 	ConnectionState() ConnectionState
 
 	// SendDatagram sends a message as a datagram, as specified in RFC 9221.
-	SendDatagram(context.Context, []byte) error
+	SendDatagram([]byte) error
 	// ReceiveDatagram gets a message received in a datagram, as specified in RFC 9221.
 	ReceiveDatagram(context.Context) ([]byte, error)
 }