Merge pull request #121 from bp0lr/master

fix bug on MakeRequestQuery
hahwul · Sep 28, 2020 · 64eb641 · 64eb641
2 parents 527cff1 + b159382
commit 64eb641
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/pkg/optimization/optimization.go b/pkg/optimization/optimization.go
@@ -82,7 +82,9 @@ func MakeRequestQuery(target, param, payload, ptype string, options model.Option
 
 	payload = url.QueryEscape(payload)
 	u, _ := url.Parse(target)
+
 	data := u.String()
+
 	if options.Data != "" {
 		tempParam, _ := url.ParseQuery(options.Data)
 		var body string
@@ -100,11 +102,22 @@ func MakeRequestQuery(target, param, payload, ptype string, options model.Option
 
 	} else {
 		tempParam := u.Query()
-		if tempParam[param] == nil {
-			data = "?" + param + "=" + payload + "&" + data
+
+		if tempParam[param] == nil {			
+			if(strings.Contains(data, "?")) {
+				data = data + "&"+param+"="+payload
+			} else{
+				data = data + "?"+param+"="+payload
+			}						
 		} else {
-			data = strings.Replace(data, param+"="+url.QueryEscape(tempParam[param][0]), param+"="+url.QueryEscape(tempParam[param][0])+payload, 1)
+			data = strings.Replace(data, param+"=" + url.QueryEscape(tempParam[param][0]), param + "=" + url.QueryEscape(tempParam[param][0])+payload, 1)
+			if tempParam[param] == nil {
+				data = "?" + param + "=" + payload + "&" + data
+			} else {
+				data = strings.Replace(data, param+"="+url.QueryEscape(tempParam[param][0]), param+"="+url.QueryEscape(tempParam[param][0])+payload, 1)
+			}
 		}
+
 		tempURL, _ := url.Parse(data)
 		tempQuery := tempURL.Query()
 		tempURL.RawQuery = tempQuery.Encode()