From 299c87376c091b0082ae2f98aaf5f74818e6d152 Mon Sep 17 00:00:00 2001
From: WRABZY <63013822+WRABZY@users.noreply.github.com>
Date: Sat, 24 Jan 2026 13:33:15 +0300
Subject: [PATCH] fix goroutine leak on fetch sleep for write requests

---
 dependency/vault_pki.go   | 7 ++++++-
 dependency/vault_write.go | 7 ++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/dependency/vault_pki.go b/dependency/vault_pki.go
index 37bc3d6a8..935be22cf 100644
--- a/dependency/vault_pki.go
+++ b/dependency/vault_pki.go
@@ -98,7 +98,12 @@ func (d *VaultPKIQuery) Fetch(clients *ClientSet, opts *QueryOptions) (interface
 	}
 	select {
 	case dur := <-d.sleepCh:
-		time.Sleep(dur)
+		select {
+		case <-time.After(dur):
+			break
+		case <-d.stopCh:
+			return nil, nil, ErrStopped
+		}
 	default:
 	}
 
diff --git a/dependency/vault_write.go b/dependency/vault_write.go
index 2232d1c79..00c7ed21b 100644
--- a/dependency/vault_write.go
+++ b/dependency/vault_write.go
@@ -61,7 +61,12 @@ func (d *VaultWriteQuery) Fetch(clients *ClientSet, opts *QueryOptions,
 	}
 	select {
 	case dur := <-d.sleepCh:
-		time.Sleep(dur)
+		select {
+		case <-time.After(dur):
+			break
+		case <-d.stopCh:
+			return nil, nil, ErrStopped
+		}
 	default:
 	}