Fixes blevesearch#1873 Adding timeout in the search handler

Likith101 · Nov 14, 2023 · 7226997 · 7226997
1 parent 0c31cd2
commit 7226997
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/http/search.go b/http/search.go
@@ -20,7 +20,6 @@ import (
 	"fmt"
 	"io"
 	"net/http"
-	"strconv"
 	"time"
 
 	"github.com/blevesearch/bleve/v2"
@@ -89,12 +88,12 @@ func (h *SearchHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 	if timeoutStr == "" {
 		ctx = context.Background()
 	} else {
-		timeout, err := strconv.Atoi(timeoutStr)
+		timeout, err := time.ParseDuration(timeoutStr)
 		if err != nil {
 			showError(w, req, fmt.Sprintf("error parsing timeout value: %v", err), 400)
 			return
 		}
-		ctx, _ = context.WithTimeout(context.Background(), time.Duration(timeout)*time.Millisecond)
+		ctx, _ = context.WithTimeout(context.Background(), timeout)
 	}
 
 	// execute the query