app/soc/cache/home.py
changeset 1419 5d0f80ad7b9f
child 1600 0aa3de1b2acc
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/app/soc/cache/home.py	Sat Feb 21 00:45:57 2009 +0000
@@ -0,0 +1,92 @@
+#!/usr/bin/python2.5
+#
+# Copyright 2009 the Melange authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Module contains sidebar memcaching functions.
+"""
+
+__authors__ = [
+    '"Sverre Rabbelier" <sverre@rabbelier.nl>',
+  ]
+
+
+import logging
+
+from google.appengine.api import memcache
+from google.appengine.api import users
+
+import soc.cache.base
+
+
+def key(entity):
+  """Returns the memcache key for the user's sidebar.
+  """
+
+  return 'homepage_for_%s_%s' % (entity.kind(), entity.key().name())
+
+
+def get(self, *args, **kwargs):
+  """Retrieves the sidebar for the specified user from the memcache.
+  """
+
+  # only cache the page for non-logged-in users
+  # TODO: figure out how to cache everything but the sidebar
+  if users.get_current_user():
+    return (None, None)
+
+  entity = self._logic.getFromKeyFields(kwargs)
+
+  # if we can't retrieve the entity, leave it to the actual method
+  if not entity:
+    return (None, None)
+
+  memcache_key = key(entity)
+  logging.info("Retrieving %s" % memcache_key)
+  return memcache.get(memcache_key), memcache_key
+
+def put(result, memcache_key, *args, **kwargs):
+  """Sets the sidebar for the specified user in the memcache.
+
+  Args:
+    sidebar: the sidebar to be cached
+  """
+
+  # no sense in storing anything if we won't query it later on
+  if users.get_current_user():
+    return
+
+  # Store sidebar for just ten minutes to force a refresh every so often
+  retention = 10*60
+
+  logging.info("Setting %s" % memcache_key)
+  memcache.add(memcache_key, result, retention)
+
+
+def flush(entity):
+  """Removes the sidebar for the current user from the memcache.
+
+  Also calls soc.cache.rights.flush for the specified user.
+
+  Args:
+    id: defaults to the current account if not set
+  """
+
+  memcache_key = key(entity)
+  logging.info("Flushing %s" % memcache_key)
+  memcache.delete(memcache_key)
+
+
+# define the cache function
+cache = soc.cache.base.getCacher(get, put)