Teach planstate_tree_walker about custom scans.

This logic was missing from ExplainPreScanNode, from which I derived
planstate_tree_walker.  But it shouldn't be missing, especially not
from a generic walker function, so add it.

KaiGai Kohei

diff --git a/src/backend/nodes/nodeFuncs.c b/src/backend/nodes/nodeFuncs.c
index 4a2447409a9e7c84678e14746c65fdf45eedd036..a11cb9ff4d10fe657800299711301f93b7c60bf9 100644 (file)
--- a/src/backend/nodes/nodeFuncs.c
+++ b/src/backend/nodes/nodeFuncs.c
@@ -3428,6 +3428,7 @@ bool
 planstate_tree_walker(PlanState *planstate, bool (*walker) (), void *context)
 {
    Plan       *plan = planstate->plan;
+   ListCell   *lc;
 
    /* initPlan-s */
    if (planstate_walk_subplans(planstate->initPlan, walker, context))
@@ -3484,6 +3485,13 @@ planstate_tree_walker(PlanState *planstate, bool (*walker) (), void *context)
            if (walker(((SubqueryScanState *) planstate)->subplan, context))
                return true;
            break;
+       case T_CustomScan:
+           foreach (lc, ((CustomScanState *) planstate)->custom_ps)
+           {
+               if (walker((PlanState *) lfirst(lc), context))
+                   return true;
+           }
+           break;
        default:
            break;
    }