We can hack Join operator not just logical relation operator.

[zhexuany]

tispark/core/src/main/scala/org/apache/spark/sql/TiStrategy.scala

Line 56 in e32fa0e

case class TiStrategy(getOrCreateTiContext: SparkSession => TiContext)(sparkSession: SparkSession)

Too many hacks here since we hack the planning but we don't have full control over the planning stage. We cannot pass context around during planning so a re-extract needed for push-down since a plan tree might contain Join which causes a single tree has multiple plans to push-down.

tispark/core/src/main/scala/org/apache/spark/sql/TiStrategy.scala

Line 114 in e32fa0e

override def apply(plan: LogicalPlan): Seq[SparkPlan] = {

Currently, we only hack the logical relation. We can do the same thing on the Join operator.