func (self *ProtobufClient) sendResponse(response *protocol.Response) {
self.requestBufferLock.RLock()
req, ok := self.requestBuffer[*response.RequestId]
self.requestBufferLock.RUnlock()
if !ok {
return
}
deleteRequest := false
switch rt := response.GetType(); rt {
case protocol.Response_END_STREAM,
protocol.Response_HEARTBEAT,
protocol.Response_ERROR:
deleteRequest = true
case protocol.Response_QUERY:
// do nothing
default:
panic(fmt.Errorf("Unknown response type: %s", rt))
}
self.requestBufferLock.Lock()
req, ok = self.requestBuffer[*response.RequestId]
if deleteRequest {
delete(self.requestBuffer, *response.RequestId)
}
self.requestBufferLock.Unlock()
if !ok {
return
}
log.Debug("ProtobufClient yielding to %s %s", req.r.Name(), response)
req.r.Yield(response)
}
