func (p *tcpServer) Handle(conn net.Conn) {
buf := make([]byte, 4)
_, err := io.ReadFull(conn, buf)
if err != nil {
p.ctx.s.logf("error: failed to read protocol - %s", err)
return
}
protoMagic := string(buf)
p.ctx.s.logf("client(%s) send protocol: %s", conn.RemoteAddr(), protoMagic)
var protocol proto.Protocol
switch protoMagic {
case " V1":
protocol = &LookupProtocolV1{ctx: p.ctx}
default:
proto.SendResponse(conn, []byte("E_BAD_PROTOCOL"))
p.ctx.s.logf("client(%s) wrong protocol: %s", conn.RemoteAddr(), protoMagic)
conn.Close()
return
}
err = protocol.IOLoop(conn)
if err != nil {
p.ctx.s.logf("client(%s) error - %s", conn.RemoteAddr(), err)
return
}
}
func (p *LookupProtocolV1) IOLoop(conn net.Conn) error {
var err error
//var line string
err = nil
client := NewClientV1(conn)
reader := bufio.NewReader(conn)
for {
line, err := reader.ReadString('\n')
if err != nil {
break
}
line = strings.TrimSpace(line)
params := strings.Split(line, " ")
resp, err := p.Exec(client, reader, params)
if err != nil {
p.ctx.s.logf("client(%s) exec error- %s", client, err)
break
}
if resp != nil {
_, err := proto.SendResponse(client, resp)
if err != nil {
p.ctx.s.logf("send response to client(%) error - %s", client, err)
break
}
}
}
p.ctx.s.logf("closing client(%s)", client)
//client may download client or server
if client.peerInfo != nil {
p.ctx.s.Hold.RemoveProducer(client.peerInfo)
}
client.Close()
return err
}