| /* |
| * |
| * Copyright 2014, 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. |
| * |
| */ |
| |
| var _ = require('underscore'); |
| var ProtoBuf = require('protobufjs'); |
| var fs = require('fs'); |
| var util = require('util'); |
| |
| var Transform = require('stream').Transform; |
| |
| var grpc = require('..'); |
| var math = grpc.load(__dirname + '/math.proto').math; |
| |
| var Server = grpc.buildServer([math.Math.service]); |
| |
| /** |
| * Server function for division. Provides the /Math/DivMany and /Math/Div |
| * functions (Div is just DivMany with only one stream element). For each |
| * DivArgs parameter, responds with a DivReply with the results of the division |
| * @param {Object} call The object containing request and cancellation info |
| * @param {function(Error, *)} cb Response callback |
| */ |
| function mathDiv(call, cb) { |
| var req = call.request; |
| if (req.divisor == 0) { |
| cb(new Error('cannot divide by zero')); |
| } |
| cb(null, { |
| quotient: req.dividend / req.divisor, |
| remainder: req.dividend % req.divisor |
| }); |
| } |
| |
| /** |
| * Server function for Fibonacci numbers. Provides the /Math/Fib function. Reads |
| * a single parameter that indicates the number of responses, and then responds |
| * with a stream of that many Fibonacci numbers. |
| * @param {stream} stream The stream for sending responses. |
| */ |
| function mathFib(stream) { |
| // Here, call is a standard writable Node object Stream |
| var previous = 0, current = 1; |
| for (var i = 0; i < stream.request.limit; i++) { |
| stream.write({num: current}); |
| var temp = current; |
| current += previous; |
| previous = temp; |
| } |
| stream.end(); |
| } |
| |
| /** |
| * Server function for summation. Provides the /Math/Sum function. Reads a |
| * stream of number parameters, then responds with their sum. |
| * @param {stream} call The stream of arguments. |
| * @param {function(Error, *)} cb Response callback |
| */ |
| function mathSum(call, cb) { |
| // Here, call is a standard readable Node object Stream |
| var sum = 0; |
| call.on('data', function(data) { |
| sum += data.num | 0; |
| }); |
| call.on('end', function() { |
| cb(null, {num: sum}); |
| }); |
| } |
| |
| function mathDivMany(stream) { |
| // Here, call is a standard duplex Node object Stream |
| util.inherits(DivTransform, Transform); |
| function DivTransform() { |
| var options = {objectMode: true}; |
| Transform.call(this, options); |
| } |
| DivTransform.prototype._transform = function(div_args, encoding, callback) { |
| if (div_args.divisor == 0) { |
| callback(new Error('cannot divide by zero')); |
| } |
| callback(null, { |
| quotient: div_args.dividend / div_args.divisor, |
| remainder: div_args.dividend % div_args.divisor |
| }); |
| }; |
| var transform = new DivTransform(); |
| stream.pipe(transform); |
| transform.pipe(stream); |
| } |
| |
| var server = new Server({ |
| 'math.Math' : { |
| Div: mathDiv, |
| Fib: mathFib, |
| Sum: mathSum, |
| DivMany: mathDivMany |
| } |
| }); |
| |
| if (require.main === module) { |
| server.bind('localhost:7070').listen(); |
| } |
| |
| /** |
| * See docs for server |
| */ |
| module.exports = server; |