Phase6

apps/worker/src/handlers/server-provisioning.ts

@@ -1,10 +1,13 @@
 import { randomUUID } from 'node:crypto';
 
 import { prisma } from '@hexahost/database';
+import { selectNodeForAllocation } from '@hexahost/scheduler';
 import { Job } from 'bullmq';
 import { z } from 'zod';
 
+import { releaseServerAllocation } from '../allocation';
 import { logger } from '../logger';
+import { loadNodeSnapshots } from '../node-snapshots';
 import { publishNodeCommand, waitForNodeResponse } from '../redis';
 import { transitionServerStatus } from '../server-state';
 
@@ -65,10 +68,9 @@ export async function processProvisionServerJob(job: Job): Promise<{ status: str
     throw new Error(`Server not found: ${serverId}`);
   }
 
-  const node = await prisma.gameNode.findFirst({
-    where: { status: 'ONLINE' },
-    orderBy: { createdAt: 'asc' },
-  });
+  const nodeSnapshots = await loadNodeSnapshots();
+  const ramMb = server.plan?.maxRamMb ?? server.ramMb;
+  const node = selectNodeForAllocation(nodeSnapshots, ramMb);
 
   if (!node) {
     await transitionServerStatus(serverId, 'ERROR', {
@@ -80,7 +82,6 @@ export async function processProvisionServerJob(job: Job): Promise<{ status: str
   }
 
   const hostPort = await findAvailableHostPort(node.id);
-  const ramMb = server.plan?.maxRamMb ?? server.ramMb;
   const dataPath = buildDataPath(serverId);
   const messageId = randomUUID();
 
@@ -95,6 +96,11 @@ export async function processProvisionServerJob(job: Job): Promise<{ status: str
       },
     });
 
+    await tx.gameNode.update({
+      where: { id: node.id },
+      data: { activeServers: { increment: 1 } },
+    });
+
     await tx.gameServer.update({
       where: { id: serverId },
       data: {
@@ -150,6 +156,7 @@ export async function processProvisionServerJob(job: Job): Promise<{ status: str
   }
 
   if (!response.success) {
+    await releaseServerAllocation(serverId);
     await transitionServerStatus(serverId, 'ERROR', {
       reason: 'Provision failed',
       correlationId: messageId,