59 lines
2.3 KiB
TypeScript
59 lines
2.3 KiB
TypeScript
// Copyright 2016-2017, Pulumi Corporation. All rights reserved.
|
|
|
|
import * as log from "../log";
|
|
import { ComputedValues } from "../resource";
|
|
import { debuggablePromise } from "./debuggable";
|
|
import { deserializeProperties, serializeProperties } from "./rpc";
|
|
import { excessiveDebugOutput, getMonitor, options, 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: ComputedValues): 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: Error, innerResponse: any) => {
|
|
log.debug(`Invoke RPC finished: tok=${tok}; err: ${err}, resp: ${innerResponse}`);
|
|
if (err) {
|
|
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();
|
|
}
|
|
}
|
|
|