Merge pull request #9 from Goryudyuma/useable-http2

fix benchmarker useable http/2
showwin · Jun 23, 2018 · a4120dd · a4120dd
2 parents aab9bd3 + d947715
commit a4120dd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/admin/benchmarker/request.go b/admin/benchmarker/request.go
@@ -12,6 +12,7 @@ import (
 	"time"
 
 	"github.com/PuerkitoBio/goquery"
+	"golang.org/x/net/http2"
 )
 
 func getInitialize() {
@@ -74,6 +75,9 @@ func httpsRequest(method string, path string, params url.Values) int {
 	tr := &http.Transport{
 		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
 	}
+	if err := http2.ConfigureTransport(tr); err != nil {
+		log.Fatalf("Failed to configure h2 transport: %s", err)
+	}
 	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
 	client := http.Client{Transport: tr}
 
@@ -92,6 +96,9 @@ func httpsRequestDoc(method string, path string, params url.Values) *goquery.Doc
 	tr := &http.Transport{
 		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
 	}
+	if err := http2.ConfigureTransport(tr); err != nil {
+		log.Fatalf("Failed to configure h2 transport: %s", err)
+	}
 	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
 	client := http.Client{Transport: tr}