martin-g commented on PR #3842:
URL: 
https://github.com/apache/datafusion-comet/pull/3842#issuecomment-4165013433

   Right!
   This is why I think a possible solution is to move the logic to 
`CometNativeScanExec#doExecuteColumnar()`. There a `CometExecRDD` is 
instantiated and returned: 
   
https://github.com/apache/datafusion-comet/blob/7878f0d01b0411f1a9442a5205649481df4b1257/spark/src/main/scala/org/apache/spark/sql/comet/CometNativeScanExec.scala#L183
   
   You can make it an anonymous class instance like:
   ```scala
   CometExecRDD(...) {
      override def compute(split: Partition, context: TaskContext): 
Iterator[ColumnarBatch] = 
      {
         val res = super.compute(split, context);
       
         // new logic here
   
         res
      }
   }
   ```


-- 
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: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to