Created
January 4, 2010 07:25
-
-
Save ericflo/268379 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from django.core.cache import cache | |
from django.conf import settings | |
from django.contrib.auth.models import User | |
ONLINE_THRESHOLD = getattr(settings, 'ONLINE_THRESHOLD', 60 * 15) | |
ONLINE_MAX = getattr(settings, 'ONLINE_MAX', 50) | |
def get_online_now(self): | |
return User.objects.filter(id__in=self.online_now_ids or []) | |
class OnlineNowMiddleware(object): | |
""" | |
Maintains a list of users who have interacted with the website recently. | |
Their user IDs are available as ``online_now_ids`` on the request object, | |
and their corresponding users are available (lazily) as the | |
``online_now`` property on the request object. | |
""" | |
def process_request(self, request): | |
# First get the index | |
uids = cache.get('online-now', []) | |
# Perform the multiget on the individual online uid keys | |
online_keys = ['online-%s' % (u,) for u in uids] | |
fresh = cache.get_many(online_keys).keys() | |
online_now_ids = [int(k.replace('online-', '')) for k in fresh] | |
# If the user is authenticated, add their id to the list | |
if request.user.is_authenticated(): | |
uid = request.user.id | |
# If their uid is already in the list, we want to bump it | |
# to the top, so we remove the earlier entry. | |
if uid in online_now_ids: | |
online_now_ids.remove(uid) | |
online_now_ids.append(uid) | |
if len(online_now_ids) > ONLINE_MAX: | |
del online_now_ids[0] | |
# Attach our modifications to the request object | |
request.__class__.online_now_ids = online_now_ids | |
request.__class__.online_now = property(get_online_now) | |
# Set the new cache | |
cache.set('online-%s' % (request.user.pk,), True, ONLINE_THRESHOLD) | |
cache.set('online-now', online_now_ids, ONLINE_THRESHOLD) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
A criticism is that this imposes an artificial ONLINE_THRESHOLD on the list of current uids. You'd possibly have to use sorted sets to circumvent that and make your solution less constrained.