源码快照 @ 2026-09-01 (source-only, 剔除非源码素材)

packages/runtime-host/src/server/process-lifecycle.ts

@@ -0,0 +1,61 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+import {
+  RuntimeHostProcessTerminationRequiredError,
+  type RuntimeHostKernel,
+} from './host-kernel.js';
+
+export interface RuntimeHostProcessLifecycleOptions {
+  closeOnDisconnect?: boolean;
+  onReady?: () => void;
+}
+
+export type RuntimeHostProcessLifecycleEnd = 'host_closed' | 'termination_requested';
+
+export async function runRuntimeHostProcessLifecycle(
+  host: Pick<RuntimeHostKernel, 'close' | 'closed'> &
+    Partial<Pick<RuntimeHostKernel, 'shutdownReason'>>,
+  options: RuntimeHostProcessLifecycleOptions = {},
+): Promise<RuntimeHostProcessLifecycleEnd> {
+  let closing = false;
+  const close = () => {
+    if (closing) return;
+    closing = true;
+    void host.close();
+  };
+
+  process.once('SIGINT', close);
+  process.once('SIGTERM', close);
+  if (options.closeOnDisconnect) process.once('disconnect', close);
+  try {
+    options.onReady?.();
+    await host.closed;
+    return closing ? 'termination_requested' : 'host_closed';
+  } catch (error) {
+    if (error instanceof RuntimeHostProcessTerminationRequiredError) {
+      process.exit(host.shutdownReason === 'retirement' ? 0 : 1);
+    }
+    throw error;
+  } finally {
+    process.off('SIGINT', close);
+    process.off('SIGTERM', close);
+    if (options.closeOnDisconnect) process.off('disconnect', close);
+  }
+}