mirror of
https://github.com/netsampler/goflow2.git
synced 2024-05-06 15:54:52 +00:00
e52a053c9c
* Improve format registration * less interleaving (json does not require importing protobuf) * generic text renderer * escape for strings in text/json formatter
42 lines
811 B
Go
42 lines
811 B
Go
package json
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/netsampler/goflow2/format"
|
|
"github.com/netsampler/goflow2/format/common"
|
|
)
|
|
|
|
type JsonDriver struct {
|
|
}
|
|
|
|
func (d *JsonDriver) Prepare() error {
|
|
common.HashFlag()
|
|
common.SelectorFlag()
|
|
return nil
|
|
}
|
|
|
|
func (d *JsonDriver) Init(context.Context) error {
|
|
err := common.ManualHashInit()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return common.ManualSelectorInit()
|
|
}
|
|
|
|
func (d *JsonDriver) Format(data interface{}) ([]byte, []byte, error) {
|
|
msg, ok := data.(proto.Message)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("message is not protobuf")
|
|
}
|
|
|
|
key := common.HashProtoLocal(msg)
|
|
return []byte(key), []byte(common.FormatMessageReflectJSON(msg, "")), nil
|
|
}
|
|
|
|
func init() {
|
|
d := &JsonDriver{}
|
|
format.RegisterFormatDriver("json", d)
|
|
}
|