micro/broker/broker.go

70 lines
1.3 KiB
Go
Raw Normal View History

package broker
import (
"code.google.com/p/go-uuid/uuid"
2015-05-24 01:16:26 +03:00
"golang.org/x/net/context"
)
type Broker interface {
Address() string
Connect() error
Disconnect() error
Init() error
2015-05-24 01:16:26 +03:00
Publish(context.Context, string, []byte) error
Subscribe(string, func(context.Context, *Message)) (Subscriber, error)
}
type Message struct {
Id string
Timestamp int64
Topic string
2015-05-24 01:16:26 +03:00
Body []byte
}
type Subscriber interface {
Topic() string
Unsubscribe() error
}
type options struct{}
type Option func(*options)
var (
Address string
Id string
DefaultBroker Broker
)
func NewBroker(addrs []string, opt ...Option) Broker {
return newHttpBroker([]string{Address}, opt...)
}
func Init() error {
if len(Id) == 0 {
Id = "broker-" + uuid.NewUUID().String()
}
if DefaultBroker == nil {
DefaultBroker = newHttpBroker([]string{Address})
}
return DefaultBroker.Init()
}
func Connect() error {
return DefaultBroker.Connect()
}
func Disconnect() error {
return DefaultBroker.Disconnect()
}
2015-05-24 01:16:26 +03:00
func Publish(ctx context.Context, topic string, body []byte) error {
return DefaultBroker.Publish(ctx, topic, body)
}
2015-05-24 01:16:26 +03:00
func Subscribe(topic string, function func(context.Context, *Message)) (Subscriber, error) {
return DefaultBroker.Subscribe(topic, function)
}