unistack-org/micro-broker-stan
4
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 5 Wiki Activity

couple of fixes

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2021-09-10 23:44:15 +03:00
parent ba37142442
commit e1ff56f784
1 changed files with 32 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
stan.go
View File

@ -97,12 +97,7 @@ func (n *subscriber) Close() error {
} }
func (n *stanBroker) Address() string { func (n *stanBroker) Address() string {
// stan does not support connected server info return strings.Join(n.addrs, ",")
if len(n.addrs) > 0 {
return n.addrs[0]
}
return ""
} }
func setAddrs(addrs []string) []string { func setAddrs(addrs []string) []string {
@ -299,22 +294,28 @@ func (n *stanBroker) Options() broker.Options {
} }
func (n *stanBroker) BatchPublish(ctx context.Context, msg []*broker.Message, opts ...broker.PublishOption) error { func (n *stanBroker) BatchPublish(ctx context.Context, msg []*broker.Message, opts ...broker.PublishOption) error {
msgs := make(map[string][][]byte) var err error
var buf []byte
var wg sync.WaitGroup var wg sync.WaitGroup
msgs := make(map[string][][]byte)
options := broker.NewPublishOptions(opts...)
wg.Add(len(msg)) wg.Add(len(msg))
for _, m := range msg { for _, m := range msg {
b, err := n.opts.Codec.Marshal(m) if options.BodyOnly {
buf = m.Body
} else {
buf, err = n.opts.Codec.Marshal(m)
if err != nil { if err != nil {
return err return err
} }
topic, _ := m.Header.Get(metadata.HeaderTopic)
msgs[topic] = append(msgs[topic], b)
} }
topic, _ := m.Header.Get(metadata.HeaderTopic)
n.RLock() msgs[topic] = append(msgs[topic], buf)
defer n.RUnlock() }
var ackErr error var ackErr error
@ -325,6 +326,9 @@ func (n *stanBroker) BatchPublish(ctx context.Context, msg []*broker.Message, op
} }
} }
n.RLock()
defer n.RUnlock()
for topic, ms := range msgs { for topic, ms := range msgs {
for _, m := range ms { for _, m := range ms {
if _, err := n.conn.PublishAsync(topic, m, ackHandler); err != nil { if _, err := n.conn.PublishAsync(topic, m, ackHandler); err != nil {
@ -339,13 +343,23 @@ func (n *stanBroker) BatchPublish(ctx context.Context, msg []*broker.Message, op
} }
func (n *stanBroker) Publish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { func (n *stanBroker) Publish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error {
b, err := n.opts.Codec.Marshal(msg) var buf []byte
var err error
options := broker.NewPublishOptions(opts...)
if options.BodyOnly {
buf = msg.Body
} else {
buf, err = n.opts.Codec.Marshal(msg)
if err != nil { if err != nil {
return err return err
} }
}
n.RLock() n.RLock()
defer n.RUnlock() defer n.RUnlock()
return n.conn.Publish(topic, b) return n.conn.Publish(topic, buf)
} }
func (n *stanBroker) BatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { func (n *stanBroker) BatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) {