Docserver: Cache the samples.

chrome/common/extensions/docs/server2/api_data_source.py

 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import logging
 
 from data_source import DataSource
 from docs_server_utils import StringIdentity
 from environment import IsPreviewServer
 from file_system import FileNotFoundError
 from future import Future, All
-from jsc_view import JSCView, GetEventByNameFromEvents
+from jsc_view import CreateJSCView, GetEventByNameFromEvents
 from platform_util import GetPlatforms
-from samples_data_source import CreateSamplesView
 
 
 class APIDataSource(DataSource):
   '''This class fetches and loads JSON APIs from the FileSystem passed in with
   |compiled_fs_factory|, so the APIs can be plugged into templates.
   '''
   def __init__(self, server_instance, request):
     file_system = server_instance.host_file_system_provider.GetMaster()
     self._json_cache = server_instance.compiled_fs_factory.ForJson(file_system)
     self._template_cache = server_instance.compiled_fs_factory.ForTemplates(
@@ skipping 19 matching lines @@
       self._event_byname_futures[platform] = Future(
           callback=lambda: GetEventByNameFromEvents(future.Get()))
     return self._event_byname_futures[platform]
 
   def _GetSchemaView(self, platform, api_name):
     object_store_key = '/'.join((platform, api_name))
     api_models = self._platform_bundle.GetAPIModels(platform)
     jsc_view_future = self._view_cache.Get(object_store_key)
     model_future = api_models.GetModel(api_name)
     content_script_apis_future = api_models.GetContentScriptAPIs()
+
+    # Parsing samples on the preview server takes forever, so disable it.
+    if IsPreviewServer():
+      samples_futures = Future(value=[])
+    else:
+      samples_future = (self._platform_bundle.GetSamplesModel(platform)
+          .FilterSamples(api_name))
+
     def resolve():
       jsc_view = jsc_view_future.Get()
       if jsc_view is None:
-        jsc_view = JSCView(
+        jsc_view = CreateJSCView(
             content_script_apis_future.Get(),
             model_future.Get(),
             self._platform_bundle.GetAvailabilityFinder(platform),
             self._json_cache,
             self._template_cache,
             self._platform_bundle.GetFeaturesBundle(platform),
             self._LoadEventByName(platform),
-            platform).ToDict()
+            platform,
+            samples_future.Get(),
+            self._request)
         self._view_cache.Set(object_store_key, jsc_view)
       return jsc_view
     return Future(callback=resolve)
 
-  def _GetImpl(self, platform, api_name):
-    jsc_view_future = self._GetSchemaView(platform, api_name)
-    def resolve():
-      jsc_view = jsc_view_future.Get()
-      # Parsing samples on the preview server takes seconds and doesn't add
-      # anything. Don't do it.
-      if not IsPreviewServer():
-        samples_model = self._platform_bundle.GetSamplesModel(platform)
-        # Creates an object that lazily gets samples.
-        jsc_view['samples'] = type('getter', (object,), {
-          'get': lambda _, platform: CreateSamplesView(
-              samples_model.FilterSamples(jsc_view['name']), self._request)
-          })()
-      return jsc_view
-    return Future(callback=resolve)
-
   def get(self, platform):
     '''Return a getter object so that templates can perform lookups such
     as apis.extensions.runtime.
     '''
     getter = lambda: 0
-    getter.get = lambda api_name: self._GetImpl(platform, api_name).Get()
+    getter.get = lambda api_name: self._GetSchemaView(platform, api_name).Get()
     return getter
 
   def GetRefreshPaths(self):
     tasks = []
     for platform in GetPlatforms():
       tasks += ['%s/%s' % (platform, api)
                 for api in
                     self._platform_bundle.GetAPIModels(platform).GetNames()]
     return tasks
 
   def Refresh(self, path):
     platform, api = path.split('/')
     logging.info('Refreshing %s/%s' % (platform, api))
-    future = self._GetImpl(platform, api)
+    future = self._GetSchemaView(platform, api)
     return All([future], except_pass=FileNotFoundError)