添加wda守护

This commit is contained in:
2025-09-12 21:36:29 +08:00
parent 9cf094986c
commit 398d980caa
3 changed files with 41 additions and 8 deletions

4
.idea/workspace.xml generated
View File

@@ -5,11 +5,9 @@
</component> </component>
<component name="ChangeListManager"> <component name="ChangeListManager">
<list default="true" id="eceeff5e-51c1-459c-a911-d21ec090a423" name="Changes" comment="20250904-初步功能已完成"> <list default="true" id="eceeff5e-51c1-459c-a911-d21ec090a423" name="Changes" comment="20250904-初步功能已完成">
<change beforePath="$PROJECT_DIR$/.idea/iOSAI.iml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/iOSAI.iml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/misc.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/misc.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" /> <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/Entity/DeviceModel.py" beforeDir="false" afterPath="$PROJECT_DIR$/Entity/DeviceModel.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/Module/DeviceInfo.py" beforeDir="false" afterPath="$PROJECT_DIR$/Module/DeviceInfo.py" afterDir="false" /> <change beforePath="$PROJECT_DIR$/Module/DeviceInfo.py" beforeDir="false" afterPath="$PROJECT_DIR$/Module/DeviceInfo.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/Module/FlaskService.py" beforeDir="false" afterPath="$PROJECT_DIR$/Module/FlaskService.py" afterDir="false" />
</list> </list>
<option name="SHOW_DIALOG" value="false" /> <option name="SHOW_DIALOG" value="false" />
<option name="HIGHLIGHT_CONFLICTS" value="true" /> <option name="HIGHLIGHT_CONFLICTS" value="true" />

View File

@@ -15,7 +15,7 @@ class DeviceModel(object):
self.scale = scale self.scale = scale
# 1 添加 2删除 # 1 添加 2删除
self.type = type self.type = type
self.ready = False
# 转字典 # 转字典
def toDict(self): def toDict(self):

View File

@@ -34,6 +34,9 @@ class Deviceinfo(object):
self._port_pool: List[int] = [] # 端口回收池 self._port_pool: List[int] = [] # 端口回收池
self._port_in_use: set[int] = set() # 正在使用的端口 self._port_in_use: set[int] = set() # 正在使用的端口
# 🔥1. 启动 WDA 健康检查线程
threading.Thread(target=self._wda_health_checker, daemon=True).start()
# region iproxy 初始化 # region iproxy 初始化
try: try:
self.iproxy_path = self._iproxy_path() self.iproxy_path = self._iproxy_path()
@@ -124,7 +127,40 @@ class Deviceinfo(object):
time.sleep(1) time.sleep(1)
# endregion # 🔥2. WDA 健康检查
def _wda_health_checker(self):
while True:
time.sleep(1)
print(len(self.deviceModelList))
with self._lock:
online = [m for m in self.deviceModelList if m.ready] # ← 只检查就绪的
print(len(online))
for model in online:
udid = model.deviceId
if not self._wda_ok(udid):
LogManager.warning(f"WDA 异常,重启通道:{udid}", udid)
with self._lock:
self._remove_model(udid)
self.connectDevice(udid)
# 🔥3. 真正做 health-check 的地方
def _wda_ok(self, udid: str) -> bool:
"""返回 True 表示 WDA 活着False 表示已死"""
try:
# 用 2 秒超时快速探测
c = wda.USBClient(udid, 8100)
# 下面这句就是“xctest launched but check failed” 的触发点
# 如果 status 里返回了 WebDriverAgent 运行信息就认为 OK
st = c.status()
if st.get("state") != "success":
return False
# 你也可以再苛刻一点,多做一次 /wda/healthcheck
# c.http.get("/wda/healthcheck")
return True
except Exception as e:
# 任何异常连接拒绝、超时、json 解析失败)都认为已死
LogManager.error(f"WDA health-check 异常:{e}", udid)
return False
# region ===================== 增删改查唯一入口(线程安全) ===================== # region ===================== 增删改查唯一入口(线程安全) =====================
def _has_model(self, udid: str) -> bool: def _has_model(self, udid: str) -> bool:
@@ -133,6 +169,7 @@ class Deviceinfo(object):
def _add_model(self, model: DeviceModel): def _add_model(self, model: DeviceModel):
if model.deviceId in self._model_index: if model.deviceId in self._model_index:
return # 防重复 return # 防重复
model.ready = True
self.deviceModelList.append(model) self.deviceModelList.append(model)
self._model_index[model.deviceId] = model self._model_index[model.deviceId] = model
try: try:
@@ -188,6 +225,7 @@ class Deviceinfo(object):
# region ===================== 端口分配与回收 ===================== # region ===================== 端口分配与回收 =====================
def _alloc_port(self) -> int: def _alloc_port(self) -> int:
print(self.screenProxy)
if self._port_pool: if self._port_pool:
port = self._port_pool.pop() port = self._port_pool.pop()
else: else:
@@ -207,9 +245,6 @@ class Deviceinfo(object):
if not self.is_device_trusted(udid): if not self.is_device_trusted(udid):
LogManager.warning("设备未信任,跳过 WDA 启动", udid) LogManager.warning("设备未信任,跳过 WDA 启动", udid)
return return
if self._has_model(udid):
LogManager.warning("设备已存在,跳过重复连接", udid)
return
try: try:
d = wda.USBClient(udid, 8100) d = wda.USBClient(udid, 8100)