Yash Tibrewal
7 years ago
parent
f9fe345553
commit
b232bacf27
14 changed files with 147 additions and 60 deletions
@ -0,0 +1,89 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2018 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef GRPCPP_IMPL_CODEGEN_SERVER_INTERCEPTOR_H |
||||||
|
#define GRPCPP_IMPL_CODEGEN_SERVER_INTERCEPTOR_H |
||||||
|
|
||||||
|
#include <vector> |
||||||
|
|
||||||
|
#include <grpc/impl/codegen/log.h> |
||||||
|
#include <grpcpp/impl/codegen/interceptor.h> |
||||||
|
#include <grpcpp/impl/codegen/string_ref.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
class ServerContext; |
||||||
|
|
||||||
|
namespace internal { |
||||||
|
template <int I> |
||||||
|
class CallNoOp; |
||||||
|
} |
||||||
|
|
||||||
|
namespace experimental { |
||||||
|
class ServerRpcInfo; |
||||||
|
|
||||||
|
class ServerInterceptorFactoryInterface { |
||||||
|
public: |
||||||
|
virtual ~ServerInterceptorFactoryInterface() {} |
||||||
|
virtual Interceptor* CreateServerInterceptor(ServerRpcInfo* info) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
class ServerRpcInfo { |
||||||
|
public: |
||||||
|
ServerRpcInfo() {} |
||||||
|
ServerRpcInfo(grpc::ServerContext* ctx, const char* method, |
||||||
|
const std::vector<std::unique_ptr< |
||||||
|
experimental::ServerInterceptorFactoryInterface>>& creators) |
||||||
|
: ctx_(ctx), method_(method) { |
||||||
|
for (const auto& creator : creators) { |
||||||
|
interceptors_.push_back(std::unique_ptr<experimental::Interceptor>( |
||||||
|
creator->CreateServerInterceptor(this))); |
||||||
|
} |
||||||
|
} |
||||||
|
~ServerRpcInfo(){}; |
||||||
|
|
||||||
|
ServerRpcInfo(const ServerRpcInfo&) = delete; |
||||||
|
ServerRpcInfo(ServerRpcInfo&&) = default; |
||||||
|
ServerRpcInfo& operator=(ServerRpcInfo&&) = default; |
||||||
|
|
||||||
|
// Getter methods
|
||||||
|
const char* method() { return method_; } |
||||||
|
grpc::ServerContext* server_context() { return ctx_; } |
||||||
|
|
||||||
|
public: |
||||||
|
/* Runs interceptor at pos \a pos. If \a reverse is set, the interceptor order
|
||||||
|
* is the reverse */ |
||||||
|
void RunInterceptor( |
||||||
|
experimental::InterceptorBatchMethods* interceptor_methods, |
||||||
|
unsigned int pos) { |
||||||
|
GPR_CODEGEN_ASSERT(pos < interceptors_.size()); |
||||||
|
interceptors_[pos]->Intercept(interceptor_methods); |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
grpc::ServerContext* ctx_ = nullptr; |
||||||
|
const char* method_ = nullptr; |
||||||
|
|
||||||
|
public: |
||||||
|
std::vector<std::unique_ptr<experimental::Interceptor>> interceptors_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace experimental
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPCPP_IMPL_CODEGEN_SERVER_INTERCEPTOR_H
|
||||||
Loading…
Reference in new issue