mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
2.6 KiB
99 lines
2.6 KiB
/* |
|
* |
|
* 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 <atomic> |
|
#include <vector> |
|
|
|
#include <grpcpp/impl/codegen/interceptor.h> |
|
#include <grpcpp/impl/codegen/string_ref.h> |
|
|
|
namespace grpc { |
|
|
|
class ServerContext; |
|
|
|
namespace internal { |
|
class InterceptorBatchMethodsImpl; |
|
} |
|
|
|
namespace experimental { |
|
class ServerRpcInfo; |
|
|
|
class ServerInterceptorFactoryInterface { |
|
public: |
|
virtual ~ServerInterceptorFactoryInterface() {} |
|
virtual Interceptor* CreateServerInterceptor(ServerRpcInfo* info) = 0; |
|
}; |
|
|
|
class ServerRpcInfo { |
|
public: |
|
~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_; } |
|
|
|
private: |
|
ServerRpcInfo(grpc::ServerContext* ctx, const char* method) |
|
: ctx_(ctx), method_(method) { |
|
ref_.store(1); |
|
} |
|
|
|
// Runs interceptor at pos \a pos. |
|
void RunInterceptor( |
|
experimental::InterceptorBatchMethods* interceptor_methods, size_t pos) { |
|
GPR_CODEGEN_ASSERT(pos < interceptors_.size()); |
|
interceptors_[pos]->Intercept(interceptor_methods); |
|
} |
|
|
|
void RegisterInterceptors( |
|
const std::vector< |
|
std::unique_ptr<experimental::ServerInterceptorFactoryInterface>>& |
|
creators) { |
|
for (const auto& creator : creators) { |
|
interceptors_.push_back(std::unique_ptr<experimental::Interceptor>( |
|
creator->CreateServerInterceptor(this))); |
|
} |
|
} |
|
|
|
void Ref() { ref_++; } |
|
void Unref() { |
|
if (--ref_ == 0) { |
|
delete this; |
|
} |
|
} |
|
|
|
grpc::ServerContext* ctx_ = nullptr; |
|
const char* method_ = nullptr; |
|
std::atomic_int ref_; |
|
std::vector<std::unique_ptr<experimental::Interceptor>> interceptors_; |
|
|
|
friend class internal::InterceptorBatchMethodsImpl; |
|
friend class grpc::ServerContext; |
|
}; |
|
|
|
} // namespace experimental |
|
} // namespace grpc |
|
|
|
#endif // GRPCPP_IMPL_CODEGEN_SERVER_INTERCEPTOR_H
|
|
|