warn user when they forget register generic service

1 files changed, 12 insertions, 4 deletions

diff --git a/include/grpc++/impl/service_type.h b/include/grpc++/impl/service_type.h
index 4925785fd4..27af88e16f 100644
--- a/include/grpc++/impl/service_type.h
+++ b/include/grpc++/impl/service_type.h
@@ -82,6 +82,15 @@ class Service {
     return false;
   }
 
+  bool has_generic_methods() const {
+    for (auto it = methods_.begin(); it != methods_.end(); ++it) {
+      if ((*it)->get() == nullptr) {
+        return true;
+      }
+    }
+    return false;
+  }
+
  protected:
   template <class Message>
   void RequestAsyncUnary(int index, ServerContext* context, Message* request,
@@ -122,16 +131,15 @@ class Service {
 
   void MarkMethodAsync(int index) {
     if (methods_[index].get() == nullptr) {
-      gpr_log(GPR_ERROR, "A method cannot be marked async and generic.");
-      abort();
+      gpr_log(GPR_ERROR, "Method already marked generic.");
+      return;
     }
     methods_[index]->ResetHandler();
   }
 
   void MarkMethodGeneric(int index) {
     if (methods_[index]->handler() == nullptr) {
-      gpr_log(GPR_ERROR, "A method cannot be marked async and generic.");
-      abort();
+      gpr_log(GPR_ERROR, "Method already marked async.");
     }
     methods_[index].reset();
   }