msgpack/example_encodeMapStringString_test.go at v2 · xichen2020/msgpack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package msgpack_test

import (
	"bytes"
	"fmt"

	"gopkg.in/vmihailenco/msgpack.v2"
)

func Example_encodeMapStringString() {
	buf := &bytes.Buffer{}

	m := map[string]string{"foo1": "bar1", "foo2": "bar2", "foo3": "bar3"}
	keys := []string{"foo1", "foo3"}

	encodedMap, err := encodeMap(m, keys...)
	if err != nil {
		panic(err)
	}

	_, err = buf.Write(encodedMap)
	if err != nil {
		panic(err)
	}

	decoder := msgpack.NewDecoder(buf)
	value, err := decoder.DecodeMap()
	if err != nil {
		panic(err)
	}

	decodedMapValue := value.(map[interface{}]interface{})

	for _, key := range keys {
		fmt.Printf("%#v: %#v, ", key, decodedMapValue[key])
	}

	// Output: "foo1": "bar1", "foo3": "bar3",
}

func Example_decodeMapStringString() {
	decodedMap := make(map[string]string)
	buf := &bytes.Buffer{}

	m := map[string]string{"foo1": "bar1", "foo2": "bar2", "foo3": "bar3"}
	keys := []string{"foo1", "foo3", "foo2"}

	encodedMap, err := encodeMap(m, keys...)
	if err != nil {
		panic(err)
	}

	_, err = buf.Write(encodedMap)
	if err != nil {
		panic(err)
	}

	decoder := msgpack.NewDecoder(buf)

	n, err := decoder.DecodeMapLen()
	if err != nil {
		panic(err)
	}

	for i := 0; i < n; i++ {
		key, err := decoder.DecodeString()
		if err != nil {
			panic(err)
		}
		value, err := decoder.DecodeString()
		if err != nil {
			panic(err)
		}
		decodedMap[key] = value
	}

	for _, key := range keys {
		fmt.Printf("%#v: %#v, ", key, decodedMap[key])
	}
	// Output: "foo1": "bar1", "foo3": "bar3", "foo2": "bar2",
}

func encodeMap(m map[string]string, keys ...string) ([]byte, error) {
	buf := &bytes.Buffer{}
	encoder := msgpack.NewEncoder(buf)

	if err := encoder.EncodeMapLen(len(keys)); err != nil {
		return nil, err
	}

	for _, key := range keys {
		if err := encoder.EncodeString(key); err != nil {
			return nil, err
		}
		if err := encoder.EncodeString(m[key]); err != nil {
			return nil, err
		}
	}

	return buf.Bytes(), nil
}