gurl

snail007 · Feb 7, 2025 · b6a238a · b6a238a
1 parent 7fb5512
commit b6a238a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/util/url/url.go b/util/url/url.go
@@ -81,6 +81,11 @@ func (s *Builder) Query(data map[string]string) *Builder {
 	return s
 }
 
+func (s *Builder) SetQuery(k, v string) *Builder {
+	s.query[k] = v
+	return s
+}
+
 func (s *Builder) HostsURL(hosts []string) (urlArr []string) {
 	h0 := s.URL.Host
 	for _, h := range hosts {

diff --git a/util/url/url_test.go b/util/url/url_test.go
@@ -32,6 +32,8 @@ func TestURLBuilder_Query(t *testing.T) {
 	assert.Equal(t, "http://a.com:88/", link.String())
 	link = NewBuilder().Host("a.com").Port("88")
 	assert.Equal(t, "http://a.com:88/", link.String())
+	link = NewBuilder().Host("a.com").Port("88").SetQuery("a", "1")
+	assert.Equal(t, "http://a.com:88/?a=1", link.String())
 }
 
 func TestGetConcatChar(t *testing.T) {