Function ReadMapBytes

msgp/iter.go:219–252

ReadMapBytes returns an iterator over key/value pairs from a MessagePack map encoded in b. The iterator yields K,V pairs, and this function also returns a closure to get the remaining bytes and any error.

(b []byte,
	readK func([]byte) (K, []byte, error),
	readV func([]byte) (V, []byte, error))

// The iterator yields K,V pairs, and this function also returns
// a closure to get the remaining bytes and any error.
func ReadMapBytes[K any, V any](b []byte,
	readK func([]byte) (K, []byte, error),
	readV func([]byte) (V, []byte, error)) (iter.Seq2[K, V], func() (remain []byte, err error)) {
	var err error
	var sz uint32
	if IsNil(b) {
		b, err = ReadNilBytes(b)
		return func(yield func(K, V) bool) {}, func() ([]byte, error) { return b, err }
	}
	sz, b, err = ReadMapHeaderBytes(b)
	if err != nil || sz == 0 {
		return func(yield func(K, V) bool) {}, func() ([]byte, error) { return b, err }
	}

	return func(yield func(K, V) bool) {
		for range sz {
			var k K
			k, b, err = readK(b)
			if err != nil {
				err = fmt.Errorf("cannot read map key: %w", err)
				return
			}
			var v V
			v, b, err = readV(b)
			if err != nil {
				err = fmt.Errorf("cannot read map value: %w", err)
				return
			}
			if !yield(k, v) {
				return
			}
		}
	}, func() ([]byte, error) { return b, err }
}

// AppendMap writes a map to the provided buffer.
// The writeK and writeV parameters specify the functions to use to write each key and value of the map.