NSQ系列之nsqlookupd代码分析二（初识nsqlookupd tcpServer）

NSQ系列之nsqlookupd代码分析二（初识nsqlookupd tcpServer）

nsqlookupd

nsqlookupd

tcpServer

httpServer

tcpServer

type tcpServer struct {
ctx *Context //上一章中提到的Contexto
}

func (p *tcpServer) Handle(clientConn net.Conn) {
p.ctx.nsqlookupd.logf("TCP: new client(%s)", clientConn.RemoteAddr())

// The client should initialize itself by sending a 4 byte sequence indicating
// the version of the protocol that it intends to communicate, this will allow us
// to gracefully upgrade the protocol away from text/line oriented to whatever...

//这里有注释了，但是我英文不好，所以还是加上自己的注释吧，如果有错误，欢迎大家指正哦
buf := make([]byte, 4)  //初始化4个字节的buf,用来获取协议中的版本号
_, err := io.ReadFull(clientConn, buf)
if err != nil {
p.ctx.nsqlookupd.logf("ERROR: failed to read protocol version - %s", err)
return
}
protocolMagic := string(buf)

p.ctx.nsqlookupd.logf("CLIENT(%s): desired protocol magic '%s'",
clientConn.RemoteAddr(), protocolMagic)

var prot protocol.Protocol
switch protocolMagic {
case "  V1":  //如果是  V1，注意4个字节哦，前面有两个空格字符
prot = &LookupProtocolV1{ctx: p.ctx} //初始化一个lookupProtocolV1的指针，将Context 组合进去 具体参考nsq/nsqlookupd/lookup_protocol_v1.go文件里的代码
default:
//如果不是  V1 则关闭连接，返回E_BAD_PROTOCOL
protocol.SendResponse(clientConn, []byte("E_BAD_PROTOCOL"))
clientConn.Close()
p.ctx.nsqlookupd.logf("ERROR: client(%s) bad protocol magic '%s'",
clientConn.RemoteAddr(), protocolMagic)
return
}
//到这里才是整个tcpServer中的重头戏，不过今天就讲到这里了。下一章就重点分析下这个IOLoop
err = prot.IOLoop(clientConn)
if err != nil {
p.ctx.nsqlookupd.logf("ERROR: client(%s) - %s", clientConn.RemoteAddr(), err)
return
}
}

tcpServer

IOLoop