mirror of https://github.com/grpc/grpc.git
Merge branch 'master' of https://github.com/grpc/grpc into channel-trace-size
ncteisen
6 years ago
commit
867192e29e
54 changed files with 514 additions and 60 deletions
@ -0,0 +1,44 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_CLIENT_INTERCEPTOR_H |
||||||
|
#define GRPCPP_IMPL_CODEGEN_CLIENT_INTERCEPTOR_H |
||||||
|
|
||||||
|
#include <grpcpp/impl/codegen/interceptor.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace experimental { |
||||||
|
class ClientInterceptor { |
||||||
|
public: |
||||||
|
virtual ~ClientInterceptor() {} |
||||||
|
|
||||||
|
virtual void Intercept(InterceptorBatchMethods* methods) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
class ClientRpcInfo {}; |
||||||
|
|
||||||
|
class ClientInterceptorFactoryInterface { |
||||||
|
public: |
||||||
|
virtual ~ClientInterceptorFactoryInterface() {} |
||||||
|
virtual ClientInterceptor* CreateClientInterceptor(ClientRpcInfo* info) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace experimental
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPCPP_IMPL_CODEGEN_CLIENT_INTERCEPTOR_H
|
||||||
@ -0,0 +1,67 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_INTERCEPTOR_H |
||||||
|
#define GRPCPP_IMPL_CODEGEN_INTERCEPTOR_H |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace experimental { |
||||||
|
class InterceptedMessage { |
||||||
|
public: |
||||||
|
template <class M> |
||||||
|
bool Extract(M* msg); // returns false if definitely invalid extraction
|
||||||
|
template <class M> |
||||||
|
M* MutableExtract(); |
||||||
|
uint64_t length(); // length on wire
|
||||||
|
}; |
||||||
|
|
||||||
|
enum class InterceptionHookPoints { |
||||||
|
/* The first two in this list are for clients and servers */ |
||||||
|
PRE_SEND_INITIAL_METADATA, |
||||||
|
PRE_SEND_MESSAGE, |
||||||
|
PRE_SEND_STATUS /* server only */, |
||||||
|
/* The following three are for hijacked clients only and can only be
|
||||||
|
registered by the global interceptor */ |
||||||
|
PRE_RECV_INITIAL_METADATA, |
||||||
|
PRE_RECV_MESSAGE, |
||||||
|
PRE_RECV_STATUS, |
||||||
|
/* The following two are for all clients and servers */ |
||||||
|
POST_RECV_INITIAL_METADATA, |
||||||
|
POST_RECV_MESSAGE, |
||||||
|
POST_RECV_STATUS /* client only */, |
||||||
|
POST_RECV_CLOSE /* server only */, |
||||||
|
NUM_INTERCEPTION_HOOKS |
||||||
|
}; |
||||||
|
|
||||||
|
class InterceptorBatchMethods { |
||||||
|
public: |
||||||
|
virtual ~InterceptorBatchMethods(); |
||||||
|
// Queries to check whether the current batch has an interception hook point
|
||||||
|
// of type \a type
|
||||||
|
virtual bool QueryInterceptionHookPoint(InterceptionHookPoints type) = 0; |
||||||
|
// Calling this will signal that the interceptor is done intercepting the
|
||||||
|
// current batch of the RPC
|
||||||
|
virtual void Proceed() = 0; |
||||||
|
// Calling this indicates that the interceptor has hijacked the RPC (only
|
||||||
|
// valid if the batch contains send_initial_metadata on the client side)
|
||||||
|
virtual void Hijack() = 0; |
||||||
|
}; |
||||||
|
} // namespace experimental
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPCPP_IMPL_CODEGEN_INTERCEPTOR_H
|
||||||
Loading…
Reference in new issue