include/grpcpp/impl/server_initializer.h

   1 /*
   2  *
   3  * Copyright 2016 gRPC authors.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  */
  18
  19 #ifndef GRPCPP_IMPL_SERVER_INITIALIZER_H
  20 #define GRPCPP_IMPL_SERVER_INITIALIZER_H
  21
  22 #include <memory>
  23 #include <vector>
  24
  25 #include <grpcpp/server.h>
  26
  27 namespace grpc {
  28
  29 class Server;
  30 class Service;
  31
  32 class ServerInitializer {
  33  public:
  34   ServerInitializer(Server* server) : server_(server) {}
  35
  36   bool RegisterService(std::shared_ptr<Service> service) {
  37     if (!server_->RegisterService(nullptr, service.get())) {
  38       return false;
  39     }
  40     default_services_.push_back(service);
  41     return true;
  42   }
  43
  44   const std::vector<grpc::string>* GetServiceList() {
  45     return &server_->services_;
  46   }
  47
  48  private:
  49   Server* server_;
  50   std::vector<std::shared_ptr<Service> > default_services_;
  51 };
  52
  53 }  // namespace grpc
  54
  55 #endif  // GRPCPP_IMPL_SERVER_INITIALIZER_H