tanbushi commited on
Commit
72a5949
·
1 Parent(s): 989566d

fix: 用路径(/health)区分健康检查, 避免HF代理丢弃Upgrade header

Browse files
Files changed (1) hide show
  1. server.py +3 -5
server.py CHANGED
@@ -57,11 +57,9 @@ async def handle_client(reader, writer):
57
  if not data:
58
  writer.close()
59
  return
60
- is_health_check = (
61
- data.startswith(b"GET / ")
62
- and b"Upgrade: websocket" not in data
63
- )
64
- if is_health_check:
65
  await health_check(reader, writer)
66
  else:
67
  await tcp_proxy(reader, writer, initial_data=data)
 
57
  if not data:
58
  writer.close()
59
  return
60
+ path = data.split(b" ", 2)[1] if data.count(b" ") >= 2 else b""
61
+ path = path.split(b"?")[0]
62
+ if path in (b"/", b"/health"):
 
 
63
  await health_check(reader, writer)
64
  else:
65
  await tcp_proxy(reader, writer, initial_data=data)