这个框架的目标是作为一个通用的框架,我希望它大而全,在日后可以直接作为我其他项目的基础模板,所以我还想继续给他添加一些功能,就当写一些demo进去吧。这篇文章,我将会添加一个队列的功能。
成都创新互联公司致力于互联网品牌建设与网络营销,包括成都网站建设、网站设计、SEO优化、网络推广、整站优化营销策划推广、电子商务、移动互联网营销等。成都创新互联公司为不同类型的客户提供良好的互联网应用定制及解决方案,成都创新互联公司核心团队十年专注互联网开发,积累了丰富的网站经验,为广大企业客户提供一站式企业网站建设服务,在网站建设行业内树立了良好口碑。
队列有很多种,我选择nsq。使用nsq需要知道以下几个概念:
了解了一些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)
}
}