Return immediately from Job.wait() if a Job has a response

goccy · Apr 11, 2024 · 2437402 · 2437402
1 parent 5ad569f
commit 2437402
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/metadata/job.go b/internal/metadata/job.go
@@ -47,6 +47,10 @@ func (j *Job) Wait(ctx context.Context) (*internaltypes.QueryResponse, error) {
 	j.mu.Lock()
 	defer j.mu.Unlock()
 
+	if j.response != nil {
+		return j.response, nil
+	}
+
 	ticker := time.NewTicker(100 * time.Millisecond)
 	defer ticker.Stop()
 	for {