fix #378 add cache for finding class

1 files changed, 27 insertions, 21 deletions

diff --git a/script/core/diagnostics/undefined-doc-name.lua b/script/core/diagnostics/undefined-doc-name.lua
index 5c1e8fbf..3d6c3cd8 100644
--- a/script/core/diagnostics/undefined-doc-name.lua
+++ b/script/core/diagnostics/undefined-doc-name.lua
@@ -4,27 +4,6 @@ local lang    = require 'language'
 local define  = require 'proto.define'
 local vm      = require 'vm'
 
-local function hasNameOfClassOrAlias(name)
-    local docs = vm.getDocTypes(name)
-    for _, otherDoc in ipairs(docs) do
-        if otherDoc.type == 'doc.class.name'
-        or otherDoc.type == 'doc.alias.name' then
-            return true
-        end
-    end
-    return false
-end
-
-local function hasNameOfGeneric(name, source)
-    if not source.typeGeneric then
-        return false
-    end
-    if not source.typeGeneric[name] then
-        return false
-    end
-    return true
-end
-
 return function (uri, callback)
     local state = files.getAst(uri)
     if not state then
@@ -35,6 +14,33 @@ return function (uri, callback)
         return
     end
 
+    local classCache = {}
+    local function hasNameOfClassOrAlias(name)
+        if classCache[name] ~= nil then
+            return classCache[name]
+        end
+        local docs = vm.getDocTypes(name)
+        for _, otherDoc in ipairs(docs) do
+            if otherDoc.type == 'doc.class.name'
+            or otherDoc.type == 'doc.alias.name' then
+                classCache[name] = true
+                return true
+            end
+        end
+        classCache[name] = false
+        return false
+    end
+
+    local function hasNameOfGeneric(name, source)
+        if not source.typeGeneric then
+            return false
+        end
+        if not source.typeGeneric[name] then
+            return false
+        end
+        return true
+    end
+
     guide.eachSource(state.ast.docs, function (source)
         if  source.type ~= 'doc.extends.name'
         and source.type ~= 'doc.type.name' then