怎么搭建golang框架

這個框架的目標是作為一個通用的框架，我希望它大而全，在日后可以直接作為我其他項目的基礎模板，所以我還想繼續給他添加一些功能，就當寫一些demo進去吧。這篇文章，我將會添加一個隊列的功能。

nsq

隊列有很多種，我選擇nsq。使用nsq需要知道以下幾個概念：

1. nsqd：負責維護隊列的組件，接受消息排隊和投遞；
2. nsqlookupd：管理nsq集群的組件；
3. nsqadmin：nsq的web管理組件；
4. topic：消息的集合。產生消息，需要指定該消息屬于哪個topic的；
5. channel：隊列消息的副本。nsq消費者與nsqd或者nsqlookupd建立一個channel，并監聽channel中的消息，從而實現消費。

了解了一些nsq的基礎概念之后，我們先把nsq環境搭建起來，這里我們還是用docker-compose好了，以下是nsq的docker-compose.yaml。

version: '3'
services:
  nsqlookupd:
    image: nsqio/nsq
    command: /nsqlookupd
    networks:
      - nsq-network
    hostname: nsqlookupd
    ports:
      - "4161:4161"
      - "4160:4160"
  nsqd:
    image: nsqio/nsq
    command: /nsqd --lookupd-tcp-address=nsqlookupd:4160 --broadcast-address=nsqd
    depends_on:
      - nsqlookupd
    hostname: nsqd
    networks:
      - nsq-network
    ports:
      - "4151:4151"
      - "4150:4150"
  nsqadmin:
    image: nsqio/nsq
    command: /nsqadmin --lookupd-http-address=nsqlookupd:4161
    depends_on:
      - nsqlookupd
    hostname: nsqadmin
    ports:
      - "4171:4171"
    networks:
      - nsq-network

networks:
  nsq-network:
    driver: bridge

啟動了nsq服務之后，我們簡單地驗證一下，發布一條消息到nsq中：

curl -d 'hello awesome' 'http://127.0.0.1:4151/pub?topic=awesome'

瀏覽器訪問：http://localhost:4171/，我們也可以看到有相應的topic和產生

接著我們修改一下代碼，老規矩，先改配置：

"nsq_config":
  "topic": "awesome"
  "channel": "ch"
  "nsqlookupd_addr": "127.0.0.1:4161"

type NsqConfig struct {
    Topic          string `yaml:"topic"`
    Channel        string `yaml:"channel"`
    NsqLookupdAddr string `yaml:"nsqlookupd_addr"`
}

然后在mq.go中添加一個啟動mq消費者的函數：

func StartMqServer() {
    conf := nsq.NewConfig()
    q, _ := nsq.NewConsumer(config.GetConfig().NsqConfig.Topic, config.GetConfig().NsqConfig.Channel, conf)
    q.AddHandler(nsq.HandlerFunc(func(message *nsq.Message) error {
        //do something when you receive a message
        logger.GetLogger().Info("receive", zap.String(config.GetConfig().NsqConfig.Topic, string(message.Body)))
        return nil
    }))
    err := q.ConnectToNSQLookupd(config.GetConfig().NsqConfig.NsqLookupdAddr)
    if err != nil {
        logger.GetLogger().Error("connect to nsqlookupd failed.", zap.Error(err))
        os.Exit(-1)
    }
}

同樣在入口文件，啟動它：

go mq.StartMqServer()

最后，我們寫個生產者，生成一條消息，測試一下：

func TestProducer(t *testing.T)  {
    config := nsq.NewConfig()
    p, err := nsq.NewProducer("127.0.0.1:4150", config)

    if err != nil {
        log.Panic(err)
    }

    err = p.Publish("awesome", []byte("hello awesome"))
    if err != nil {
        log.Panic(err)
    }
}