1，原生 Switch 的渲染内容

对如下代码来说：

import { BrowserRouter as Router, Route, Switch } from "react-router-dom";
function News() {
    return <div className="page news">News</div>;
}

function Goods() {
    return <div className="page goods">Goods</div>;
}

export default function App() {
    return (
        <Router>
            <Switch>
                <Route path="/page1" component={News}></Route>
                <Route path="/page2" component={Goods}></Route>
            </Switch>
        </Router>
    );
}

React 插件展示的内容：

可以看到，除了也使用了 Router 的上下文之外，只加载了一个 Route 组件。

2，实现

经测试，Switch 的子元素有如下规则：

• 如果不是 Route 组件，则会报错。
• 如果只有一个 Route 组件，则得到的 props.children 的类型是对象。
• 如果有多个 Route 组件，则得到的 props.children 的类型是数组。

所以，除了做以上特殊的判断外，再加上渲染第一个匹配到的组件的逻辑即可。

import React, { Component } from "react";
import ctx from "./RouterContext";
import { Route } from "./Route";
import matchPath from "./matchPath";

export class Switch extends Component {
    getChildren = ({ location }) => {
        let children = [];
        if (Array.isArray(this.props.children)) {
            children = this.props.children;
        } else if (typeof this.props.children === "object") {
            children = [this.props.children];
        }
        for (const child of children) {
            if (child.type !== Route) {
                throw new TypeError("子元素非 Route 组件");
            }
            const { path = "/", exact = false, strict = false, sensitive = false } = child.props;
            const result = matchPath(path, location.pathname, {
                exact,
                strict,
                sensitive,
            });
            if (result) {
                return child;
            }
        }
        return null;
    };
    render() {
        return <ctx.Consumer>{this.getChildren}</ctx.Consumer>;
    }
}

注意到，判断是否是 Route 组件，可通过引入的 Route 组件直接进行判断。

---

以上。