pulumi/sdk/nodejs/runtime/settings.ts
pat@pulumi.com 97f99d7fa1 Do not disconnect from the engine prematurely.
The `nodejs` language support is implemented as two programs: one that
manages the initial connection to the engine and provides the language
serivce itself, and another that the language service invokes in order
to run a `nodejs` Pulumi program. The latter is responsible for running
the user's program and communicating its resource requests to the
engine. Currently, `run` effectively assumes that the user's program
will run synchronously from start to finish, and will disconnect from
the engine once the user's program has completed. This assumption breaks
if the user's program requires multiple turns of the event loop to
finish its root resource requests. For example, the following program
would fail to create its second resource because the engine will be
disconnected once it reaches its `await`:

```
(async () => {
    let a = new Resource();
    await somePromise();
    let = new Resource();
})();
```

These changes fix this issue by disconnecting from the engine during
process shutdown rather than after the user's program has finished its
first turn through the event loop.
2017-10-26 12:16:32 -07:00

127 lines
4.3 KiB
TypeScript

// Copyright 2016-2017, Pulumi Corporation. All rights reserved.
import { debuggablePromise } from "./debuggable";
/**
* excessiveDebugOutput enables, well, pretty excessive debug output pertaining to resources and properties.
*/
export let excessiveDebugOutput: boolean = false;
/**
* Options is a bag of settings that controls the behavior of previews and deployments
*/
export interface Options {
readonly project?: string; // the name of the current project.
readonly stack?: string; // the name of the current stack being deployed into.
readonly engine?: Object; // a live connection to the engine, used for logging, etc.
readonly monitor?: Object; // a live connection to the resource monitor that tracks deployments.
readonly parallel?: number; // the degree of parallelism for resource operations (default is serial).
readonly dryRun?: boolean; // whether we are performing a preview (true) or a real deployment (false).
readonly includeStackTraces?: boolean; // whether we include full stack traces in resource errors or not.
}
/**
* options are the current deployment options being used for this entire session.
*/
export let options: Options = {
dryRun: false,
includeStackTraces: true,
};
/**
* configured is set to true once configuration has been set.
*/
let configured: boolean;
/**
* hasMonitor returns true if we are currently connected to a resource monitoring service.
*/
export function hasMonitor(): boolean {
return !!options.monitor;
}
/**
* getMonitor returns the current resource monitoring service client for RPC communications.
*/
export function getMonitor(): Object | undefined {
if (!configured) {
configured = true;
console.warn("warning: Pulumi Fabric monitor is missing; no resources will be created");
}
return options.monitor;
}
/**
* getEngine returns the current engine, if any, for RPC communications back to the resource engine.
*/
export function getEngine(): Object | undefined {
return options.engine;
}
/**
* serialize returns true if resource operations should be serialized.
*/
export function serialize(): boolean {
return !options.parallel || options.parallel <= 1;
}
/**
* configure initializes the current resource monitor and engine RPC connections, and whether we are performing a "dry
* run" (preview), versus a real deployment, and so on. It may only be called once.
*/
export function configure(opts: Options): void {
if (configured) {
throw new Error("Cannot configure runtime settings more than once");
}
Object.assign(options, opts);
configured = true;
}
/**
* disconnect permanently disconnects from the server, closing the connections. It waits for the existing RPC
* queue to drain. If any RPCs come in afterwards, however, they will crash the process.
*/
export function disconnect(): void {
let done: Promise<any> | undefined;
const closeCallback: () => Promise<void> = () => {
if (done !== rpcDone) {
// If the done promise has changed, some activity occurred in between callbacks. Wait again.
done = rpcDone;
return debuggablePromise(done.then(closeCallback));
}
disconnectSync();
return Promise.resolve();
};
closeCallback();
}
/**
* disconnectSync permanently disconnects from the server, closing the connections. Unlike `disconnect`. it does not
* wait for the existing RPC queue to drain. Any RPCs that come in after this call will crash the process.
*/
export function disconnectSync(): void {
// Otherwise, actually perform the close activities.
if (options.monitor) {
(<any>options.monitor).close();
}
if (options.engine) {
(<any>options.engine).close();
}
}
/**
* rpcDone resolves when the last known client-side RPC call finishes.
*/
let rpcDone: Promise<any> = Promise.resolve();
/**
* rpcKeepAlive registers a pending call to ensure that we don't prematurely disconnect from the server. It returns
* a function that, when invoked, signals that the RPC has completed.
*/
export function rpcKeepAlive(): () => void {
let done: (() => void) | undefined = undefined;
const donePromise = debuggablePromise(new Promise<void>((resolve) => { done = resolve; }));
rpcDone = rpcDone.then(() => donePromise);
return done!;
}