nsqd的架构及源码分析

原创

wx63993a9e4baf6 2023-08-08 01:43:29 ©著作权

文章标签 nsqd源码 nsq源码 sql Group 客户端 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者wx63993a9e4baf6的原创作品，请联系作者获取转载授权，否则将追究法律责任

文章目录

一 nsq的整体代码结构

二回顾nsq的整体架构图

三 nsqd进程的作用

四 nsqd启动流程的源码分析

五本篇博客总结

在博客 nsq整体架构及各个部件作用详解_YZF_Kevin的博客-博客中我们讲了nsq的整体框架，各个部件的大致作用。如果没看过的，建议大家去学习下，不然理解后续的内容会有难度

这篇博客开始我们来看下每个部件的详细功能，从源码入手分析其内部实现原理

一 nsq的整体代码结构

建议大家也下载nsq的代码，一边看博客一边看代码印象更深刻。nsq的官方git代码地址：GitHub - nsqio/nsq: A realtime distributed messaging platform

nsq代码结构如下，图中有注释，大家先有个整体印象，知道各个模块的代码在哪就行

nsqd的架构及源码分析_nsqd源码

二回顾nsq的整体架构图

nsqd的架构及源码分析_nsqd源码_02

图中最上面的四个节点就是nsqd进程，至少要有1个，可以多开。我们画了4个，分别是nsq1，nsq2，nsq3，nsq4

注意看nsqd的连接关系，每个nsqd节点和所有客户端都有连接(tcp+http)，且每个nsqd节点和所有的nsqlookupd节点也有连接(tcp)

三 nsqd进程的作用

1. topic的创建，清空，暂停，重新激活，删除，持久化(保存到文件，从文件加载)，同步给nsqlookupd进程

2. channel的创建，清空，暂停，重新激活，删除，持久化(保存到文件，从文件加载)，同步给nsqlookupd进程

3. message的监听，中转，持久化(保存到文件，从文件加载)，主动推送消息给各个客户端，超时重发，消息计数

4. 配置修改，运行状态（协程、内存）统计

5. 抽检channel的延迟队列，飞行队列，消息超时的重新入队

可以说，nsqd进程是整个nsq平台的核心，消息队列架构简单的话，只有一个nsqd进程就够了。

四 nsqd启动流程的源码分析

nsqd的代码主要在两块

1. 代码框架及main函数，目录在 nsq/apps/nsqd/*

2. 实现代码，目录在 nsq/nsqd/*

值得一提的是nsqd，nsqlookupd，nsqadmin这三个进程的框架都使用了go-svc包，这个包很简单，使用者只需实现它的三个函数即可

Init() 配置，初始化等操作

Start() 真正启动

Stop() 结束时的关闭操作

好了，我们看nsqd的入口，也就是main函数，代码在nsq/apps/nsqd/main.go，代码如下(已加注释)

type program struct {
	once 		sync.Once
	nsqd 		*nsqd.NSQD	// nsqd对象
}

// nsqd的启动入口
func main() {
	prg := &program{}
	// Run内部会调用Init()，Start()，监听到这两个系统信号时会调用Stop()
	if err := svc.Run(prg, syscall.SIGINT, syscall.SIGTERM); err != nil {
		logFatal("%s", err)
	}
}

main()函数内部只有一个对象program，也只有一处调用svc.Run()，这个函数内部会调用program.Init()和program.Start()

其中program.Init()函数，主要是创建并检测nsqd的配置，然后根据配置创建出一个nsqd实例

重点在program.Start()函数，代码如下(已加注释)

// nsqd的启动，重点在调用的Main()函数
func (p *program) Start() error {
	// 加载元数据，并创建初始化出所有的topic对象，所有的channel对象
	err := p.nsqd.LoadMetadata()
	if err != nil {
		logFatal("failed to load metadata - %s", err)
	}
	// 再持久化元数据到文件（不要觉得奇怪，因为上面的LoadMetadata()函数可能会过滤掉一些无效的topic，channel，这里再重写算是刷新了元数据）
	err = p.nsqd.PersistMetadata()
	if err != nil {
		logFatal("failed to persist metadata - %s", err)
	}
	// 启动一个新协程，专门运行nsqd的Main()循环，注意这个Main()是永不退出的(除非出错)
	go func() {
		err := p.nsqd.Main()
		if err != nil {
			p.Stop()
			os.Exit(1)
		}
	}()
	return nil
}

对上面的代码解释下，program.Start()函数一共干了3件事

1. nsqd.LoadMetadata(), 这个函数根据配置加载旧nsqd元数据。这些元数据包含版本号，topic，channel，过滤掉不合法的topic和channel，合法的topic和channel都创建出对象，并且为每个topic建立处理循环

2. nsqd.PersistMetadata(), 把过滤后的topic和channel再保存到文件nsqd.dat，算是把旧数据过滤了一遍

3. 新启动一个协程，调用nsqd.Main()，这个Main()是nsqd的核心，启动了nsqd的全部服务。除非遇到错误，否则永不退出

接下来看nsqd.Main()的内部实现，代码在nsq/nsqd/nsqd.go，代码如下(已加注释)

// nsqd主协程（内部启动tcp循环，http循环，https循环, 扫描队列池，和nsqlookupd循环），永不退出，除非严重错误
func (n *NSQD) Main() error {
	exitCh := make(chan error)
	var once sync.Once

	// 退出函数（独立协程运行，一直监听，遇到错误
	exitFunc := func(err error) {
		once.Do(func() {
			if err != nil {
				n.logf(LOG_FATAL, "%s", err)
			}
			exitCh <- err
		})
	}

	// TCP服务，独立协程
	n.waitGroup.Wrap(func() {
		exitFunc(protocol.TCPServer(n.tcpListener, n.tcpServer, n.logf))
	})

	// HTTP服务，独立协程
	if n.httpListener != nil {
		httpServer := newHTTPServer(n, false, n.getOpts().TLSRequired == TLSRequired)
		n.waitGroup.Wrap(func() {
			exitFunc(http_api.Serve(n.httpListener, httpServer, "HTTP", n.logf))
		})
	}

	// HTTPS服务，独立协程
	if n.httpsListener != nil {
		httpsServer := newHTTPServer(n, true, true)
		n.waitGroup.Wrap(func() {
			exitFunc(http_api.Serve(n.httpsListener, httpsServer, "HTTPS", n.logf))
		})
	}

	// 独立协程，抽检扫描各个队列
	n.waitGroup.Wrap(n.queueScanLoop)

	// 独立协程，和nsqlookupd的循环（连接和重连，心跳维持，topic，channel变化通知等）
	n.waitGroup.Wrap(n.lookupLoop)

	if n.getOpts().StatsdAddress != "" {
		n.waitGroup.Wrap(n.statsdLoop)
	}

	err := <-exitCh
	return err
}

对上面的代码解释下，nsqd.Main()主要干了6件事

1. 开一个新协程，启动tcp服务并一直监听，为客户端一共tcp服务。我们的客户端最常用，因为生产消息，中转消息，处理消息都是这里实现的

2. 开一个新协程，启动http服务并一直监听，为客户端提供htttp服务

3. 开一个新协程，启动https服务并一直监听，为客户端提供htttps服务

4. 开一个新协程，建立并维持扫描池，这些扫描协程会扫描所有channel的延迟队列，飞行队列，如果消息超时了就重新入队。很有意思的是，nsqd作者很大方地承认他抄袭了redis的抽检策略，内部实现也确实是类redis操作，这个我们后面再讲，todo

5. 开一个新协程，和nsqlookupd建立循环，主要是连接和重连，心跳维持，实时报告自己的topic和channel变化

6. 开一个新协程，做统计操作，统计topic，channel，消息，内存，GC等等