grpc_library.h 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. //
  2. //
  3. // Copyright 2015 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. #ifndef GRPCPP_IMPL_GRPC_LIBRARY_H
  19. #define GRPCPP_IMPL_GRPC_LIBRARY_H
  20. #include <iostream>
  21. #include <grpc/grpc.h>
  22. #include <grpcpp/impl/codegen/config.h>
  23. namespace grpc {
  24. namespace internal {
  25. /// Classes that require gRPC to be initialized should inherit from this class.
  26. class GrpcLibrary {
  27. public:
  28. explicit GrpcLibrary(bool call_grpc_init = true) : grpc_init_called_(false) {
  29. if (call_grpc_init) {
  30. grpc_init();
  31. grpc_init_called_ = true;
  32. }
  33. }
  34. virtual ~GrpcLibrary() {
  35. if (grpc_init_called_) {
  36. grpc_shutdown();
  37. }
  38. }
  39. private:
  40. bool grpc_init_called_;
  41. };
  42. } // namespace internal
  43. } // namespace grpc
  44. #endif // GRPCPP_IMPL_GRPC_LIBRARY_H