1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
| package nsqclient
|
| import (
| "context"
|
| "kingdee-dbapi/config"
| "kingdee-dbapi/logger"
| )
|
| var producerCli Producer
|
| func InitNsqProducer() error {
| var err error
| producerCli, err = NewProducer(config.Options.NsqServer)
| if err != nil {
| logger.Error("创建nsq生产客户端失败, %s", err.Error())
| }
|
| return err
| }
|
| func Produce(topic string, msg []byte) bool {
| if producerCli == nil {
| err := InitNsqProducer()
| if err != nil {
| logger.Error("创建nsq生产客户端失败, %s", err.Error())
| return false
| }
| }
|
| err := producerCli.Publish(topic, msg)
| if err != nil {
| logger.Error("nsp发布消息失败,主题:%s, %s", topic, err.Error())
| }
|
| return err == nil
| }
|
| func InitNsqConsumer(topic, channel string, handle func(data []byte) error) {
| if c, err := NewNsqConsumer(context.Background(), topic, channel); err != nil {
| logger.Error("创建nsq消费客户端失败, %s", err.Error())
| return
| } else {
| c.AddHandler(handle)
| if err := c.Run(config.Options.NsqServer, 1); err != nil {
| logger.Error("运行nsq消费客户端失败, %s", err.Error())
| }
|
| }
| }
|
|
