real-mj-song commented on code in PR #14917:
URL: https://github.com/apache/pinot/pull/14917#discussion_r1931189114


##########
pinot-broker/src/main/java/org/apache/pinot/broker/requesthandler/MultiStageBrokerRequestHandler.java:
##########
@@ -269,11 +271,17 @@ protected BrokerResponse handleRequest(long requestId, 
String query, SqlNodeAndO
         
requestContext.setErrorCode(QueryException.EXECUTION_TIMEOUT_ERROR_CODE);
         return new 
BrokerResponseNative(QueryException.EXECUTION_TIMEOUT_ERROR);
       } catch (Throwable t) {
+        ProcessingException queryException = 
QueryException.QUERY_EXECUTION_ERROR;
+        if (t instanceof BadQueryRequestException) {
+          // provide more specific error code if available
+          queryException = QueryException.QUERY_VALIDATION_ERROR;

Review Comment:
   For `SingleStageBrokerRequestHandler`, this is not needed because 
`BadQueryRequestException` thrown by `ServerQueryExecutorV1Impl` is stored in 
BrokerResponseNative object. It's directly parsed in 
PinotClientRequest::getPinotQueryResponse.
   
   For this `MultiStageBrokerRequestHandler`, this was needed because query was 
handled by `_queryDispatcher.submitAndReduce`.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org
For additional commands, e-mail: commits-h...@pinot.apache.org

Reply via email to