ssob/encoder.go

43 lines
670 B
Go
Raw Normal View History

2019-10-04 12:28:45 +02:00
package ssob
import (
"fmt"
2019-10-04 12:28:45 +02:00
"io"
"sync"
)
type Encoder struct {
mutex sync.Mutex
w io.Writer
}
func NewEncoder(w io.Writer) *Encoder {
enc := new(Encoder)
enc.w = w
return enc
}
func (enc *Encoder) Encode(e interface{}) (err error) {
2021-09-08 13:27:39 +02:00
_, v, err := parseInterface(e)
if err != nil {
return err
}
2021-09-06 23:33:50 +02:00
var b []byte
2021-09-08 13:27:39 +02:00
if b, err = marshalVal(v.Interface()); err != nil {
2021-09-06 23:33:50 +02:00
return fmt.Errorf("ssob: Cannot encode value: %w", err)
}
2021-10-21 11:46:02 +02:00
wb := marshalUint32(uint32(len(b)))
wb = append(wb, b...)
2021-09-06 23:33:50 +02:00
enc.mutex.Lock()
defer enc.mutex.Unlock()
2021-10-21 11:46:02 +02:00
if _, err = enc.w.Write(wb); err != nil {
2021-09-06 23:33:50 +02:00
return fmt.Errorf("ssob: Cannot encode value: %w", err)
}
return nil
2021-09-02 00:08:59 +02:00
}