func (p *tcpServer) Handle(clientConn net.Conn) {
// 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...
clientConn.SetReadDeadline(time.Now().Add(p.ctx.getOpts().ClientTimeout))
buf := make([]byte, 4)
_, err := io.ReadFull(clientConn, buf)
if err != nil {
nsqd.NsqLogger().Logf(" failed to read protocol version - %s from client: %v", err, clientConn.RemoteAddr())
clientConn.Close()
return
}
protocolMagic := string(buf)
if nsqd.NsqLogger().Level() >= levellogger.LOG_DEBUG {
nsqd.NsqLogger().LogDebugf("new CLIENT(%s): desired protocol magic '%s'",
clientConn.RemoteAddr(), protocolMagic)
}
var prot protocol.Protocol
switch protocolMagic {
case " V2":
prot = &protocolV2{ctx: p.ctx}
default:
protocol.SendFramedResponse(clientConn, frameTypeError, []byte("E_BAD_PROTOCOL"))
clientConn.Close()
nsqd.NsqLogger().LogErrorf("client(%s) bad protocol magic '%s'",
clientConn.RemoteAddr(), protocolMagic)
return
}
err = prot.IOLoop(clientConn)
if err != nil {
nsqd.NsqLogger().Logf("client(%s) error - %s", clientConn.RemoteAddr(), err)
return
}
}
func internalSend(client *nsqd.ClientV2, frameType int32, data []byte, needFlush bool) error {
client.LockWrite()
defer client.UnlockWrite()
if client.Writer == nil {
return errors.New("client closed")
}
var zeroTime time.Time
if client.HeartbeatInterval > 0 {
client.SetWriteDeadline(time.Now().Add(client.HeartbeatInterval))
} else {
client.SetWriteDeadline(zeroTime)
}
_, err := protocol.SendFramedResponse(client.Writer, frameType, data)
if err != nil {
return err
}
if needFlush || frameType != frameTypeMessage {
err = client.Flush()
}
return err
}
