fix: 修复没有测速文件的问题

Author: tianxiu2b2t

core/cluster.py

@@ -711,7 +711,7 @@ async def _fetch_cluster_name(self):
     def hit(self, cluster_id: str, bytes: int):
         self._clusters[cluster_id].counter.hits += 1
         self._clusters[cluster_id].counter.bytes += bytes
-    
+
     async def get_response_file(self, hash: str) -> ResponseFile:
         storage = self.storages.get_weight_storage()
         if storage is None:
@@ -726,14 +726,17 @@ async def get_response_file(self, hash: str) -> ResponseFile:
                 0
             )
 
-    async def get_measure_file(self, size: int) -> ResponseFile:
+    async def get_measure_file(self, size: int) -> Optional[ResponseFile]:
         if not self.storages._online_storages:
             logger.twarning("cluster.get_measure_file.no_storage")
             return ResponseFile(
                 0
             )
         storage = self.storages._online_storages[0]
-        return await storage.get_file(f"measure/{size}")
+        try:
+            return await storage.get_file(f"measure/{size}")
+        except:
+            return None
 
 
 if cfg.concurrency_enable_cluster:

core/config.py

@@ -92,7 +92,7 @@ def concurrency_enable_cluster(self) -> bool:
 
 
 API_VERSION = "1.13.1"
-VERSION = "4.0.10"
+VERSION = "4.0.11"
 PROJECT = "PythonOpenBMCLAPI"
 USER_AGENT = f"openbmclapi-cluster/{API_VERSION} {PROJECT}/{VERSION}"
 ROOT_PATH = Path(__file__).parent.parent

core/storage/abc.py

@@ -121,6 +121,21 @@ def path(self) -> 'CPath':
         return self._path
 
 
+    async def write_measure(self, size: int):
+        path = f"measure/{size}"
+        file = await self.get_file(path)
+        size = size * 1024 * 1024
+        if isinstance(file, (ResponseFileRemote, ResponseFileMemory, ResponseFileLocal)) and file.size == size:
+            return
+        
+        tmp = tempfile.TemporaryFile()
+        tmp.write(b'\x00' * size)
+        await self.upload(
+            path,
+            tmp,
+            size
+        )
+
     def get_py_check_path(self) -> 'CPath':
         return self.path / ".py_check"