TTB-Network
diff --git a/‎core/__init__.py‎
Lines changed: 2 additions & 1 deletion b/‎core/__init__.py‎
Lines changed: 2 additions & 1 deletion
diff --git a/‎core/dashboard.py‎
Lines changed: 8 additions & 1 deletion b/‎core/dashboard.py‎
Lines changed: 8 additions & 1 deletion
diff --git a/‎core/storage/alist.py‎
Lines changed: 0 additions & 1 deletion b/‎core/storage/alist.py‎
Lines changed: 0 additions & 1 deletion
diff --git a/‎core/web.py‎
Lines changed: 45 additions & 0 deletions b/‎core/web.py‎
Lines changed: 45 additions & 0 deletions
@@ -226,7 +226,8 @@ def access_log(request: fastapi.Request, response: fastapi.Response, total_time:
     )
 
 @web.app.middleware("http")
-async def auth_middleware(request: fastapi.Request, call_next):
+async def access_log_middleware(request: fastapi.Request, call_next):
+    web.query_per_second_statistics.add()
     start_time = runtime.get_perf_counter_ns()
     try:
         result = await call_next(request)
 
@@ -1,11 +1,12 @@
 from collections import defaultdict
-from typing import Any
+from typing import Any, Optional
 import anyio
 from bson import ObjectId
 import fastapi
 from fastapi.staticfiles import StaticFiles
 
 from .config import ROOT_PATH, DEBUG
+from .web import query_per_second_statistics
 
 
 class StreamNotice:
@@ -51,6 +52,12 @@ def _():
     @app.get("/api/receive")
     async def _(request: fastapi.Request):
         return notice.add_client(request)
+    
+    @app.get("/api/qps")
+    async def _(request: fastapi.Request, interval: Optional[int] = 5):
+        if interval is None:
+            return query_per_second_statistics.get_all()
+        return query_per_second_statistics.merge_data(interval)
 
     @app.get("/")
     @app.get("/{page}")
 
@@ -114,7 +114,6 @@ async def _check(self):
 
     async def check_measure(self, size: int):
         path = str(self._path / "measure" / size)
-        print(path)
         async with aiohttp.ClientSession(
             base_url=self._endpoint,
             headers={
 
@@ -1,4 +1,5 @@
 from collections import defaultdict
+import datetime
 import ssl
 import anyio
 import anyio.abc
@@ -8,6 +9,7 @@
 import uvicorn
 import tianxiu2b2t.anyio.streams as streams
 from tianxiu2b2t.anyio import concurrency
+from tianxiu2b2t.utils import runtime
 
 from . import utils, abc
 from .logger import logger
@@ -35,6 +37,48 @@ def __exit__(self, *args):
             del forwards[self.sockname]
             del forwards_count[self.sockname]
 
+class QueryPerSecondStatistics:
+    def __init__(
+        self,
+        expires: int = 600
+    ):
+        self._timer = lambda: runtime.monotonic()
+        self._data: defaultdict[int, int] = defaultdict(int)
+        self._expires = expires
+
+
+    def add(self):
+        self._data[int(self._timer())] += 1
+        self.expire()
+    
+    def expire(self):
+        t = self._timer()
+        for k, v in list(self._data.items()):
+            if k + self._expires < t:
+                del self._data[k]
+
+    def get_all(self) -> dict[datetime.datetime, int]:
+        now = datetime.datetime.now().replace(microsecond=0)
+        t = self._timer()
+        data = {}
+        for k, v in self._data.items():
+            if k >= t:
+                continue
+            data[(now - datetime.timedelta(seconds=t - k)).replace(microsecond=0)] += v
+        return data
+    
+    def merge_data(self, interval: int = 5) -> dict[datetime.datetime, int]:
+        timestamp = datetime.datetime.fromtimestamp(datetime.datetime.now().timestamp() // interval * interval)
+        res: defaultdict[datetime.datetime, int] = defaultdict(int)
+        cur = int(self._timer() // interval)
+        for k, v in self._data.items():
+            c = int(k // interval)
+            if c > cur:
+                continue
+            res[timestamp + datetime.timedelta(seconds=c * interval)] += v
+        return res
+
+
 app = fastapi.FastAPI(
     redoc_url=None,
     docs_url=None,
@@ -45,6 +89,7 @@ def __exit__(self, *args):
 tls_listener: streams.AutoTLSListener | None = None
 forwards: dict[tuple[str, int], tuple[str, int]] = {}
 forwards_count: defaultdict[tuple[str, int], int] = defaultdict(int)
+query_per_second_statistics = QueryPerSecondStatistics()
 
 async def get_free_port():
     listener = await anyio.create_tcp_listener()
Original file line number	Diff line number	Diff line change
`@@ -226,7 +226,8 @@ def access_log(request: fastapi.Request, response: fastapi.Response, total_time:`
`226`	`226`	`)`
`227`	`227`
`228`	`228`	`@web.app.middleware("http")`
`229`		`-async def auth_middleware(request: fastapi.Request, call_next):`
	`229`	`+async def access_log_middleware(request: fastapi.Request, call_next):`
	`230`	`+ web.query_per_second_statistics.add()`
`230`	`231`	`start_time = runtime.get_perf_counter_ns()`
`231`	`232`	`try:`
`232`	`233`	`result = await call_next(request)`