怎么搭建golang框架

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

nsq

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

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

了解了一些nsq的基礎(chǔ)概念之后，我們先把nsq環(huán)境搭建起來，這里我們還是用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服務(wù)之后，我們簡單地驗證一下，發(fā)布一條消息到nsq中：

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

瀏覽器訪問：http://localhost:4171/，我們也可以看到有相應(yīng)的topic和產(chǎn)生

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

"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消費者的函數(shù)：

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()

最后，我們寫個生產(chǎn)者，生成一條消息，測試一下：

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)
    }
}