fix #1129 / memorize withRoutes in renderRoutes.js

umijs · Oct 12, 2018 · d466112 · d466112
1 parent 5d07ae1
commit d466112
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/packages/umi/src/renderRoutes.js b/packages/umi/src/renderRoutes.js
@@ -1,7 +1,26 @@
 import React from 'react';
 import { Switch, Route, Redirect } from 'react-router-dom';
 
+const RouteInstanceMap = {
+  remove(key) {
+    key._routeInternalComponent = undefined;
+  },
+  get(key) {
+    return key._routeInternalComponent;
+  },
+  has(key) {
+    return key._routeInternalComponent !== undefined;
+  },
+  set(key, value) {
+    key._routeInternalComponent = value;
+  },
+};
+
 function withRoutes(route) {
+  if (RouteInstanceMap.has(route)) {
+    return RouteInstanceMap.get(route);
+  }
+
   const Routes = route.Routes;
   let len = Routes.length - 1;
   let Component = args => {
@@ -19,7 +38,7 @@ function withRoutes(route) {
     len -= 1;
   }
 
-  return args => {
+  const ret = args => {
     const { render, ...rest } = args;
     return (
       <Route
@@ -30,6 +49,8 @@ function withRoutes(route) {
       />
     );
   };
+  RouteInstanceMap.set(route, ret);
+  return ret;
 }
 
 export default function renderRoutes(