add access count metrics

BennyThink · BennyThink · commit 7bf2e70d3d9c · 2021-02-06T17:21:33.000+08:00
diff --git a/web/index.html b/web/index.html
@@ -32,14 +32,14 @@
             width: 75%;
         }
 
-        a::after {
-            content: attr(title);
-            display: none;
-        }
+        /*a::after {*/
+        /*    content: attr(title);*/
+        /*    display: none;*/
+        /*}*/
 
-        a:hover::after {
-            display: block;
-        }
+        /*a:hover::after {*/
+        /*    display: block;*/
+        /*}*/
 
     </style>
 </head>
@@ -67,6 +67,7 @@ <h2>有问题联系 <a style="text-decoration: none;color: green" href="https://
 </div>
 </body>
 <script src="js/axios.min.js"></script>
+<script src="js/analytics.js"></script>
 
 <script>
     // get top
@@ -99,5 +100,7 @@ <h2>有问题联系 <a style="text-decoration: none;color: green" href="https://
 
     }
 
+    accessMetrics();
+
 </script>
 </html>
diff --git a/web/js/analytics.js b/web/js/analytics.js
@@ -0,0 +1,9 @@
+function accessMetrics() {
+    axios.post('/api/metrics')
+        .then(function (response) {
+            // console.log(response);
+        })
+        .catch(function (error) {
+            // console.log(error);
+        });
+}
diff --git a/web/resource.html b/web/resource.html
@@ -660,7 +660,7 @@
                 getData: function getData() {
                     var me = this;
                     let resource_id = document.URL.split("id=")[1]
-                    $.get("/api/resource?id="+resource_id, function (json) {
+                    $.get("/api/resource?id=" + resource_id, function (json) {
                         // json.data.info.expire < (Date.now()/1000)
                         if (json.status != 1 || !json.data) {
                             return location.href = location.origin
@@ -854,6 +854,9 @@
         // });
     }
 
+    $.post("/api/metrics", function (data) {
+    });
+
 </script>
 </body>
 </html>
diff --git a/web/search.html b/web/search.html
@@ -6,7 +6,6 @@
     <link rel="icon" type="image/x-icon" href="favicon.ico">
     <link href="https://fonts.googleapis.com/css?family=Pacifico&display=swap" rel="stylesheet">
     <link href="/css/normalize.min.css" rel="stylesheet">
-    <script src="js/axios.min.js"></script>
     <style>
         h1 {
             font-family: Pacifico, sans-serif;
@@ -54,6 +53,9 @@ <h2>在 <a style="text-decoration: none;color: deepskyblue" href="https://t.me/y
     <h2>有问题请联系 <a style="text-decoration: none;color: green" href="https://t.me/BennyThink">Benny小可爱</a></h2>
 </div>
 </body>
+<script src="js/axios.min.js"></script>
+<script src="js/analytics.js"></script>
+
 <script>
     let kwe = document.URL.split("kw=")[1];
     let kw = decodeURI(kwe).toLowerCase().replace(" ", "");
@@ -93,5 +95,7 @@ <h2>有问题请联系 <a style="text-decoration: none;color: green" href="https
             div.innerHTML = `<h2>没有搜索到结果 (ノへ￣、)</h2>`
         }
     }
+
+    accessMetrics();
 </script>
 </html>
diff --git a/web/server.py b/web/server.py
@@ -12,6 +12,7 @@
 from platform import uname
 import os
 import contextlib
+from http import HTTPStatus
 from concurrent.futures import ThreadPoolExecutor
 from tornado import web, ioloop, httpserver, gen, options
 from tornado.log import enable_pretty_logging
@@ -98,17 +99,31 @@ def get(self):
         self.write(resp)
 
 
-class PingHandler(BaseHandler):
+class MetricsHandler(BaseHandler):
     executor = ThreadPoolExecutor(50)
 
     @run_on_executor()
-    def ping(self):
-        os.system('ping z.cn -n 10')
-        return 'Ping complete'
+    def set_metrics(self):
+        metrics_name = self.get_query_argument("type", "access")
+        db['metrics'].find_one_and_update(
+            {'type': metrics_name}, {'$inc': {'count': 1}}
+        )
+        self.set_status(HTTPStatus.CREATED)
+        return {}
+
+    @run_on_executor()
+    def get_metrics(self):
+        metrics_name = self.get_query_argument("type", "access")
+        return db['metrics'].find_one({'type': metrics_name}, {'_id': False})
 
     @gen.coroutine
     def get(self):
-        resp = yield self.ping()
+        resp = yield self.get_metrics()
+        self.write(resp)
+
+    @gen.coroutine
+    def post(self):
+        resp = yield self.set_metrics()
         self.write(resp)
 
 
@@ -118,6 +133,7 @@ class RunServer:
     handlers = [
         (r'/api/resource', ResourceHandler),
         (r'/api/top', TopHandler),
+        (r'/api/metrics', MetricsHandler),
         (r'/', IndexHandler),
         (r'/(.*\.html|.*\.js|.*\.css|.*\.png|.*\.jpg|.*\.ico|.*\.gif|.*\.woff2)', web.StaticFileHandler,
          {'path': static_path}),
