mirror of https://github.com/grpc/grpc-node.git
730 lines
24 KiB
C++
730 lines
24 KiB
C++
/*
|
|
*
|
|
* Copyright 2015, Google Inc.
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are
|
|
* met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above
|
|
* copyright notice, this list of conditions and the following disclaimer
|
|
* in the documentation and/or other materials provided with the
|
|
* distribution.
|
|
* * Neither the name of Google Inc. nor the names of its
|
|
* contributors may be used to endorse or promote products derived from
|
|
* this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
*/
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <map>
|
|
|
|
#include <node.h>
|
|
|
|
#include "grpc/support/log.h"
|
|
#include "grpc/grpc.h"
|
|
#include "grpc/support/alloc.h"
|
|
#include "grpc/support/time.h"
|
|
#include "byte_buffer.h"
|
|
#include "call.h"
|
|
#include "channel.h"
|
|
#include "completion_queue_async_worker.h"
|
|
#include "timeval.h"
|
|
|
|
using std::unique_ptr;
|
|
using std::shared_ptr;
|
|
using std::vector;
|
|
|
|
namespace grpc {
|
|
namespace node {
|
|
|
|
using Nan::Callback;
|
|
using Nan::EscapableHandleScope;
|
|
using Nan::HandleScope;
|
|
using Nan::Maybe;
|
|
using Nan::MaybeLocal;
|
|
using Nan::ObjectWrap;
|
|
using Nan::Persistent;
|
|
using Nan::Utf8String;
|
|
|
|
using v8::Array;
|
|
using v8::Boolean;
|
|
using v8::Exception;
|
|
using v8::External;
|
|
using v8::Function;
|
|
using v8::FunctionTemplate;
|
|
using v8::Integer;
|
|
using v8::Local;
|
|
using v8::Number;
|
|
using v8::Object;
|
|
using v8::ObjectTemplate;
|
|
using v8::Uint32;
|
|
using v8::String;
|
|
using v8::Value;
|
|
|
|
Callback *Call::constructor;
|
|
Persistent<FunctionTemplate> Call::fun_tpl;
|
|
|
|
bool EndsWith(const char *str, const char *substr) {
|
|
return strcmp(str+strlen(str)-strlen(substr), substr) == 0;
|
|
}
|
|
|
|
bool CreateMetadataArray(Local<Object> metadata, grpc_metadata_array *array,
|
|
shared_ptr<Resources> resources) {
|
|
HandleScope scope;
|
|
grpc_metadata_array_init(array);
|
|
Local<Array> keys = Nan::GetOwnPropertyNames(metadata).ToLocalChecked();
|
|
for (unsigned int i = 0; i < keys->Length(); i++) {
|
|
Local<String> current_key = Nan::To<String>(
|
|
Nan::Get(keys, i).ToLocalChecked()).ToLocalChecked();
|
|
Local<Value> value_array = Nan::Get(metadata, current_key).ToLocalChecked();
|
|
if (!value_array->IsArray()) {
|
|
return false;
|
|
}
|
|
array->capacity += Local<Array>::Cast(value_array)->Length();
|
|
}
|
|
array->metadata = reinterpret_cast<grpc_metadata*>(
|
|
gpr_malloc(array->capacity * sizeof(grpc_metadata)));
|
|
for (unsigned int i = 0; i < keys->Length(); i++) {
|
|
Local<String> current_key(keys->Get(i)->ToString());
|
|
Utf8String *utf8_key = new Utf8String(current_key);
|
|
resources->strings.push_back(unique_ptr<Utf8String>(utf8_key));
|
|
Local<Array> values = Local<Array>::Cast(
|
|
Nan::Get(metadata, current_key).ToLocalChecked());
|
|
for (unsigned int j = 0; j < values->Length(); j++) {
|
|
Local<Value> value = Nan::Get(values, j).ToLocalChecked();
|
|
grpc_metadata *current = &array->metadata[array->count];
|
|
current->key = **utf8_key;
|
|
// Only allow binary headers for "-bin" keys
|
|
if (EndsWith(current->key, "-bin")) {
|
|
if (::node::Buffer::HasInstance(value)) {
|
|
current->value = ::node::Buffer::Data(value);
|
|
current->value_length = ::node::Buffer::Length(value);
|
|
PersistentValue *handle = new PersistentValue(value);
|
|
resources->handles.push_back(unique_ptr<PersistentValue>(handle));
|
|
} else {
|
|
return false;
|
|
}
|
|
} else {
|
|
if (value->IsString()) {
|
|
Local<String> string_value = Nan::To<String>(value).ToLocalChecked();
|
|
Utf8String *utf8_value = new Utf8String(string_value);
|
|
resources->strings.push_back(unique_ptr<Utf8String>(utf8_value));
|
|
current->value = **utf8_value;
|
|
current->value_length = string_value->Length();
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
array->count += 1;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
Local<Value> ParseMetadata(const grpc_metadata_array *metadata_array) {
|
|
EscapableHandleScope scope;
|
|
grpc_metadata *metadata_elements = metadata_array->metadata;
|
|
size_t length = metadata_array->count;
|
|
std::map<const char*, size_t> size_map;
|
|
std::map<const char*, size_t> index_map;
|
|
|
|
for (unsigned int i = 0; i < length; i++) {
|
|
const char *key = metadata_elements[i].key;
|
|
if (size_map.count(key)) {
|
|
size_map[key] += 1;
|
|
} else {
|
|
size_map[key] = 1;
|
|
}
|
|
index_map[key] = 0;
|
|
}
|
|
Local<Object> metadata_object = Nan::New<Object>();
|
|
for (unsigned int i = 0; i < length; i++) {
|
|
grpc_metadata* elem = &metadata_elements[i];
|
|
Local<String> key_string = Nan::New(elem->key).ToLocalChecked();
|
|
Local<Array> array;
|
|
MaybeLocal<Value> maybe_array = Nan::Get(metadata_object, key_string);
|
|
if (maybe_array.IsEmpty() || !maybe_array.ToLocalChecked()->IsArray()) {
|
|
array = Nan::New<Array>(size_map[elem->key]);
|
|
Nan::Set(metadata_object, key_string, array);
|
|
} else {
|
|
array = Local<Array>::Cast(maybe_array.ToLocalChecked());
|
|
}
|
|
if (EndsWith(elem->key, "-bin")) {
|
|
Nan::Set(array, index_map[elem->key],
|
|
MakeFastBuffer(
|
|
Nan::CopyBuffer(elem->value,
|
|
elem->value_length).ToLocalChecked()));
|
|
} else {
|
|
Nan::Set(array, index_map[elem->key],
|
|
Nan::New(elem->value).ToLocalChecked());
|
|
}
|
|
index_map[elem->key] += 1;
|
|
}
|
|
return scope.Escape(metadata_object);
|
|
}
|
|
|
|
Local<Value> Op::GetOpType() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::New(GetTypeString()).ToLocalChecked());
|
|
}
|
|
|
|
Op::~Op() {
|
|
}
|
|
|
|
class SendMetadataOp : public Op {
|
|
public:
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::True());
|
|
}
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
if (!value->IsObject()) {
|
|
return false;
|
|
}
|
|
grpc_metadata_array array;
|
|
MaybeLocal<Object> maybe_metadata = Nan::To<Object>(value);
|
|
if (maybe_metadata.IsEmpty()) {
|
|
return false;
|
|
}
|
|
if (!CreateMetadataArray(maybe_metadata.ToLocalChecked(),
|
|
&array, resources)) {
|
|
return false;
|
|
}
|
|
out->data.send_initial_metadata.count = array.count;
|
|
out->data.send_initial_metadata.metadata = array.metadata;
|
|
return true;
|
|
}
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "send_metadata";
|
|
}
|
|
};
|
|
|
|
class SendMessageOp : public Op {
|
|
public:
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::True());
|
|
}
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
if (!::node::Buffer::HasInstance(value)) {
|
|
return false;
|
|
}
|
|
Local<Object> object_value = Nan::To<Object>(value).ToLocalChecked();
|
|
MaybeLocal<Value> maybe_flag_value = Nan::Get(
|
|
object_value, Nan::New("grpcWriteFlags").ToLocalChecked());
|
|
if (!maybe_flag_value.IsEmpty()) {
|
|
Local<Value> flag_value = maybe_flag_value.ToLocalChecked();
|
|
if (flag_value->IsUint32()) {
|
|
Maybe<uint32_t> maybe_flag = Nan::To<uint32_t>(flag_value);
|
|
out->flags = maybe_flag.FromMaybe(0) & GRPC_WRITE_USED_MASK;
|
|
}
|
|
}
|
|
out->data.send_message = BufferToByteBuffer(value);
|
|
PersistentValue *handle = new PersistentValue(value);
|
|
resources->handles.push_back(unique_ptr<PersistentValue>(handle));
|
|
return true;
|
|
}
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "send_message";
|
|
}
|
|
};
|
|
|
|
class SendClientCloseOp : public Op {
|
|
public:
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::True());
|
|
}
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
return true;
|
|
}
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "client_close";
|
|
}
|
|
};
|
|
|
|
class SendServerStatusOp : public Op {
|
|
public:
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::True());
|
|
}
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
if (!value->IsObject()) {
|
|
return false;
|
|
}
|
|
Local<Object> server_status = Nan::To<Object>(value).ToLocalChecked();
|
|
MaybeLocal<Value> maybe_metadata = Nan::Get(
|
|
server_status, Nan::New("metadata").ToLocalChecked());
|
|
if (maybe_metadata.IsEmpty()) {
|
|
return false;
|
|
}
|
|
if (!maybe_metadata.ToLocalChecked()->IsObject()) {
|
|
return false;
|
|
}
|
|
Local<Object> metadata = Nan::To<Object>(
|
|
maybe_metadata.ToLocalChecked()).ToLocalChecked();
|
|
MaybeLocal<Value> maybe_code = Nan::Get(server_status,
|
|
Nan::New("code").ToLocalChecked());
|
|
if (maybe_code.IsEmpty()) {
|
|
return false;
|
|
}
|
|
if (!maybe_code.ToLocalChecked()->IsUint32()) {
|
|
return false;
|
|
}
|
|
uint32_t code = Nan::To<uint32_t>(maybe_code.ToLocalChecked()).FromJust();
|
|
MaybeLocal<Value> maybe_details = Nan::Get(
|
|
server_status, Nan::New("details").ToLocalChecked());
|
|
if (maybe_details.IsEmpty()) {
|
|
return false;
|
|
}
|
|
if (!maybe_details.ToLocalChecked()->IsString()) {
|
|
return false;
|
|
}
|
|
Local<String> details = Nan::To<String>(
|
|
maybe_details.ToLocalChecked()).ToLocalChecked();
|
|
grpc_metadata_array array;
|
|
if (!CreateMetadataArray(metadata, &array, resources)) {
|
|
return false;
|
|
}
|
|
out->data.send_status_from_server.trailing_metadata_count = array.count;
|
|
out->data.send_status_from_server.trailing_metadata = array.metadata;
|
|
out->data.send_status_from_server.status =
|
|
static_cast<grpc_status_code>(code);
|
|
Utf8String *str = new Utf8String(details);
|
|
resources->strings.push_back(unique_ptr<Utf8String>(str));
|
|
out->data.send_status_from_server.status_details = **str;
|
|
return true;
|
|
}
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "send_status";
|
|
}
|
|
};
|
|
|
|
class GetMetadataOp : public Op {
|
|
public:
|
|
GetMetadataOp() {
|
|
grpc_metadata_array_init(&recv_metadata);
|
|
}
|
|
|
|
~GetMetadataOp() {
|
|
grpc_metadata_array_destroy(&recv_metadata);
|
|
}
|
|
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(ParseMetadata(&recv_metadata));
|
|
}
|
|
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
out->data.recv_initial_metadata = &recv_metadata;
|
|
return true;
|
|
}
|
|
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "metadata";
|
|
}
|
|
|
|
private:
|
|
grpc_metadata_array recv_metadata;
|
|
};
|
|
|
|
class ReadMessageOp : public Op {
|
|
public:
|
|
ReadMessageOp() {
|
|
recv_message = NULL;
|
|
}
|
|
~ReadMessageOp() {
|
|
if (recv_message != NULL) {
|
|
grpc_byte_buffer_destroy(recv_message);
|
|
}
|
|
}
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(ByteBufferToBuffer(recv_message));
|
|
}
|
|
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
out->data.recv_message = &recv_message;
|
|
return true;
|
|
}
|
|
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "read";
|
|
}
|
|
|
|
private:
|
|
grpc_byte_buffer *recv_message;
|
|
};
|
|
|
|
class ClientStatusOp : public Op {
|
|
public:
|
|
ClientStatusOp() {
|
|
grpc_metadata_array_init(&metadata_array);
|
|
status_details = NULL;
|
|
details_capacity = 0;
|
|
}
|
|
|
|
~ClientStatusOp() {
|
|
grpc_metadata_array_destroy(&metadata_array);
|
|
gpr_free(status_details);
|
|
}
|
|
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
out->data.recv_status_on_client.trailing_metadata = &metadata_array;
|
|
out->data.recv_status_on_client.status = &status;
|
|
out->data.recv_status_on_client.status_details = &status_details;
|
|
out->data.recv_status_on_client.status_details_capacity = &details_capacity;
|
|
return true;
|
|
}
|
|
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
Local<Object> status_obj = Nan::New<Object>();
|
|
Nan::Set(status_obj, Nan::New("code").ToLocalChecked(),
|
|
Nan::New<Number>(status));
|
|
if (status_details != NULL) {
|
|
Nan::Set(status_obj, Nan::New("details").ToLocalChecked(),
|
|
Nan::New(status_details).ToLocalChecked());
|
|
}
|
|
Nan::Set(status_obj, Nan::New("metadata").ToLocalChecked(),
|
|
ParseMetadata(&metadata_array));
|
|
return scope.Escape(status_obj);
|
|
}
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "status";
|
|
}
|
|
private:
|
|
grpc_metadata_array metadata_array;
|
|
grpc_status_code status;
|
|
char *status_details;
|
|
size_t details_capacity;
|
|
};
|
|
|
|
class ServerCloseResponseOp : public Op {
|
|
public:
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::New<Boolean>(cancelled));
|
|
}
|
|
|
|
bool ParseOp(Local<Value> value, grpc_op *out,
|
|
shared_ptr<Resources> resources) {
|
|
out->data.recv_close_on_server.cancelled = &cancelled;
|
|
return true;
|
|
}
|
|
|
|
protected:
|
|
std::string GetTypeString() const {
|
|
return "cancelled";
|
|
}
|
|
|
|
private:
|
|
int cancelled;
|
|
};
|
|
|
|
tag::tag(Callback *callback, OpVec *ops,
|
|
shared_ptr<Resources> resources) :
|
|
callback(callback), ops(ops), resources(resources){
|
|
}
|
|
|
|
tag::~tag() {
|
|
delete callback;
|
|
delete ops;
|
|
}
|
|
|
|
Local<Value> GetTagNodeValue(void *tag) {
|
|
EscapableHandleScope scope;
|
|
struct tag *tag_struct = reinterpret_cast<struct tag *>(tag);
|
|
Local<Object> tag_obj = Nan::New<Object>();
|
|
for (vector<unique_ptr<Op> >::iterator it = tag_struct->ops->begin();
|
|
it != tag_struct->ops->end(); ++it) {
|
|
Op *op_ptr = it->get();
|
|
Nan::Set(tag_obj, op_ptr->GetOpType(), op_ptr->GetNodeValue());
|
|
}
|
|
return scope.Escape(tag_obj);
|
|
}
|
|
|
|
Callback *GetTagCallback(void *tag) {
|
|
struct tag *tag_struct = reinterpret_cast<struct tag *>(tag);
|
|
return tag_struct->callback;
|
|
}
|
|
|
|
void DestroyTag(void *tag) {
|
|
struct tag *tag_struct = reinterpret_cast<struct tag *>(tag);
|
|
delete tag_struct;
|
|
}
|
|
|
|
Call::Call(grpc_call *call) : wrapped_call(call) {
|
|
}
|
|
|
|
Call::~Call() {
|
|
grpc_call_destroy(wrapped_call);
|
|
}
|
|
|
|
void Call::Init(Local<Object> exports) {
|
|
HandleScope scope;
|
|
Local<FunctionTemplate> tpl = Nan::New<FunctionTemplate>(New);
|
|
tpl->SetClassName(Nan::New("Call").ToLocalChecked());
|
|
tpl->InstanceTemplate()->SetInternalFieldCount(1);
|
|
Nan::SetPrototypeMethod(tpl, "startBatch", StartBatch);
|
|
Nan::SetPrototypeMethod(tpl, "cancel", Cancel);
|
|
Nan::SetPrototypeMethod(tpl, "cancelWithStatus", CancelWithStatus);
|
|
Nan::SetPrototypeMethod(tpl, "getPeer", GetPeer);
|
|
fun_tpl.Reset(tpl);
|
|
Local<Function> ctr = Nan::GetFunction(tpl).ToLocalChecked();
|
|
Nan::Set(exports, Nan::New("Call").ToLocalChecked(), ctr);
|
|
constructor = new Callback(ctr);
|
|
}
|
|
|
|
bool Call::HasInstance(Local<Value> val) {
|
|
HandleScope scope;
|
|
return Nan::New(fun_tpl)->HasInstance(val);
|
|
}
|
|
|
|
Local<Value> Call::WrapStruct(grpc_call *call) {
|
|
EscapableHandleScope scope;
|
|
if (call == NULL) {
|
|
return scope.Escape(Nan::Null());
|
|
}
|
|
const int argc = 1;
|
|
Local<Value> argv[argc] = {Nan::New<External>(
|
|
reinterpret_cast<void *>(call))};
|
|
MaybeLocal<Object> maybe_instance = Nan::NewInstance(
|
|
constructor->GetFunction(), argc, argv);
|
|
if (maybe_instance.IsEmpty()) {
|
|
return scope.Escape(Nan::Null());
|
|
} else {
|
|
return scope.Escape(maybe_instance.ToLocalChecked());
|
|
}
|
|
}
|
|
|
|
NAN_METHOD(Call::New) {
|
|
if (info.IsConstructCall()) {
|
|
Call *call;
|
|
if (info[0]->IsExternal()) {
|
|
Local<External> ext = info[0].As<External>();
|
|
// This option is used for wrapping an existing call
|
|
grpc_call *call_value =
|
|
reinterpret_cast<grpc_call *>(ext->Value());
|
|
call = new Call(call_value);
|
|
} else {
|
|
if (!Channel::HasInstance(info[0])) {
|
|
return Nan::ThrowTypeError("Call's first argument must be a Channel");
|
|
}
|
|
if (!info[1]->IsString()) {
|
|
return Nan::ThrowTypeError("Call's second argument must be a string");
|
|
}
|
|
if (!(info[2]->IsNumber() || info[2]->IsDate())) {
|
|
return Nan::ThrowTypeError(
|
|
"Call's third argument must be a date or a number");
|
|
}
|
|
// These arguments are at the end because they are optional
|
|
grpc_call *parent_call = NULL;
|
|
if (Call::HasInstance(info[4])) {
|
|
Call *parent_obj = ObjectWrap::Unwrap<Call>(
|
|
Nan::To<Object>(info[4]).ToLocalChecked());
|
|
parent_call = parent_obj->wrapped_call;
|
|
} else if (!(info[4]->IsUndefined() || info[4]->IsNull())) {
|
|
return Nan::ThrowTypeError(
|
|
"Call's fifth argument must be another call, if provided");
|
|
}
|
|
gpr_uint32 propagate_flags = GRPC_PROPAGATE_DEFAULTS;
|
|
if (info[5]->IsUint32()) {
|
|
propagate_flags = Nan::To<uint32_t>(info[5]).FromJust();
|
|
} else if (!(info[5]->IsUndefined() || info[5]->IsNull())) {
|
|
return Nan::ThrowTypeError(
|
|
"Call's sixth argument must be propagate flags, if provided");
|
|
}
|
|
Local<Object> channel_object = Nan::To<Object>(info[0]).ToLocalChecked();
|
|
Channel *channel = ObjectWrap::Unwrap<Channel>(channel_object);
|
|
if (channel->GetWrappedChannel() == NULL) {
|
|
return Nan::ThrowError("Call cannot be created from a closed channel");
|
|
}
|
|
Utf8String method(info[1]);
|
|
double deadline = Nan::To<double>(info[2]).FromJust();
|
|
grpc_channel *wrapped_channel = channel->GetWrappedChannel();
|
|
grpc_call *wrapped_call;
|
|
if (info[3]->IsString()) {
|
|
Utf8String host_override(info[3]);
|
|
wrapped_call = grpc_channel_create_call(
|
|
wrapped_channel, parent_call, propagate_flags,
|
|
CompletionQueueAsyncWorker::GetQueue(), *method,
|
|
*host_override, MillisecondsToTimespec(deadline), NULL);
|
|
} else if (info[3]->IsUndefined() || info[3]->IsNull()) {
|
|
wrapped_call = grpc_channel_create_call(
|
|
wrapped_channel, parent_call, propagate_flags,
|
|
CompletionQueueAsyncWorker::GetQueue(), *method,
|
|
NULL, MillisecondsToTimespec(deadline), NULL);
|
|
} else {
|
|
return Nan::ThrowTypeError("Call's fourth argument must be a string");
|
|
}
|
|
call = new Call(wrapped_call);
|
|
info.This()->SetHiddenValue(Nan::New("channel_").ToLocalChecked(),
|
|
channel_object);
|
|
}
|
|
call->Wrap(info.This());
|
|
info.GetReturnValue().Set(info.This());
|
|
} else {
|
|
const int argc = 4;
|
|
Local<Value> argv[argc] = {info[0], info[1], info[2], info[3]};
|
|
MaybeLocal<Object> maybe_instance = constructor->GetFunction()->NewInstance(
|
|
argc, argv);
|
|
if (maybe_instance.IsEmpty()) {
|
|
// There's probably a pending exception
|
|
return;
|
|
} else {
|
|
info.GetReturnValue().Set(maybe_instance.ToLocalChecked());
|
|
}
|
|
}
|
|
}
|
|
|
|
NAN_METHOD(Call::StartBatch) {
|
|
if (!Call::HasInstance(info.This())) {
|
|
return Nan::ThrowTypeError("startBatch can only be called on Call objects");
|
|
}
|
|
if (!info[0]->IsObject()) {
|
|
return Nan::ThrowError("startBatch's first argument must be an object");
|
|
}
|
|
if (!info[1]->IsFunction()) {
|
|
return Nan::ThrowError("startBatch's second argument must be a callback");
|
|
}
|
|
Local<Function> callback_func = info[1].As<Function>();
|
|
Call *call = ObjectWrap::Unwrap<Call>(info.This());
|
|
shared_ptr<Resources> resources(new Resources);
|
|
Local<Object> obj = Nan::To<Object>(info[0]).ToLocalChecked();
|
|
Local<Array> keys = Nan::GetOwnPropertyNames(obj).ToLocalChecked();
|
|
size_t nops = keys->Length();
|
|
vector<grpc_op> ops(nops);
|
|
unique_ptr<OpVec> op_vector(new OpVec());
|
|
for (unsigned int i = 0; i < nops; i++) {
|
|
unique_ptr<Op> op;
|
|
MaybeLocal<Value> maybe_key = Nan::Get(keys, i);
|
|
if (maybe_key.IsEmpty() || (!maybe_key.ToLocalChecked()->IsUint32())) {
|
|
return Nan::ThrowError(
|
|
"startBatch's first argument's keys must be integers");
|
|
}
|
|
uint32_t type = Nan::To<uint32_t>(maybe_key.ToLocalChecked()).FromJust();
|
|
ops[i].op = static_cast<grpc_op_type>(type);
|
|
ops[i].flags = 0;
|
|
ops[i].reserved = NULL;
|
|
switch (type) {
|
|
case GRPC_OP_SEND_INITIAL_METADATA:
|
|
op.reset(new SendMetadataOp());
|
|
break;
|
|
case GRPC_OP_SEND_MESSAGE:
|
|
op.reset(new SendMessageOp());
|
|
break;
|
|
case GRPC_OP_SEND_CLOSE_FROM_CLIENT:
|
|
op.reset(new SendClientCloseOp());
|
|
break;
|
|
case GRPC_OP_SEND_STATUS_FROM_SERVER:
|
|
op.reset(new SendServerStatusOp());
|
|
break;
|
|
case GRPC_OP_RECV_INITIAL_METADATA:
|
|
op.reset(new GetMetadataOp());
|
|
break;
|
|
case GRPC_OP_RECV_MESSAGE:
|
|
op.reset(new ReadMessageOp());
|
|
break;
|
|
case GRPC_OP_RECV_STATUS_ON_CLIENT:
|
|
op.reset(new ClientStatusOp());
|
|
break;
|
|
case GRPC_OP_RECV_CLOSE_ON_SERVER:
|
|
op.reset(new ServerCloseResponseOp());
|
|
break;
|
|
default:
|
|
return Nan::ThrowError("Argument object had an unrecognized key");
|
|
}
|
|
if (!op->ParseOp(obj->Get(type), &ops[i], resources)) {
|
|
return Nan::ThrowTypeError("Incorrectly typed arguments to startBatch");
|
|
}
|
|
op_vector->push_back(std::move(op));
|
|
}
|
|
Callback *callback = new Callback(callback_func);
|
|
grpc_call_error error = grpc_call_start_batch(
|
|
call->wrapped_call, &ops[0], nops, new struct tag(
|
|
callback, op_vector.release(), resources), NULL);
|
|
if (error != GRPC_CALL_OK) {
|
|
return Nan::ThrowError(nanErrorWithCode("startBatch failed", error));
|
|
}
|
|
CompletionQueueAsyncWorker::Next();
|
|
}
|
|
|
|
NAN_METHOD(Call::Cancel) {
|
|
if (!Call::HasInstance(info.This())) {
|
|
return Nan::ThrowTypeError("cancel can only be called on Call objects");
|
|
}
|
|
Call *call = ObjectWrap::Unwrap<Call>(info.This());
|
|
grpc_call_error error = grpc_call_cancel(call->wrapped_call, NULL);
|
|
if (error != GRPC_CALL_OK) {
|
|
return Nan::ThrowError(nanErrorWithCode("cancel failed", error));
|
|
}
|
|
}
|
|
|
|
NAN_METHOD(Call::CancelWithStatus) {
|
|
Nan::HandleScope scope;
|
|
if (!HasInstance(info.This())) {
|
|
return Nan::ThrowTypeError("cancel can only be called on Call objects");
|
|
}
|
|
if (!info[0]->IsUint32()) {
|
|
return Nan::ThrowTypeError(
|
|
"cancelWithStatus's first argument must be a status code");
|
|
}
|
|
if (!info[1]->IsString()) {
|
|
return Nan::ThrowTypeError(
|
|
"cancelWithStatus's second argument must be a string");
|
|
}
|
|
Call *call = ObjectWrap::Unwrap<Call>(info.This());
|
|
grpc_status_code code = static_cast<grpc_status_code>(
|
|
Nan::To<uint32_t>(info[0]).FromJust());
|
|
Utf8String details(info[0]);
|
|
grpc_call_cancel_with_status(call->wrapped_call, code, *details, NULL);
|
|
}
|
|
|
|
NAN_METHOD(Call::GetPeer) {
|
|
Nan::HandleScope scope;
|
|
if (!HasInstance(info.This())) {
|
|
return Nan::ThrowTypeError("getPeer can only be called on Call objects");
|
|
}
|
|
Call *call = ObjectWrap::Unwrap<Call>(info.This());
|
|
char *peer = grpc_call_get_peer(call->wrapped_call);
|
|
Local<Value> peer_value = Nan::New(peer).ToLocalChecked();
|
|
gpr_free(peer);
|
|
info.GetReturnValue().Set(peer_value);
|
|
}
|
|
|
|
} // namespace node
|
|
} // namespace grpc
|