NSQLookupd Main方法分析
上一篇
http://my.oschina.net/astute/blog/296955
已经分析了 nsqlookupd 启动时的命令行解析，最终构造了 NSQLookupd 结构体，这是一个很重要的结构体。
type NSQLookupd struct {
options *nsqlookupdOptions // 命令行参数
tcpAddr *net.TCPAddr // tcp 端口
httpAddr *net.TCPAddr // http 端口
tcpListener net.Listener
httpListener net.Listener
waitGroup util.WaitGroupWrapper
DB *RegistrationDB
}
这篇文章分析下 daemon.Main() 执行过程
1、创建 context，内部只有一个指向NSQLookupd的指针
context := &Context{l}
2、创建 listener；Listen 方法支持面向流的监听，包括 unix domain stream socket。
tcpListener, err := net.Listen("tcp", l.tcpAddr.String())
3、创建 tcpServer struct，内部只含有一个 context 指针，为什么需要对 NSQLookupd
tcpServer 实现了 Handle(net.Conn) 方法，表明实现了 TCPHandler 接口
tcpServer := &tcpServer{context: context}
4、分析 l.waitGroup.Wrap(func() { util.TCPServer(tcpListener, tcpServer) })
waitGroup 是结构体 util.WaitGroupWrapper 实例，此结构体继承自 sync.WaitGroup；WaitGroup 等待一组协程结束。主协程调用 Add 方法来设置等待协程的数目；func() { util.TCPServer(tcpListener, tcpServer) } 这是一个匿名函数。
总结：主协程调用 Wrap 方法，调用 Add，设置等待的协程数目为1，启动新的协程去调用 匿名函数，主协程最后会在 exitChan 通道上等待消息，如果收到中断信号，关闭 tcpListener, httpListner，最后在 waitGroup 上等待协程结束。需要 waitGroup 的目的，就是为了 主协程 和 监听协程之间的状态同步。
5、监听协程执行的逻辑如下，简化版本，忽略日志和错误处理。监听协程循环 Accept，然后分配新的协程调用 handler 处理接收的 clientConn。处理客户端连接的协程，我把它叫做 工作协程。
func TCPServer(listener net.Listener, handler TCPHandler) {
for {
clientConn, err := listener.Accept()
go handler.Handle(clientConn)
}
}
6、工作协程的处理逻辑，最终调用的是 tcpServer 的 Handle 方法，这是最重要的一个方法，重点分析
func (p *tcpServer) Handle(clientConn net.Conn)
buf := make([]byte, 4)
_, err := io.ReadFull(clientConn, buf)
读取客户端发送过来的前4个字节，里面含有协议的版本号。看到了没有，完全阻塞式编程，没有屎一样的事件驱动！
判断版本号后，最终调用 LookupProtocolV1 的 IOLoop 方法
7、方法分析 func (p *LookupProtocolV1) IOLoop(conn net.Conn) error
client := NewClientV1(conn) // ClientV1 struct 继承 net.Conn，封装一个 PeerInfo
reader := bufio.NewReader(client) // 创建一个带缓冲的 reader
for { // 此部分代码有省略
line, err = reader.ReadString('\n’) // 按行来处理
line = strings.TrimSpace(line)
params := strings.Split(line, " ")
response, err := p.Exec(client, reader, params) // 执行命令
if response != nil {
_, err = util.SendResponse(client, response) // 发送 response
}
}
IO 出错，或者执行命令出错，会退出循环。
8、命令分派
func (p *LookupProtocolV1) Exec(client *ClientV1, reader *bufio.Reader, params []string) ([]byte, error)
switch params[0] {
case "PING":
return p.PING(client, params)
// .........
}
根据第一个参数做命令区分，调用响应的指令
9、命令执行，以 ping 命令为例，看看命令如何执行
func (p *LookupProtocolV1) PING(client *ClientV1, params []string) ([]byte, error)
atomic.StoreInt64(&client.peerInfo.lastUpdate, now.UnixNano()) // 更新客户端的状态
return []byte("OK"), nil // 返回 “OK”，经由第 7 步的 util.SendResponse 发送给 client
至此 tcpServer 的处理流程就介绍完了。
10、和 2，3 步对应的，还会创建 httpListener, httpServer, 主要负责 topic，channel 相关的操作
httpListener, err := net.Listen("tcp", l.httpAddr.String())
httpServer := &httpServer{context: context}
处理 httpServer 的协程， 会调用 func() { util.HTTPServer(httpListener, httpServer, "HTTP") }
11、分析 util.HTTPServer 函数
关键是如下两句
server := &http.Server{
Handler: handler,
}
err := server.Serve(listener)
默认的 http handler 就是 httpServer 本身
12、分析 httpServer 的 ServeHTTP 方法
err := s.v1Router(w, req) // 路由请求
13、分析 v1Router
switch req.URL.Path {
case "/ping”:
//......
}
按照请求路径分派
14、以 "/topics” 为例，看如何处理请求
执行函数 util.NegotiateAPIResponseWrapper(w, req,
func() (interface{}, error) { return s.doTopics(req) })
内部调用传入的匿名函数，也就是执行 s.doTopics(req)，会把所有的 topic 对应的 producer 信息组成 map[string]interface{} 返回给客户端