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.
103 lines
2.8 KiB
103 lines
2.8 KiB
// |
|
// |
|
// Copyright 2015 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_METADATA_MAP_H |
|
#define GRPCPP_IMPL_METADATA_MAP_H |
|
|
|
#include <grpc/grpc.h> |
|
#include <grpcpp/support/slice.h> |
|
|
|
#include <map> |
|
|
|
namespace grpc { |
|
|
|
namespace internal { |
|
|
|
const char kBinaryErrorDetailsKey[] = "grpc-status-details-bin"; |
|
|
|
class MetadataMap { |
|
public: |
|
MetadataMap() { Setup(); } |
|
|
|
~MetadataMap() { Destroy(); } |
|
|
|
std::string GetBinaryErrorDetails() { |
|
// if filled_, extract from the multimap for O(log(n)) |
|
if (filled_) { |
|
auto iter = map_.find(kBinaryErrorDetailsKey); |
|
if (iter != map_.end()) { |
|
return std::string(iter->second.begin(), iter->second.length()); |
|
} |
|
} |
|
// if not yet filled, take the O(n) lookup to avoid allocating the |
|
// multimap until it is requested. |
|
// TODO(ncteisen): plumb this through core as a first class object, just |
|
// like code and message. |
|
else { |
|
for (size_t i = 0; i < arr_.count; i++) { |
|
if (strncmp(reinterpret_cast<const char*>( |
|
GRPC_SLICE_START_PTR(arr_.metadata[i].key)), |
|
kBinaryErrorDetailsKey, |
|
GRPC_SLICE_LENGTH(arr_.metadata[i].key)) == 0) { |
|
return std::string(reinterpret_cast<const char*>( |
|
GRPC_SLICE_START_PTR(arr_.metadata[i].value)), |
|
GRPC_SLICE_LENGTH(arr_.metadata[i].value)); |
|
} |
|
} |
|
} |
|
return std::string(); |
|
} |
|
|
|
std::multimap<grpc::string_ref, grpc::string_ref>* map() { |
|
FillMap(); |
|
return &map_; |
|
} |
|
grpc_metadata_array* arr() { return &arr_; } |
|
|
|
void Reset() { |
|
filled_ = false; |
|
map_.clear(); |
|
Destroy(); |
|
Setup(); |
|
} |
|
|
|
private: |
|
bool filled_ = false; |
|
grpc_metadata_array arr_; |
|
std::multimap<grpc::string_ref, grpc::string_ref> map_; |
|
|
|
void Destroy() { grpc_metadata_array_destroy(&arr_); } |
|
|
|
void Setup() { memset(&arr_, 0, sizeof(arr_)); } |
|
|
|
void FillMap() { |
|
if (filled_) return; |
|
filled_ = true; |
|
for (size_t i = 0; i < arr_.count; i++) { |
|
// TODO(yangg) handle duplicates? |
|
map_.insert(std::pair<grpc::string_ref, grpc::string_ref>( |
|
StringRefFromSlice(&arr_.metadata[i].key), |
|
StringRefFromSlice(&arr_.metadata[i].value))); |
|
} |
|
} |
|
}; |
|
} // namespace internal |
|
|
|
} // namespace grpc |
|
|
|
#endif // GRPCPP_IMPL_METADATA_MAP_H
|
|
|