unistack-org/micro-broker-nats
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 8 Wiki Activity

We now support configurable codecs for broker

Browse Source
This commit is contained in:
Asim Aslam 2016-12-06 19:55:09 +00:00 committed by Vasiliy Tolstov
parent 90f40cbad4
commit f8714d4cd9
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nats.go
View File

@ -1,10 +1,10 @@
package nats
import (
"encoding/json"
"strings"
"github.com/micro/go-micro/broker"
"github.com/micro/go-micro/broker/codec/json"
"github.com/micro/go-micro/cmd"
"github.com/nats-io/nats"
)
@ -100,7 +100,7 @@ func (n *nbroker) Options() broker.Options {
}
func (n *nbroker) Publish(topic string, msg *broker.Message, opts ...broker.PublishOption) error {
b, err := json.Marshal(msg)
b, err := n.opts.Codec.Marshal(msg)
if err != nil {
return err
}
@ -118,7 +118,7 @@ func (n *nbroker) Subscribe(topic string, handler broker.Handler, opts ...broker
fn := func(msg *nats.Msg) {
var m *broker.Message
if err := json.Unmarshal(msg.Data, &m); err != nil {
if err := n.opts.Codec.Unmarshal(msg.Data, &m); err != nil {
return
}
handler(&publication{m: m, t: topic})
@ -143,7 +143,11 @@ func (n *nbroker) String() string {
}
func NewBroker(opts ...broker.Option) broker.Broker {
var options broker.Options
options := broker.Options{
// Default codec
Codec: json.NewCodec(),
}
for _, o := range opts {
o(&options)
}