feat: allow find or set value in JSON using path like a.b.c.

Author: nicola-strappazzon

README.md

@@ -7,6 +7,7 @@ A [Tinybird](https://www.tinybird.co/) module for Go. Why need this module? It p
 - Lightweight and fast.
 - Native Go implementation. No C-bindings, just pure Go
 - Connection pooling for HTTP.
+- Read and write values in nested JSON structures using path-based access.
 - Test your code with mocks.
 - Allow JSON, [NDJSON](http://ndjson.org/) and CSV between tinybird and this module.
 - Parallelize HTTP requests.

data.go

@@ -1,6 +1,10 @@
 package tinybird
 
-import "encoding/json"
+import (
+	"encoding/json"
+	"errors"
+	"strings"
+)
 
 // Row is a generic map-based structure that can hold fields of any type.
 // Each key corresponds to a field name, and each value can be any Go type.
@@ -41,3 +45,63 @@ func (d Data) ToString() (out string) {
 
 	return string(tmp)
 }
+
+func (d Data) Get(in string) (out any) {
+	if len(d) == 0 {
+		return nil
+	}
+
+	parts := strings.Split(in, ".")
+	out = d[0]
+
+	for _, part := range parts {
+		row, ok := out.(Row)
+		if !ok {
+			return nil
+		}
+
+		out, ok = row[part]
+		if !ok {
+			return nil
+		}
+	}
+
+	return out
+}
+
+func (d *Data) Set(in string, value any) error {
+	if d == nil || len(*d) == 0 {
+		return errors.New("empty Data")
+	}
+
+	parts := strings.Split(in, ".")
+	if len(parts) == 0 {
+		return errors.New("empty path")
+	}
+
+	current := (*d)[0]
+
+	for i, p := range parts {
+		if i == len(parts)-1 {
+			current[p] = value
+			return nil
+		}
+
+		next, exists := current[p]
+		if !exists {
+			child := Row{}
+			current[p] = child
+			current = child
+			continue
+		}
+
+		child, ok := next.(Row)
+		if !ok {
+			return errors.New("path collision at " + p)
+		}
+
+		current = child
+	}
+
+	return nil
+}

data_test.go

@@ -70,3 +70,25 @@ func TestToString_OnNilAndEmpty(t *testing.T) {
 	assert.Equal(t, "null", dNil.ToString())
 	assert.Equal(t, "[]", dEmpty.ToString())
 }
+
+func TestGet(t *testing.T) {
+	d := tinybird.Data{
+		{"a": tinybird.Row{"b": 1}},
+		{"b": "ok"},
+	}
+
+	assert.Nil(t, d.Get(""))
+	assert.Nil(t, d.Get("ab"))
+	assert.Equal(t, tinybird.Row(tinybird.Row{"b": 1}), d.Get("a"))
+	assert.Equal(t, 1, d.Get("a.b"))
+}
+
+func TestSet(t *testing.T) {
+	d := tinybird.Data{
+		{"a": tinybird.Row{"b": 1}},
+		{"b": "ok"},
+	}
+
+	assert.Nil(t, d.Set("a.b", 2))
+	assert.Equal(t, 2, d.Get("a.b"))
+}