NSQ消息队列的使用

NSQ 是由国外的一个短链服务商bitly使用golang开发的一个消息队列系统，正好使用到了这个东西，在这里简单的记录下。

获取客户端

nsq的golang客户端是官方版本的

go get github.com/nsqio/go-nsq

即可

简单的消费者和生产者使用

该客户端有原始的command函数用于一些基础操作，也有consumer和producer的封装，我这里是直接使用了封装了。

• consumer

消费者比较简单，只要监听队列消息，并处理就可以了，下面是一个简单的例子。

type NSQHandler struct {
}

func (this *NSQHandler) HandleMessage(message *nsq.Message) error {
    log.Println("recv:", string(message.Body))
    return nil
}

func testNSQ() {
    waiter := sync.WaitGroup{}
    waiter.Add(1)

    go func() {
        defer waiter.Done()

        consumer, err := nsq.NewConsumer("test", "ch1", nsq.NewConfig())
        if nil != err {
            log.Println(err)
            return
        }

        consumer.AddHandler(&NSQHandler{})
        err = consumer.ConnectToNSQD("10.100.156.207:4150")
        if nil != err {
            log.Println(err)
            return
        }

        select {}
    }()

    waiter.Wait()
}

创建好consumer后，只需要自己创建一个struct并实现HandleMessage方法即可，当有消息时候，再去处理消息。

需要注意的是，AddHandler的回调是在别的routine中执行的，并且可以添加多个handler用于处理消息，这里可能需要注意下线程的同步问题。

• producer

生产者也和消费者差不多，首先需要创建一个producer

func (this *MsgQueue) Init(addr string) error {
    var err error
    this.addr = addr

    //  try to connect
    cfg := nsq.NewConfig()
    this.producer, err = nsq.NewProducer(addr, cfg)
    if nil != err {
        return err
    }

    //  try to ping
    err = this.producer.Ping()
    if nil != err {
        this.producer.Stop()
        this.producer = nil
        return err
    }

    return nil
}

producer封装了较多的方法，分为同步和异步两种。带Async后缀的，都是异步的。

同步是收到了nsq的回应后再返回的函数，所以可能会堵塞，而异步的操作，则调用方需要传入一个chan用于接收结果，当有结果返回或者是超时的情况下，相应的内容会写到该chan中。

在这里我用了同步的api，毕竟消息队列假如出了什么问题，那么整个服务就不可用了，而且同步改异步也不会太麻烦，以后可以做下修改。

publish的方法也很简单，提供一个topic和数据就行了。