From a31dc49bbb8d1896b906ae1f6e9c206e696bb4d1 Mon Sep 17 00:00:00 2001
From: gVisor bot <gvisor-bot@google.com>
Date: Sun, 13 Oct 2019 12:11:26 +0800
Subject: [PATCH] Optimization: use context in urltest speed test (#356)

---
 adapters/outbound/urltest.go | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/adapters/outbound/urltest.go b/adapters/outbound/urltest.go
index 2bdb872d..33f80717 100644
--- a/adapters/outbound/urltest.go
+++ b/adapters/outbound/urltest.go
@@ -75,12 +75,15 @@ func (u *URLTest) Destroy() {
 
 func (u *URLTest) loop() {
 	tick := time.NewTicker(u.interval)
-	go u.speedTest()
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	go u.speedTest(ctx)
 Loop:
 	for {
 		select {
 		case <-tick.C:
-			go u.speedTest()
+			go u.speedTest(ctx)
 		case <-u.done:
 			break Loop
 		}
@@ -104,13 +107,13 @@ func (u *URLTest) fallback() {
 	u.fast = fast
 }
 
-func (u *URLTest) speedTest() {
+func (u *URLTest) speedTest(ctx context.Context) {
 	if !atomic.CompareAndSwapInt32(&u.once, 0, 1) {
 		return
 	}
 	defer atomic.StoreInt32(&u.once, 0)
 
-	ctx, cancel := context.WithTimeout(context.Background(), defaultURLTestTimeout)
+	ctx, cancel := context.WithTimeout(ctx, defaultURLTestTimeout)
 	defer cancel()
 	picker := picker.WithoutAutoCancel(ctx)
 	for _, p := range u.proxies {