nsq源码阅读 nsqlookupd源码三 tcp.go tcp_server.go

本文详细解析了NSQ中TCP服务的实现原理及处理流程,包括如何通过TCP监听客户端连接请求,以及如何处理客户端发送的协议版本信息。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

nsqlookupd.go中对TCP的处理代码为

protocol.TCPServer(tcpListener, tcpServer, l.opts.Logger)
用户监听处理nsqd发送的信息

TCPServer在internal/protocol/tcp_server.go中定义:

package protocol

import (
	"fmt"
	"net"
	"runtime"
	"strings"

	"github.com/nsqio/nsq/internal/app"
)

type TCPHandler interface {
	Handle(net.Conn)
}

//启动TCP服务,接受客户端连接,并注册处理方法
func TCPServer(listener net.Listener, handler TCPHandler, l app.Logger) {
	l.Output(2, fmt.Sprintf("TCP: listening on %s", listener.Addr()))

	for {
		clientConn, err := listener.Accept()
		//针对不同的错误级别,采用不同的处理方式
		if err != nil {
			if nerr, ok := err.(net.Error); ok && nerr.Temporary() {
				l.Output(2, fmt.Sprintf("NOTICE: temporary Accept() failure - %s", err))
				runtime.Gosched()
				continue
			}
			// there's no direct way to detect this error because it is not exposed
			if !strings.Contains(err.Error(), "use of closed network connection") {
				l.Output(2, fmt.Sprintf("ERROR: listener.Accept() - %s", err))
			}
			break
		}
		//golang接口的应用
		go handler.Handle(clientConn)
	}

	l.Output(2, fmt.Sprintf("TCP: closing %s", listener.Addr()))
}

protocol.tcpServer()的第二个参数,是一个tcpServer实例,在文件nsqlookupd/tcp.go中定义,改实例实现了internal/protocol/tcp_server.go中的TCPHandler接口,

tcp.go:

package nsqlookupd

import (
	"io"
	"net"

	"github.com/nsqio/nsq/internal/protocol"
)

type tcpServer struct {
	ctx *Context
}

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...
	//客户端初始化自己时,需要发送一个4字节序列数据指明它想使用的协议版本
	buf := make([]byte, 4)
	//读取协议内容
	//函数原型
	// func ReadFull(r Reader, buf []byte) (n int, err error)
	//把对象 r 中的数据读出来,然后存入一个缓冲区 buf 中
	_, err := io.ReadFull(clientConn, buf)
	if err != nil {
		p.ctx.nsqlookupd.logf("ERROR: failed to read protocol version - %s", err)
		return
	}
	//将读取到的字节流转换成string
	protocolMagic := string(buf)

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

	//internal/protocol/protocol.go中定义了Protocol接口
	//接口中方法原型 IOLoop(conn net.Conn) error
	var prot protocol.Protocol
	switch protocolMagic {
	case "  V1":
		//当前只支持"  V1"协议(前面两个空格,所以共4个字节)
		//协议在nsqlookupd/lookup_protocol_v1.go中定义
		//创建LookupProtocolV1实例
		//LookupProtocolV1实现了Protocol接口
		prot = &LookupProtocolV1{ctx: p.ctx}
	default:
		//如果不是"  V1"协议,则断开链接,打印错误信息,返回
		protocol.SendResponse(clientConn, []byte("E_BAD_PROTOCOL"))
		clientConn.Close()
		p.ctx.nsqlookupd.logf("ERROR: client(%s) bad protocol magic '%s'",
			clientConn.RemoteAddr(), protocolMagic)
		return
	}

	//如果是"  V1"协议,则调用LookupProtocolV1的IOLoop方法
	err = prot.IOLoop(clientConn)
	if err != nil {
		p.ctx.nsqlookupd.logf("ERROR: client(%s) - %s", clientConn.RemoteAddr(), err)
		return
	}
}










评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值