mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-11-10 15:04:45 +00:00
761 lines
23 KiB
C++
761 lines
23 KiB
C++
/*
|
|
*
|
|
* Copyright 2015 gRPC authors.
|
|
*
|
|
* Licensed 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.
|
|
*
|
|
*/
|
|
|
|
#include <map>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include <node.h>
|
|
|
|
#include "byte_buffer.h"
|
|
#include "call.h"
|
|
#include "call_credentials.h"
|
|
#include "channel.h"
|
|
#include "completion_queue.h"
|
|
#include "grpc/grpc.h"
|
|
#include "grpc/grpc_security.h"
|
|
#include "grpc/support/alloc.h"
|
|
#include "grpc/support/log.h"
|
|
#include "grpc/support/time.h"
|
|
#include "slice.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;
|
|
|
|
/**
|
|
* Helper function for throwing errors with a grpc_call_error value.
|
|
* Modified from the answer by Gus Goose to
|
|
* http://stackoverflow.com/questions/31794200.
|
|
*/
|
|
Local<Value> nanErrorWithCode(const char *msg, grpc_call_error code) {
|
|
EscapableHandleScope scope;
|
|
Local<Object> err = Nan::Error(msg).As<Object>();
|
|
Nan::Set(err, Nan::New("code").ToLocalChecked(), Nan::New<Uint32>(code));
|
|
return scope.Escape(err);
|
|
}
|
|
|
|
bool CreateMetadataArray(Local<Object> metadata, grpc_metadata_array *array) {
|
|
HandleScope scope;
|
|
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_zalloc(array->capacity * sizeof(grpc_metadata)));
|
|
for (unsigned int i = 0; i < keys->Length(); i++) {
|
|
Local<String> current_key(Nan::To<String>(keys->Get(i)).ToLocalChecked());
|
|
Local<Array> values =
|
|
Local<Array>::Cast(Nan::Get(metadata, current_key).ToLocalChecked());
|
|
grpc_slice key_slice = CreateSliceFromString(current_key);
|
|
grpc_slice key_intern_slice = grpc_slice_intern(key_slice);
|
|
grpc_slice_unref(key_slice);
|
|
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 = key_intern_slice;
|
|
// Only allow binary headers for "-bin" keys
|
|
if (grpc_is_binary_header(key_intern_slice)) {
|
|
if (::node::Buffer::HasInstance(value)) {
|
|
current->value = CreateSliceFromBuffer(value);
|
|
} else {
|
|
return false;
|
|
}
|
|
} else {
|
|
if (value->IsString()) {
|
|
Local<String> string_value = Nan::To<String>(value).ToLocalChecked();
|
|
current->value = CreateSliceFromString(string_value);
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
array->count += 1;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void DestroyMetadataArray(grpc_metadata_array *array) {
|
|
for (size_t i = 0; i < array->count; i++) {
|
|
// Don't unref keys because they are interned
|
|
grpc_slice_unref(array->metadata[i].value);
|
|
}
|
|
grpc_metadata_array_destroy(array);
|
|
}
|
|
|
|
Local<Value> ParseMetadata(const grpc_metadata_array *metadata_array) {
|
|
EscapableHandleScope scope;
|
|
grpc_metadata *metadata_elements = metadata_array->metadata;
|
|
size_t length = metadata_array->count;
|
|
Local<Object> metadata_object = Nan::New<Object>();
|
|
for (unsigned int i = 0; i < length; i++) {
|
|
grpc_metadata *elem = &metadata_elements[i];
|
|
// TODO(murgatroid99): Use zero-copy string construction instead
|
|
Local<String> key_string = CopyStringFromSlice(elem->key);
|
|
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>(0);
|
|
Nan::Set(metadata_object, key_string, array);
|
|
} else {
|
|
array = Local<Array>::Cast(maybe_array.ToLocalChecked());
|
|
}
|
|
if (grpc_is_binary_header(elem->key)) {
|
|
Nan::Set(array, array->Length(), CreateBufferFromSlice(elem->value));
|
|
} else {
|
|
// TODO(murgatroid99): Use zero-copy string construction instead
|
|
Nan::Set(array, array->Length(), CopyStringFromSlice(elem->value));
|
|
}
|
|
}
|
|
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:
|
|
SendMetadataOp() { grpc_metadata_array_init(&send_metadata); }
|
|
~SendMetadataOp() { DestroyMetadataArray(&send_metadata); }
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::True());
|
|
}
|
|
bool ParseOp(Local<Value> value, grpc_op *out) {
|
|
if (!value->IsObject()) {
|
|
return false;
|
|
}
|
|
MaybeLocal<Object> maybe_metadata = Nan::To<Object>(value);
|
|
if (maybe_metadata.IsEmpty()) {
|
|
return false;
|
|
}
|
|
if (!CreateMetadataArray(maybe_metadata.ToLocalChecked(), &send_metadata)) {
|
|
return false;
|
|
}
|
|
out->data.send_initial_metadata.count = send_metadata.count;
|
|
out->data.send_initial_metadata.metadata = send_metadata.metadata;
|
|
return true;
|
|
}
|
|
bool IsFinalOp() { return false; }
|
|
void OnComplete(bool success) {}
|
|
|
|
protected:
|
|
std::string GetTypeString() const { return "send_metadata"; }
|
|
|
|
private:
|
|
grpc_metadata_array send_metadata;
|
|
};
|
|
|
|
class SendMessageOp : public Op {
|
|
public:
|
|
SendMessageOp() { send_message = NULL; }
|
|
~SendMessageOp() {
|
|
if (send_message != NULL) {
|
|
grpc_byte_buffer_destroy(send_message);
|
|
}
|
|
}
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::True());
|
|
}
|
|
bool ParseOp(Local<Value> value, grpc_op *out) {
|
|
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;
|
|
}
|
|
}
|
|
send_message = BufferToByteBuffer(value);
|
|
out->data.send_message.send_message = send_message;
|
|
return true;
|
|
}
|
|
|
|
bool IsFinalOp() { return false; }
|
|
void OnComplete(bool success) {}
|
|
|
|
protected:
|
|
std::string GetTypeString() const { return "send_message"; }
|
|
|
|
private:
|
|
grpc_byte_buffer *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) { return true; }
|
|
bool IsFinalOp() { return false; }
|
|
void OnComplete(bool success) {}
|
|
|
|
protected:
|
|
std::string GetTypeString() const { return "client_close"; }
|
|
};
|
|
|
|
class SendServerStatusOp : public Op {
|
|
public:
|
|
SendServerStatusOp() {
|
|
details = grpc_empty_slice();
|
|
grpc_metadata_array_init(&status_metadata);
|
|
}
|
|
~SendServerStatusOp() {
|
|
grpc_slice_unref(details);
|
|
DestroyMetadataArray(&status_metadata);
|
|
}
|
|
Local<Value> GetNodeValue() const {
|
|
EscapableHandleScope scope;
|
|
return scope.Escape(Nan::True());
|
|
}
|
|
bool ParseOp(Local<Value> value, grpc_op *out) {
|
|
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();
|
|
if (!CreateMetadataArray(metadata, &status_metadata)) {
|
|
return false;
|
|
}
|
|
out->data.send_status_from_server.trailing_metadata_count =
|
|
status_metadata.count;
|
|
out->data.send_status_from_server.trailing_metadata =
|
|
status_metadata.metadata;
|
|
out->data.send_status_from_server.status =
|
|
static_cast<grpc_status_code>(code);
|
|
this->details = CreateSliceFromString(details);
|
|
out->data.send_status_from_server.status_details = &this->details;
|
|
return true;
|
|
}
|
|
bool IsFinalOp() { return true; }
|
|
void OnComplete(bool success) {}
|
|
|
|
protected:
|
|
std::string GetTypeString() const { return "send_status"; }
|
|
|
|
private:
|
|
grpc_slice details;
|
|
grpc_metadata_array status_metadata;
|
|
};
|
|
|
|
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) {
|
|
out->data.recv_initial_metadata.recv_initial_metadata = &recv_metadata;
|
|
return true;
|
|
}
|
|
bool IsFinalOp() { return false; }
|
|
void OnComplete(bool success) {}
|
|
|
|
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) {
|
|
out->data.recv_message.recv_message = &recv_message;
|
|
return true;
|
|
}
|
|
bool IsFinalOp() { return false; }
|
|
void OnComplete(bool success) {}
|
|
|
|
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 = grpc_empty_slice();
|
|
}
|
|
|
|
~ClientStatusOp() {
|
|
grpc_metadata_array_destroy(&metadata_array);
|
|
grpc_slice_unref(status_details);
|
|
}
|
|
|
|
bool ParseOp(Local<Value> value, grpc_op *out) {
|
|
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;
|
|
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));
|
|
Nan::Set(status_obj, Nan::New("details").ToLocalChecked(),
|
|
CopyStringFromSlice(status_details));
|
|
Nan::Set(status_obj, Nan::New("metadata").ToLocalChecked(),
|
|
ParseMetadata(&metadata_array));
|
|
return scope.Escape(status_obj);
|
|
}
|
|
bool IsFinalOp() { return true; }
|
|
void OnComplete(bool success) {}
|
|
|
|
protected:
|
|
std::string GetTypeString() const { return "status"; }
|
|
|
|
private:
|
|
grpc_metadata_array metadata_array;
|
|
grpc_status_code status;
|
|
grpc_slice status_details;
|
|
};
|
|
|
|
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) {
|
|
out->data.recv_close_on_server.cancelled = &cancelled;
|
|
return true;
|
|
}
|
|
bool IsFinalOp() { return false; }
|
|
void OnComplete(bool success) {}
|
|
|
|
protected:
|
|
std::string GetTypeString() const { return "cancelled"; }
|
|
|
|
private:
|
|
int cancelled;
|
|
};
|
|
|
|
tag::tag(Callback *callback, OpVec *ops, Call *call, Local<Value> call_value)
|
|
: callback(callback),
|
|
async_resource(NULL),
|
|
ops(ops),
|
|
call(call) {
|
|
HandleScope scope;
|
|
async_resource = new Nan::AsyncResource("grpc:tag"); // Needs handle scope.
|
|
call_persist.Reset(call_value);
|
|
}
|
|
|
|
tag::~tag() {
|
|
delete callback;
|
|
delete async_resource;
|
|
delete ops;
|
|
}
|
|
|
|
void CompleteTag(void *tag, const char *error_message) {
|
|
HandleScope scope;
|
|
struct tag *tag_struct = reinterpret_cast<struct tag *>(tag);
|
|
Callback *callback = tag_struct->callback;
|
|
if (error_message == NULL) {
|
|
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());
|
|
}
|
|
Local<Value> argv[] = {Nan::Null(), tag_obj};
|
|
callback->Call(2, argv, tag_struct->async_resource);
|
|
} else {
|
|
Local<Value> argv[] = {Nan::Error(error_message)};
|
|
callback->Call(1, argv, tag_struct->async_resource);
|
|
}
|
|
bool success = (error_message == NULL);
|
|
bool is_final_op = false;
|
|
for (vector<unique_ptr<Op> >::iterator it = tag_struct->ops->begin();
|
|
it != tag_struct->ops->end(); ++it) {
|
|
Op *op_ptr = it->get();
|
|
op_ptr->OnComplete(success);
|
|
if (op_ptr->IsFinalOp()) {
|
|
is_final_op = true;
|
|
}
|
|
}
|
|
if (tag_struct->call == NULL) {
|
|
return;
|
|
}
|
|
tag_struct->call->CompleteBatch(is_final_op);
|
|
}
|
|
|
|
void DestroyTag(void *tag) {
|
|
struct tag *tag_struct = reinterpret_cast<struct tag *>(tag);
|
|
delete tag_struct;
|
|
}
|
|
|
|
void Call::DestroyCall() {
|
|
if (this->wrapped_call != NULL) {
|
|
grpc_call_unref(this->wrapped_call);
|
|
this->wrapped_call = NULL;
|
|
}
|
|
}
|
|
|
|
Call::Call(grpc_call *call)
|
|
: wrapped_call(call), pending_batches(0), has_final_op_completed(false) {
|
|
peer = grpc_call_get_peer(call);
|
|
}
|
|
|
|
Call::~Call() {
|
|
DestroyCall();
|
|
gpr_free(peer);
|
|
}
|
|
|
|
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);
|
|
Nan::SetPrototypeMethod(tpl, "setCredentials", SetCredentials);
|
|
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);
|
|
}
|
|
|
|
grpc_call *Call::GetWrappedCall() { return this->wrapped_call; }
|
|
|
|
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());
|
|
}
|
|
}
|
|
|
|
void Call::CompleteBatch(bool is_final_op) {
|
|
if (is_final_op) {
|
|
this->has_final_op_completed = true;
|
|
}
|
|
this->pending_batches--;
|
|
if (this->has_final_op_completed && this->pending_batches == 0) {
|
|
this->DestroyCall();
|
|
}
|
|
}
|
|
|
|
NAN_METHOD(Call::New) {
|
|
/* Arguments:
|
|
* 0: Channel to make the call on
|
|
* 1: Method
|
|
* 2: Deadline
|
|
* 3: host
|
|
* 4: parent Call
|
|
* 5: propagation flags
|
|
*/
|
|
if (info.IsConstructCall()) {
|
|
Call *call;
|
|
if (!info[0]->IsExternal()) {
|
|
return Nan::ThrowTypeError(
|
|
"Call can only be created with Channel.createCall");
|
|
}
|
|
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);
|
|
call->Wrap(info.This());
|
|
info.GetReturnValue().Set(info.This());
|
|
return;
|
|
} else {
|
|
return Nan::ThrowTypeError(
|
|
"Call can only be created with Channel.createCall");
|
|
}
|
|
}
|
|
|
|
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());
|
|
if (call->wrapped_call == NULL) {
|
|
/* This implies that the call has completed and has been destroyed. To
|
|
* emulate
|
|
* previous behavior, we should call the callback immediately with an error,
|
|
* as though the batch had failed in core */
|
|
Local<Value> argv[] = {
|
|
Nan::Error("The async function failed because the call has completed")};
|
|
Nan::Call(callback_func, Nan::New<Object>(), 1, argv);
|
|
return;
|
|
}
|
|
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])) {
|
|
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(), call, info.This()), NULL);
|
|
if (error != GRPC_CALL_OK) {
|
|
return Nan::ThrowError(nanErrorWithCode("startBatch failed", error));
|
|
}
|
|
call->pending_batches++;
|
|
CompletionQueueNext();
|
|
}
|
|
|
|
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());
|
|
if (call->wrapped_call == NULL) {
|
|
/* Cancel is supposed to be idempotent. If the call has already finished,
|
|
* cancel should just complete silently */
|
|
return;
|
|
}
|
|
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());
|
|
if (call->wrapped_call == NULL) {
|
|
/* Cancel is supposed to be idempotent. If the call has already finished,
|
|
* cancel should just complete silently */
|
|
return;
|
|
}
|
|
grpc_status_code code =
|
|
static_cast<grpc_status_code>(Nan::To<uint32_t>(info[0]).FromJust());
|
|
if (code == GRPC_STATUS_OK) {
|
|
return Nan::ThrowRangeError(
|
|
"cancelWithStatus cannot be called with OK status");
|
|
}
|
|
Utf8String details(info[1]);
|
|
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());
|
|
Local<Value> peer_value = Nan::New(call->peer).ToLocalChecked();
|
|
info.GetReturnValue().Set(peer_value);
|
|
}
|
|
|
|
NAN_METHOD(Call::SetCredentials) {
|
|
Nan::HandleScope scope;
|
|
if (!HasInstance(info.This())) {
|
|
return Nan::ThrowTypeError(
|
|
"setCredentials can only be called on Call objects");
|
|
}
|
|
if (!CallCredentials::HasInstance(info[0])) {
|
|
return Nan::ThrowTypeError(
|
|
"setCredentials' first argument must be a CallCredentials");
|
|
}
|
|
Call *call = ObjectWrap::Unwrap<Call>(info.This());
|
|
if (call->wrapped_call == NULL) {
|
|
return Nan::ThrowError(
|
|
"Cannot set credentials on a call that has already started");
|
|
}
|
|
CallCredentials *creds_object = ObjectWrap::Unwrap<CallCredentials>(
|
|
Nan::To<Object>(info[0]).ToLocalChecked());
|
|
grpc_call_credentials *creds = creds_object->GetWrappedCredentials();
|
|
grpc_call_error error = GRPC_CALL_ERROR;
|
|
if (creds) {
|
|
error = grpc_call_set_credentials(call->wrapped_call, creds);
|
|
}
|
|
info.GetReturnValue().Set(Nan::New<Uint32>(error));
|
|
}
|
|
|
|
} // namespace node
|
|
} // namespace grpc
|