Support set query string.

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

gofight.go

@@ -14,6 +14,13 @@
 //      "X-Version": version,
 //    })
 //
+// Set query string: Using SetQUERY to generate query string data.
+//
+//    SetQUERY(gofight.H{
+//      "a": "1",
+//      "b": "2",
+//    })
+//
 // POST FORM Data: Using SetFORM to generate form data.
 //
 //    SetFORM(gofight.H{
@@ -212,6 +219,23 @@ func (rc *RequestConfig) SetFORM(body H) *RequestConfig {
 	return rc
 }
 
+// SetQUERY supply query string.
+func (rc *RequestConfig) SetQUERY(query H) *RequestConfig {
+	f := make(url.Values)
+
+	for k, v := range query {
+		f.Set(k, v)
+	}
+
+	if strings.Contains(rc.Path, "?") {
+		rc.Path = rc.Path + "&" + f.Encode()
+	} else {
+		rc.Path = rc.Path + "?" + f.Encode()
+	}
+
+	return rc
+}
+
 // SetBody supply raw body.
 func (rc *RequestConfig) SetBody(body string) *RequestConfig {
 	if len(body) > 0 {

gofight_test.go

@@ -380,3 +380,23 @@ func TestEchoOptions(t *testing.T) {
 			assert.Equal(t, http.StatusOK, r.Status())
 		})
 }
+
+func TestSetQueryString(t *testing.T) {
+	r := New()
+
+	r.GET("/hello").
+		SetQUERY(H{
+			"a": "1",
+			"b": "2",
+		})
+
+	assert.Equal(t, "/hello?a=1&b=2", r.Path)
+
+	r.GET("/hello?foo=bar").
+		SetQUERY(H{
+			"a": "1",
+			"b": "2",
+		})
+
+	assert.Equal(t, "/hello?foo=bar&a=1&b=2", r.Path)
+}