File: example_transcoding_test.go

package info (click to toggle)
golang-github-fxamacker-cbor 2.9.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,800 kB
  • sloc: makefile: 2
file content (85 lines) | stat: -rw-r--r-- 2,046 bytes parent folder | download
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
// Copyright (c) Faye Amacker. All rights reserved.
// Licensed under the MIT License. See LICENSE in the project root for license information.

package cbor_test

import (
	"encoding/json"
	"fmt"
	"io"
	"reflect"

	"github.com/fxamacker/cbor/v2"
)

type TranscoderFunc func(io.Writer, io.Reader) error

func (f TranscoderFunc) Transcode(w io.Writer, r io.Reader) error {
	return f(w, r)
}

func ExampleTranscoder_fromJSON() {
	enc, _ := cbor.EncOptions{
		JSONMarshalerTranscoder: TranscoderFunc(func(w io.Writer, r io.Reader) error {
			d := json.NewDecoder(r)

			for {
				token, err := d.Token()
				if err == io.EOF {
					return nil
				}
				if err != nil {
					return err
				}
				switch token {
				case json.Delim('['):
					if _, err := w.Write([]byte{0x9f}); err != nil {
						return err
					}
				case json.Delim('{'):
					if _, err := w.Write([]byte{0xbf}); err != nil {
						return err
					}
				case json.Delim(']'), json.Delim('}'):
					if _, err := w.Write([]byte{0xff}); err != nil {
						return err
					}
				default:
					b, err := cbor.Marshal(token)
					if err != nil {
						return err
					}
					if _, err := w.Write(b); err != nil {
						return err
					}
				}
			}
		}),
	}.EncMode()

	got, _ := enc.Marshal(json.RawMessage(`{"a": [true, "z", {"y": 3.14}], "b": {"c": null}}`))
	diag, _ := cbor.Diagnose(got)
	fmt.Println(diag)
	// Output: {_ "a": [_ true, "z", {_ "y": 3.14}], "b": {_ "c": null}}
}

func ExampleTranscoder_toJSON() {
	var dec cbor.DecMode
	dec, _ = cbor.DecOptions{
		DefaultMapType: reflect.TypeOf(map[string]any{}),
		JSONUnmarshalerTranscoder: TranscoderFunc(func(w io.Writer, r io.Reader) error {
			var tmp any
			if err := dec.NewDecoder(r).Decode(&tmp); err != nil {
				return err
			}
			return json.NewEncoder(w).Encode(tmp)
		}),
	}.DecMode()

	var got json.RawMessage
	if err := dec.Unmarshal(cbor.RawMessage{0xa2, 0x61, 'a', 0x01, 0x61, 'b', 0x83, 0xf4, 0xf5, 0xf6}, &got); err != nil {
		panic(err)
	}
	fmt.Println(string(got))
	// Output: {"a":1,"b":[false,true,null]}
}