mirror of
https://github.com/grpc/grpc-node.git
synced 2025-12-08 18:23:54 +00:00
Resolved merge conflict with upstream/master
This commit is contained in:
commit
be555df5b0
49
client.js
49
client.js
@ -45,10 +45,22 @@ util.inherits(GrpcClientStream, Duplex);
|
||||
* from stream.Duplex.
|
||||
* @constructor
|
||||
* @param {grpc.Call} call Call object to proxy
|
||||
* @param {object} options Stream options
|
||||
* @param {function(*):Buffer=} serialize Serialization function for requests
|
||||
* @param {function(Buffer):*=} deserialize Deserialization function for
|
||||
* responses
|
||||
*/
|
||||
function GrpcClientStream(call, options) {
|
||||
Duplex.call(this, options);
|
||||
function GrpcClientStream(call, serialize, deserialize) {
|
||||
Duplex.call(this, {objectMode: true});
|
||||
if (!serialize) {
|
||||
serialize = function(value) {
|
||||
return value;
|
||||
};
|
||||
}
|
||||
if (!deserialize) {
|
||||
deserialize = function(value) {
|
||||
return value;
|
||||
};
|
||||
}
|
||||
var self = this;
|
||||
var finished = false;
|
||||
// Indicates that a read is currently pending
|
||||
@ -56,6 +68,32 @@ function GrpcClientStream(call, options) {
|
||||
// Indicates that a write is currently pending
|
||||
var writing = false;
|
||||
this._call = call;
|
||||
|
||||
/**
|
||||
* Serialize a request value to a buffer. Always maps null to null. Otherwise
|
||||
* uses the provided serialize function
|
||||
* @param {*} value The value to serialize
|
||||
* @return {Buffer} The serialized value
|
||||
*/
|
||||
this.serialize = function(value) {
|
||||
if (value === null || value === undefined) {
|
||||
return null;
|
||||
}
|
||||
return serialize(value);
|
||||
};
|
||||
|
||||
/**
|
||||
* Deserialize a response buffer to a value. Always maps null to null.
|
||||
* Otherwise uses the provided deserialize function.
|
||||
* @param {Buffer} buffer The buffer to deserialize
|
||||
* @return {*} The deserialized value
|
||||
*/
|
||||
this.deserialize = function(buffer) {
|
||||
if (buffer === null) {
|
||||
return null;
|
||||
}
|
||||
return deserialize(buffer);
|
||||
};
|
||||
/**
|
||||
* Callback to be called when a READ event is received. Pushes the data onto
|
||||
* the read queue and starts reading again if applicable
|
||||
@ -126,6 +164,9 @@ GrpcClientStream.prototype._write = function(chunk, encoding, callback) {
|
||||
* Make a request on the channel to the given method with the given arguments
|
||||
* @param {grpc.Channel} channel The channel on which to make the request
|
||||
* @param {string} method The method to request
|
||||
* @param {function(*):Buffer} serialize Serialization function for requests
|
||||
* @param {function(Buffer):*} deserialize Deserialization function for
|
||||
* responses
|
||||
* @param {array=} metadata Array of metadata key/value pairs to add to the call
|
||||
* @param {(number|Date)=} deadline The deadline for processing this request.
|
||||
* Defaults to infinite future.
|
||||
@ -133,6 +174,8 @@ GrpcClientStream.prototype._write = function(chunk, encoding, callback) {
|
||||
*/
|
||||
function makeRequest(channel,
|
||||
method,
|
||||
serialize,
|
||||
deserialize,
|
||||
metadata,
|
||||
deadline) {
|
||||
if (deadline === undefined) {
|
||||
|
||||
65
server.js
65
server.js
@ -47,10 +47,22 @@ util.inherits(GrpcServerStream, Duplex);
|
||||
* from stream.Duplex.
|
||||
* @constructor
|
||||
* @param {grpc.Call} call Call object to proxy
|
||||
* @param {object} options Stream options
|
||||
* @param {function(*):Buffer=} serialize Serialization function for responses
|
||||
* @param {function(Buffer):*=} deserialize Deserialization function for
|
||||
* requests
|
||||
*/
|
||||
function GrpcServerStream(call, options) {
|
||||
Duplex.call(this, options);
|
||||
function GrpcServerStream(call, serialize, deserialize) {
|
||||
Duplex.call(this, {objectMode: true});
|
||||
if (!serialize) {
|
||||
serialize = function(value) {
|
||||
return value;
|
||||
};
|
||||
}
|
||||
if (!deserialize) {
|
||||
deserialize = function(value) {
|
||||
return value;
|
||||
};
|
||||
}
|
||||
this._call = call;
|
||||
// Indicate that a status has been sent
|
||||
var finished = false;
|
||||
@ -59,6 +71,33 @@ function GrpcServerStream(call, options) {
|
||||
'code' : grpc.status.OK,
|
||||
'details' : 'OK'
|
||||
};
|
||||
|
||||
/**
|
||||
* Serialize a response value to a buffer. Always maps null to null. Otherwise
|
||||
* uses the provided serialize function
|
||||
* @param {*} value The value to serialize
|
||||
* @return {Buffer} The serialized value
|
||||
*/
|
||||
this.serialize = function(value) {
|
||||
if (value === null || value === undefined) {
|
||||
return null;
|
||||
}
|
||||
return serialize(value);
|
||||
};
|
||||
|
||||
/**
|
||||
* Deserialize a request buffer to a value. Always maps null to null.
|
||||
* Otherwise uses the provided deserialize function.
|
||||
* @param {Buffer} buffer The buffer to deserialize
|
||||
* @return {*} The deserialized value
|
||||
*/
|
||||
this.deserialize = function(buffer) {
|
||||
if (buffer === null) {
|
||||
return null;
|
||||
}
|
||||
return deserialize(buffer);
|
||||
};
|
||||
|
||||
/**
|
||||
* Send the pending status
|
||||
*/
|
||||
@ -75,7 +114,6 @@ function GrpcServerStream(call, options) {
|
||||
* @param {Error} err The error object
|
||||
*/
|
||||
function setStatus(err) {
|
||||
console.log('Server setting status to', err);
|
||||
var code = grpc.status.INTERNAL;
|
||||
var details = 'Unknown Error';
|
||||
|
||||
@ -113,7 +151,7 @@ function GrpcServerStream(call, options) {
|
||||
return;
|
||||
}
|
||||
var data = event.data;
|
||||
if (self.push(data) && data != null) {
|
||||
if (self.push(deserialize(data)) && data != null) {
|
||||
self._call.startRead(readCallback);
|
||||
} else {
|
||||
reading = false;
|
||||
@ -155,7 +193,7 @@ GrpcServerStream.prototype._read = function(size) {
|
||||
*/
|
||||
GrpcServerStream.prototype._write = function(chunk, encoding, callback) {
|
||||
var self = this;
|
||||
self._call.startWrite(chunk, function(event) {
|
||||
self._call.startWrite(self.serialize(chunk), function(event) {
|
||||
callback();
|
||||
}, 0);
|
||||
};
|
||||
@ -211,12 +249,13 @@ function Server(options) {
|
||||
}
|
||||
}, 0);
|
||||
call.serverEndInitialMetadata(0);
|
||||
var stream = new GrpcServerStream(call);
|
||||
var stream = new GrpcServerStream(call, handler.serialize,
|
||||
handler.deserialize);
|
||||
Object.defineProperty(stream, 'cancelled', {
|
||||
get: function() { return cancelled;}
|
||||
});
|
||||
try {
|
||||
handler(stream, data.metadata);
|
||||
handler.func(stream, data.metadata);
|
||||
} catch (e) {
|
||||
stream.emit('error', e);
|
||||
}
|
||||
@ -237,14 +276,20 @@ function Server(options) {
|
||||
* handle/respond to.
|
||||
* @param {function} handler Function that takes a stream of request values and
|
||||
* returns a stream of response values
|
||||
* @param {function(*):Buffer} serialize Serialization function for responses
|
||||
* @param {function(Buffer):*} deserialize Deserialization function for requests
|
||||
* @return {boolean} True if the handler was set. False if a handler was already
|
||||
* set for that name.
|
||||
*/
|
||||
Server.prototype.register = function(name, handler) {
|
||||
Server.prototype.register = function(name, handler, serialize, deserialize) {
|
||||
if (this.handlers.hasOwnProperty(name)) {
|
||||
return false;
|
||||
}
|
||||
this.handlers[name] = handler;
|
||||
this.handlers[name] = {
|
||||
func: handler,
|
||||
serialize: serialize,
|
||||
deserialize: deserialize
|
||||
};
|
||||
return true;
|
||||
};
|
||||
|
||||
|
||||
@ -49,7 +49,7 @@ describe('Interop tests', function() {
|
||||
done();
|
||||
});
|
||||
// This depends on not using a binary stream
|
||||
it.skip('should pass empty_unary', function(done) {
|
||||
it('should pass empty_unary', function(done) {
|
||||
interop_client.runTest(port, name_override, 'empty_unary', true, done);
|
||||
});
|
||||
it('should pass large_unary', function(done) {
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user