aboutsummaryrefslogtreecommitdiff
path: root/tracker
diff options
context:
space:
mode:
authorDimitri Sokolyuk <demon@dim13.org>2016-06-16 00:07:12 +0200
committerDimitri Sokolyuk <demon@dim13.org>2016-06-16 00:07:12 +0200
commit662556e27a4d7a53013bae7c4d2a65fc541d6fa2 (patch)
tree324be740df1b36baa3293b5ff0ebcafd951b4809 /tracker
parentded9f5352c3995e20af6fb61bc4c138474969657 (diff)
Add query encoder
Diffstat (limited to 'tracker')
-rw-r--r--tracker/query.go37
-rw-r--r--tracker/query_test.go20
2 files changed, 57 insertions, 0 deletions
diff --git a/tracker/query.go b/tracker/query.go
new file mode 100644
index 0000000..bb741dd
--- /dev/null
+++ b/tracker/query.go
@@ -0,0 +1,37 @@
+package tracker
+
+import (
+ "errors"
+ "net/url"
+ "reflect"
+ "strconv"
+ "strings"
+)
+
+func MarshalQuery(v interface{}) (string, error) {
+ val := reflect.ValueOf(v)
+ return marshalQuery(val)
+}
+
+func marshalQuery(v reflect.Value) (string, error) {
+ if v.Kind() != reflect.Struct {
+ return "", errors.New("must be a struct")
+ }
+ q := url.Values{}
+ t := v.Type()
+ for i := 0; i < t.NumField(); i++ {
+ name := strings.ToLower(t.Field(i).Name)
+ f := v.Field(i)
+ switch f.Kind() {
+ case reflect.Int:
+ if x := int(f.Int()); x != 0 {
+ q.Add(name, strconv.Itoa(int(f.Int())))
+ }
+ case reflect.String:
+ if s := f.String(); s != "" {
+ q.Add(name, f.String())
+ }
+ }
+ }
+ return "?" + q.Encode(), nil
+}
diff --git a/tracker/query_test.go b/tracker/query_test.go
new file mode 100644
index 0000000..3959255
--- /dev/null
+++ b/tracker/query_test.go
@@ -0,0 +1,20 @@
+package tracker
+
+import "testing"
+
+func TestMarshalQuery(t *testing.T) {
+ q := struct {
+ A string
+ B int
+ C string
+ }{
+ A: "test",
+ B: 100,
+ }
+ v, err := MarshalQuery(q)
+ if err != nil {
+ t.Error(err)
+ }
+ t.Log(v)
+
+}