68911900fd
* Graceful RPC shutdown: CLI side * Handle unavailable resource monitor in language hosts * Fix a comment * Don't commit package-lock.json * fix mangled pylint pragma * Rebase against master and fix Gopkg.lock * Code review feedback * Fix a race between closing the callerEventsOpt channel and terminating a goroutine that writes to it * glog -> logging
75 lines
2.9 KiB
TypeScript
75 lines
2.9 KiB
TypeScript
// Copyright 2016-2018, Pulumi Corporation. All rights reserved.
|
|
|
|
import * as grpc from "grpc";
|
|
import * as log from "../log";
|
|
import { Inputs } from "../resource";
|
|
import { debuggablePromise } from "./debuggable";
|
|
import { deserializeProperties, serializeProperties } from "./rpc";
|
|
import { excessiveDebugOutput, getMonitor, rpcKeepAlive, serialize } from "./settings";
|
|
|
|
const gstruct = require("google-protobuf/google/protobuf/struct_pb.js");
|
|
const resproto = require("../proto/resource_pb.js");
|
|
|
|
/**
|
|
* invoke dynamically invokes the function, tok, which is offered by a provider plugin. The inputs
|
|
* can be a bag of computed values (Ts or Promise<T>s), and the result is a Promise<any> that
|
|
* resolves when the invoke finishes.
|
|
*/
|
|
export async function invoke(tok: string, props: Inputs): Promise<any> {
|
|
log.debug(`Invoking function: tok=${tok}` +
|
|
excessiveDebugOutput ? `, props=${JSON.stringify(props)}` : ``);
|
|
|
|
// Wait for all values to be available, and then perform the RPC.
|
|
const done = rpcKeepAlive();
|
|
try {
|
|
const obj = gstruct.Struct.fromJavaScript(
|
|
await serializeProperties(`invoke:${tok}`, props));
|
|
log.debug(`Invoke RPC prepared: tok=${tok}` + excessiveDebugOutput ? `, obj=${JSON.stringify(obj)}` : ``);
|
|
|
|
// Fetch the monitor and make an RPC request.
|
|
const monitor: any = getMonitor();
|
|
|
|
const req = new resproto.InvokeRequest();
|
|
req.setTok(tok);
|
|
req.setArgs(obj);
|
|
const resp: any = await debuggablePromise(new Promise((innerResolve, innerReject) =>
|
|
monitor.invoke(req, (err: grpc.ServiceError, innerResponse: any) => {
|
|
log.debug(`Invoke RPC finished: tok=${tok}; err: ${err}, resp: ${innerResponse}`);
|
|
if (err) {
|
|
// If the monitor is unavailable, it is in the process of shutting down or has already
|
|
// shut down. Don't emit an error and don't do any more RPCs.
|
|
if (err.code === grpc.status.UNAVAILABLE) {
|
|
log.debug("Resource monitor is terminating");
|
|
waitForDeath();
|
|
}
|
|
|
|
innerReject(err);
|
|
}
|
|
else {
|
|
innerResolve(innerResponse);
|
|
}
|
|
})));
|
|
|
|
// If there were failures, propagate them.
|
|
const failures: any = resp.getFailuresList();
|
|
if (failures && failures.length) {
|
|
throw new Error(`Invoke of '${tok}' failed: ${failures[0].reason} (${failures[0].property})`);
|
|
}
|
|
|
|
// Finally propagate any other properties that were given to us as outputs.
|
|
return deserializeProperties(resp.getReturn());
|
|
}
|
|
finally {
|
|
done();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* waitForDeath loops forever. See the comments in resource.ts on the function with
|
|
* the same name for an explanation as to why this exists.
|
|
*/
|
|
function waitForDeath(): never {
|
|
// tslint:disable-next-line
|
|
while (true) {}
|
|
}
|