beam/data: convenience Message type for chained manipulation

Docker-DCO-1.1-Signed-off-by: Solomon Hykes <solomon@docker.com> (github: shykes)

beam/data/message.go

@@ -0,0 +1,47 @@
+package data
+
+import (
+)
+
+type Message string
+
+func Empty() Message {
+	return Message(Encode(nil))
+}
+
+func (m Message) Add(k, v string) Message {
+	data, err := Decode(string(m))
+	if err != nil {
+		return m
+	}
+	if values, exists := data[k]; exists {
+		data[k] = append(values, v)
+	} else {
+		data[k] = []string{v}
+	}
+	return Message(Encode(data))
+}
+
+func (m Message) Set(k string, v ...string) Message {
+	data, err := Decode(string(m))
+	if err != nil {
+		panic(err)
+		return m
+	}
+	data[k] = v
+	return Message(Encode(data))
+}
+
+func (m Message) Del(k string) Message {
+	data, err := Decode(string(m))
+	if err != nil {
+		panic(err)
+		return m
+	}
+	delete(data, k)
+	return Message(Encode(data))
+}
+
+func (m Message) String() string {
+	return string(m)
+}

beam/data/message_test.go

@@ -0,0 +1,53 @@
+package data
+
+import (
+	"testing"
+)
+
+func TestEmptyMessage(t *testing.T) {
+	m := Empty()
+	if m.String() != Encode(nil) {
+		t.Fatalf("%v != %v", m.String(), Encode(nil))
+	}
+}
+
+func TestSetMessage(t *testing.T) {
+	m := Empty().Set("foo", "bar")
+	output := m.String()
+	expectedOutput := "000;3:foo,6:3:bar,,"
+	if output != expectedOutput {
+		t.Fatalf("'%v' != '%v'", output, expectedOutput)
+	}
+	decodedOutput, err := Decode(output)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(decodedOutput) != 1 {
+		t.Fatalf("wrong output data: %#v\n", decodedOutput)
+	}
+}
+
+func TestSetMessageTwice(t *testing.T) {
+	m := Empty().Set("foo", "bar").Set("ga", "bu")
+	output := m.String()
+	expectedOutput := "000;3:foo,6:3:bar,,2:ga,5:2:bu,,"
+	if output != expectedOutput {
+		t.Fatalf("'%v' != '%v'", output, expectedOutput)
+	}
+	decodedOutput, err := Decode(output)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(decodedOutput) != 2 {
+		t.Fatalf("wrong output data: %#v\n", decodedOutput)
+	}
+}
+
+func TestSetDelMessage(t *testing.T) {
+	m := Empty().Set("foo", "bar").Del("foo")
+	output := m.String()
+	expectedOutput := Encode(nil)
+	if output != expectedOutput {
+		t.Fatalf("'%v' != '%v'", output, expectedOutput)
+	}
+}