bingshuo/guanghu-ice-heart
Watch
1
0
Fork
You've already forked guanghu-ice-heart
 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity

fix(pool): migrate legacy coordinator state before worker heartbeats

Browse source
This commit is contained in:
冰朔 2026-09-11 16:01:27 +08:00
commit 828a9f2f3f
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
server-tools/persona-compute-pool/entry_service.py
View file

@ -20,6 +20,9 @@ def read_state():
def write_state(value): def write_state(value):
STATE.parent.mkdir(parents=True, exist_ok=True); tmp=STATE.with_suffix('.tmp'); tmp.write_text(json.dumps(value,ensure_ascii=False,indent=2)+'\n'); tmp.replace(STATE) STATE.parent.mkdir(parents=True, exist_ok=True); tmp=STATE.with_suffix('.tmp'); tmp.write_text(json.dumps(value,ensure_ascii=False,indent=2)+'\n'); tmp.replace(STATE)
def normalized_state():
value=read_state(); value.setdefault('workers',{}); value.setdefault('leases',{}); value.setdefault('registered_workers',len(value['workers'])); return value
def authorized(handler): def authorized(handler):
try: try:
subject=handler.connection.getpeercert().get('subject',()) subject=handler.connection.getpeercert().get('subject',())
@ -39,13 +42,13 @@ class Handler(BaseHTTPRequestHandler):
try: payload=json.loads(self.rfile.read(int(self.headers.get('Content-Length','0')) or 0) or b'{}') try: payload=json.loads(self.rfile.read(int(self.headers.get('Content-Length','0')) or 0) or b'{}')
except Exception: self.send_json(400,{'error':'JSON_INVALID'}); return except Exception: self.send_json(400,{'error':'JSON_INVALID'}); return
with LOCK: with LOCK:
state=read_state() state=normalized_state()
if self.path=='/v1/worker/heartbeat': if self.path=='/v1/worker/heartbeat':
node=payload.get('node_id'); node=payload.get('node_id');
if not node or node=='JD-FD-PRIMARY': self.send_json(403,{'error':'MOTHER_NODE_FORBIDDEN'}); return if not node or node=='JD-FD-PRIMARY': self.send_json(403,{'error':'MOTHER_NODE_FORBIDDEN'}); return
peer=next((value for group in self.connection.getpeercert().get('subject',()) for key,value in group if key=='commonName'),None) peer=next((value for group in self.connection.getpeercert().get('subject',()) for key,value in group if key=='commonName'),None)
if node != peer: self.send_json(403,{'error':'WORKER_CERT_ID_MISMATCH'}); return if node != peer: self.send_json(403,{'error':'WORKER_CERT_ID_MISMATCH'}); return
state['workers'][node]={**payload,'last_seen':payload.get('observed_at'),'authority_granted':False}; write_state(state); self.send_json(200,{'outcome':'HEARTBEAT_ACCEPTED','node_id':node,'authority_granted':False}); return state['workers'][node]={**payload,'last_seen':payload.get('observed_at'),'authority_granted':False}; state['registered_workers']=len(state['workers']); write_state(state); self.send_json(200,{'outcome':'HEARTBEAT_ACCEPTED','node_id':node,'authority_granted':False}); return
if self.path=='/v1/lease/request': if self.path=='/v1/lease/request':
self.send_json(200,{'outcome':'LEASE_QUEUED','state':'PENDING_DETERMINISTIC_SELECTION','authority_granted':False}); return self.send_json(200,{'outcome':'LEASE_QUEUED','state':'PENDING_DETERMINISTIC_SELECTION','authority_granted':False}); return
self.send_json(404,{'error':'NOT_FOUND'}) self.send_json(404,{'error':'NOT_FOUND'})