traverse - Support traverseConfig

This commit is contained in:
2019-09-11 16:54:32 +05:30
parent 8326ec5bf7
commit e7e969cf65
+6 -2
View File
@@ -136,8 +136,12 @@ function* traverseGenerator(
* ```
*
*/
function traverse(node: FiberNode, fn: (node: FiberNode) => any) {
const nodeIterator = traverseGenerator(node);
function traverse(
node: FiberNode,
fn: (node: FiberNode) => any,
traverseConfig?: TTraverseConfig
) {
const nodeIterator = traverseGenerator(node, traverseConfig);
for (const tmpNode of nodeIterator) {
fn.call(null, tmpNode);
}