aboutsummaryrefslogtreecommitdiff
path: root/elegoo.go
blob: a7ed182dbdd0940c5c00c45212097276175c583a (plain)
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
package elegoo

import (
	"bufio"
	"io"

	"github.com/dim13/cobs"
	proto "github.com/golang/protobuf/proto"
)

//go:generate sh -c "protoc --go_out=. --nanopb_out=elegoo/ *.proto"

func Send(w io.Writer, pb proto.Message) error {
	buf := new(proto.Buffer)
	if err := buf.EncodeMessage(pb); err != nil {
		return err
	}
	_, err := w.Write(cobs.Encode(buf.Bytes()))
	return err
}

func Recv(buf *bufio.Reader, pb proto.Message) error {
	block, err := buf.ReadBytes(0)
	if err != nil {
		return err
	}
	return proto.NewBuffer(cobs.Decode(block)).DecodeMessage(pb)
}