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 }