micro/examples/pubsub/main.go

62 lines
1.1 KiB
Go
Raw Normal View History

2015-04-26 22:45:46 +03:00
package main
import (
"fmt"
"time"
2015-05-23 19:55:08 +03:00
log "github.com/golang/glog"
2015-05-05 21:05:06 +03:00
"github.com/myodc/go-micro/broker"
2015-05-23 19:55:08 +03:00
"github.com/myodc/go-micro/cmd"
2015-05-24 01:16:26 +03:00
c "github.com/myodc/go-micro/context"
"golang.org/x/net/context"
2015-04-26 22:45:46 +03:00
)
var (
topic = "go.micro.topic.foo"
)
func pub() {
tick := time.NewTicker(time.Second)
i := 0
for _ = range tick.C {
2015-05-27 00:39:48 +03:00
ctx := c.WithMetadata(context.Background(), map[string]string{
2015-05-24 01:16:26 +03:00
"id": fmt.Sprintf("%d", i),
})
2015-04-26 22:45:46 +03:00
msg := fmt.Sprintf("%d: %s", i, time.Now().String())
2015-05-24 01:16:26 +03:00
if err := broker.Publish(ctx, topic, []byte(msg)); err != nil {
2015-05-23 19:55:08 +03:00
log.Errorf("[pub] failed: %v", err)
} else {
fmt.Println("[pub] pubbed message:", msg)
}
2015-04-26 22:45:46 +03:00
i++
}
}
func sub() {
2015-05-24 01:16:26 +03:00
_, err := broker.Subscribe(topic, func(ctx context.Context, msg *broker.Message) {
2015-05-27 00:39:48 +03:00
md, _ := c.GetMetadata(ctx)
2015-05-24 01:16:26 +03:00
fmt.Println("[sub] received message:", string(msg.Body), "context", md)
2015-04-26 22:45:46 +03:00
})
if err != nil {
fmt.Println(err)
}
}
func main() {
2015-05-23 19:55:08 +03:00
cmd.Init()
if err := broker.Init(); err != nil {
log.Fatalf("Broker Init error: %v", err)
}
if err := broker.Connect(); err != nil {
log.Fatalf("Broker Connect error: %v", err)
}
2015-04-26 22:45:46 +03:00
go pub()
go sub()
<-time.After(time.Second * 10)
}