From a27e0035083b8e66ee446c949f98405c1ac2791d Mon Sep 17 00:00:00 2001
From: Dmitrii Stepanov <d.stepanov@yadro.com>
Date: Mon, 21 Apr 2025 10:53:51 +0300
Subject: [PATCH] [#1709] qos: Add context.Done check before schedule request

Do not push request to schedule queue, if context is already cancelled.

Change-Id: Ieef837a7d423158e3dbb0c3b4efecaa20744c845
Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
---
 internal/qos/limiter.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/internal/qos/limiter.go b/internal/qos/limiter.go
index 5851d7626..2d7de32fc 100644
--- a/internal/qos/limiter.go
+++ b/internal/qos/limiter.go
@@ -149,6 +149,11 @@ func (n *mClockLimiter) WriteRequest(ctx context.Context) (ReleaseFunc, error) {
 }
 
 func requestArrival(ctx context.Context, s scheduler, stats map[string]*stat) (ReleaseFunc, error) {
+	select {
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	default:
+	}
 	tag, ok := tagging.IOTagFromContext(ctx)
 	if !ok {
 		tag = IOTagClient.String()