添加页面加载耗时记录

DjangoBlog/settings.py

@@ -58,6 +58,7 @@ MIDDLEWARE_CLASSES = [
     'django.contrib.messages.middleware.MessageMiddleware',
     'django.middleware.clickjacking.XFrameOptionsMiddleware',
     'django.middleware.http.ConditionalGetMiddleware',
+    'blog.middleware.OnlineMiddleware'
 ]
 
 ROOT_URLCONF = 'DjangoBlog.urls'

blog/middleware.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+# encoding: utf-8
+
+
+"""
+@version: ??
+@author: liangliangyy
+@license: MIT Licence
+@contact: liangliangyy@gmail.com
+@site: https://www.lylinux.org/
+@software: PyCharm
+@file: middleware.py
+@time: 2017/1/19 上午12:36
+"""
+
+import time
+from ipware.ip import get_real_ip
+from django.core.cache import cache
+
+
+class OnlineMiddleware(object):
+    def process_request(self, request):
+        self.start_time = time.time()
+
+    def process_view(self, request, view_func, view_args, view_kwargs):
+        """
+        处理当前在线人数
+        """
+        http_user_agent = request.META.get('HTTP_USER_AGENT', [])
+        if 'Spider' in http_user_agent or 'spider' in http_user_agent:
+            return
+
+        online_ips = cache.get("online_ips", [])
+
+        if online_ips:
+            online_ips = cache.get_many(online_ips).keys()
+
+        ip = get_real_ip(request)
+
+        cache.set(ip, 0, 5 * 60)
+
+        if ip not in online_ips:
+            online_ips.append(ip)
+
+        cache.set("online_ips", online_ips)
+
+    def process_response(self, request, response):
+        cast_time = time.time() - self.start_time
+        response.content = response.content.replace(b'<!!LOAD_TIMES!!>', str.encode(str(cast_time)[:5]))
+        return response

requirements.txt

@@ -13,4 +13,5 @@ requests
 Unidecode
 Whoosh
 mistune
-pygments
+pygments
+django-ipware

templates/share_layout/footer.html

@@ -22,7 +22,8 @@
         <a href="https://www.djangoproject.com/" rel="nofollow" target="blank">Django</a>
         &nbsp;|&nbsp;
         <a href="https://github.com/liangliangyy/DjangoBlog" rel="nofollow" target="blank"> LiangLiangyy</a>
-
+        |
+        本页面加载耗时:<!!LOAD_TIMES!!>s
     </div>
 </footer><!-- #colophon -->