| /* |
| * |
| * 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. |
| * |
| */ |
| |
| 'use strict'; |
| |
| var grpc = require('grpc'); |
| |
| var _ = require('lodash'); |
| |
| var health_messages = require('./v1/health_pb'); |
| var health_service = require('./v1/health_grpc_pb'); |
| |
| function HealthImplementation(statusMap) { |
| this.statusMap = _.clone(statusMap); |
| } |
| |
| HealthImplementation.prototype.setStatus = function(service, status) { |
| this.statusMap[service] = status; |
| }; |
| |
| HealthImplementation.prototype.check = function(call, callback){ |
| var service = call.request.getService(); |
| var status = _.get(this.statusMap, service, null); |
| if (status === null) { |
| callback({code:grpc.status.NOT_FOUND}); |
| } else { |
| var response = new health_messages.HealthCheckResponse(); |
| response.setStatus(status); |
| callback(null, response); |
| } |
| }; |
| |
| module.exports = { |
| Client: health_service.HealthClient, |
| service: health_service.HealthService, |
| Implementation: HealthImplementation |
| }; |