Matthew Stevenson
6 years ago
commit
65bf49abba
227 changed files with 5790 additions and 5357 deletions
@ -0,0 +1,78 @@ |
||||
//
|
||||
// Copyright 2019 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.
|
||||
//
|
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/backend_metric.h" |
||||
|
||||
#include "src/core/lib/gprpp/string_view.h" |
||||
#include "udpa/data/orca/v1/orca_load_report.upb.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
namespace { |
||||
|
||||
template <typename EntryType> |
||||
Map<StringView, double, StringLess> ParseMap( |
||||
udpa_data_orca_v1_OrcaLoadReport* msg, |
||||
EntryType** (*entry_func)(udpa_data_orca_v1_OrcaLoadReport*, size_t*), |
||||
upb_strview (*key_func)(const EntryType*), |
||||
double (*value_func)(const EntryType*), Arena* arena) { |
||||
Map<StringView, double, StringLess> result; |
||||
size_t size; |
||||
const auto* const* entries = entry_func(msg, &size); |
||||
for (size_t i = 0; i < size; ++i) { |
||||
upb_strview key_view = key_func(entries[i]); |
||||
char* key = static_cast<char*>(arena->Alloc(key_view.size + 1)); |
||||
memcpy(key, key_view.data, key_view.size); |
||||
result[StringView(key, key_view.size)] = value_func(entries[i]); |
||||
} |
||||
return result; |
||||
} |
||||
|
||||
} // namespace
|
||||
|
||||
const LoadBalancingPolicy::BackendMetricData* ParseBackendMetricData( |
||||
const grpc_slice& serialized_load_report, Arena* arena) { |
||||
upb::Arena upb_arena; |
||||
udpa_data_orca_v1_OrcaLoadReport* msg = |
||||
udpa_data_orca_v1_OrcaLoadReport_parse( |
||||
reinterpret_cast<const char*>( |
||||
GRPC_SLICE_START_PTR(serialized_load_report)), |
||||
GRPC_SLICE_LENGTH(serialized_load_report), upb_arena.ptr()); |
||||
if (msg == nullptr) return nullptr; |
||||
LoadBalancingPolicy::BackendMetricData* backend_metric_data = |
||||
arena->New<LoadBalancingPolicy::BackendMetricData>(); |
||||
backend_metric_data->cpu_utilization = |
||||
udpa_data_orca_v1_OrcaLoadReport_cpu_utilization(msg); |
||||
backend_metric_data->mem_utilization = |
||||
udpa_data_orca_v1_OrcaLoadReport_mem_utilization(msg); |
||||
backend_metric_data->requests_per_second = |
||||
udpa_data_orca_v1_OrcaLoadReport_rps(msg); |
||||
backend_metric_data->request_cost = |
||||
ParseMap<udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry>( |
||||
msg, udpa_data_orca_v1_OrcaLoadReport_mutable_request_cost, |
||||
udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_key, |
||||
udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_value, arena); |
||||
backend_metric_data->utilization = |
||||
ParseMap<udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry>( |
||||
msg, udpa_data_orca_v1_OrcaLoadReport_mutable_utilization, |
||||
udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_key, |
||||
udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_value, arena); |
||||
return backend_metric_data; |
||||
} |
||||
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,36 @@ |
||||
//
|
||||
// Copyright 2019 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 GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_BACKEND_METRIC_H |
||||
#define GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_BACKEND_METRIC_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include <grpc/slice.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/lb_policy.h" |
||||
#include "src/core/lib/gprpp/arena.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
// Parses the serialized load report and allocates a BackendMetricData
|
||||
// object on the arena.
|
||||
const LoadBalancingPolicy::BackendMetricData* ParseBackendMetricData( |
||||
const grpc_slice& serialized_load_report, Arena* arena); |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_BACKEND_METRIC_H */ |
||||
@ -0,0 +1,34 @@ |
||||
/* This file was generated by upbc (the upb compiler) from the input
|
||||
* file: |
||||
* |
||||
* envoy/api/v2/cluster/filter.proto |
||||
* |
||||
* Do not edit -- your changes will be discarded when the file is |
||||
* regenerated. */ |
||||
|
||||
#include <stddef.h> |
||||
#include "upb/msg.h" |
||||
#include "envoy/api/v2/cluster/filter.upb.h" |
||||
#include "google/protobuf/any.upb.h" |
||||
#include "validate/validate.upb.h" |
||||
#include "gogoproto/gogo.upb.h" |
||||
|
||||
#include "upb/port_def.inc" |
||||
|
||||
static const upb_msglayout *const envoy_api_v2_cluster_Filter_submsgs[1] = { |
||||
&google_protobuf_Any_msginit, |
||||
}; |
||||
|
||||
static const upb_msglayout_field envoy_api_v2_cluster_Filter__fields[2] = { |
||||
{1, UPB_SIZE(0, 0), 0, 0, 9, 1}, |
||||
{2, UPB_SIZE(8, 16), 0, 0, 11, 1}, |
||||
}; |
||||
|
||||
const upb_msglayout envoy_api_v2_cluster_Filter_msginit = { |
||||
&envoy_api_v2_cluster_Filter_submsgs[0], |
||||
&envoy_api_v2_cluster_Filter__fields[0], |
||||
UPB_SIZE(16, 32), 2, false, |
||||
}; |
||||
|
||||
#include "upb/port_undef.inc" |
||||
|
||||
@ -0,0 +1,69 @@ |
||||
/* This file was generated by upbc (the upb compiler) from the input
|
||||
* file: |
||||
* |
||||
* envoy/api/v2/cluster/filter.proto |
||||
* |
||||
* Do not edit -- your changes will be discarded when the file is |
||||
* regenerated. */ |
||||
|
||||
#ifndef ENVOY_API_V2_CLUSTER_FILTER_PROTO_UPB_H_ |
||||
#define ENVOY_API_V2_CLUSTER_FILTER_PROTO_UPB_H_ |
||||
|
||||
#include "upb/generated_util.h" |
||||
#include "upb/msg.h" |
||||
#include "upb/decode.h" |
||||
#include "upb/encode.h" |
||||
|
||||
#include "upb/port_def.inc" |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
struct envoy_api_v2_cluster_Filter; |
||||
typedef struct envoy_api_v2_cluster_Filter envoy_api_v2_cluster_Filter; |
||||
extern const upb_msglayout envoy_api_v2_cluster_Filter_msginit; |
||||
struct google_protobuf_Any; |
||||
extern const upb_msglayout google_protobuf_Any_msginit; |
||||
|
||||
|
||||
/* envoy.api.v2.cluster.Filter */ |
||||
|
||||
UPB_INLINE envoy_api_v2_cluster_Filter *envoy_api_v2_cluster_Filter_new(upb_arena *arena) { |
||||
return (envoy_api_v2_cluster_Filter *)upb_msg_new(&envoy_api_v2_cluster_Filter_msginit, arena); |
||||
} |
||||
UPB_INLINE envoy_api_v2_cluster_Filter *envoy_api_v2_cluster_Filter_parse(const char *buf, size_t size, |
||||
upb_arena *arena) { |
||||
envoy_api_v2_cluster_Filter *ret = envoy_api_v2_cluster_Filter_new(arena); |
||||
return (ret && upb_decode(buf, size, ret, &envoy_api_v2_cluster_Filter_msginit, arena)) ? ret : NULL; |
||||
} |
||||
UPB_INLINE char *envoy_api_v2_cluster_Filter_serialize(const envoy_api_v2_cluster_Filter *msg, upb_arena *arena, size_t *len) { |
||||
return upb_encode(msg, &envoy_api_v2_cluster_Filter_msginit, arena, len); |
||||
} |
||||
|
||||
UPB_INLINE upb_strview envoy_api_v2_cluster_Filter_name(const envoy_api_v2_cluster_Filter *msg) { return UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(0, 0)); } |
||||
UPB_INLINE const struct google_protobuf_Any* envoy_api_v2_cluster_Filter_typed_config(const envoy_api_v2_cluster_Filter *msg) { return UPB_FIELD_AT(msg, const struct google_protobuf_Any*, UPB_SIZE(8, 16)); } |
||||
|
||||
UPB_INLINE void envoy_api_v2_cluster_Filter_set_name(envoy_api_v2_cluster_Filter *msg, upb_strview value) { |
||||
UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(0, 0)) = value; |
||||
} |
||||
UPB_INLINE void envoy_api_v2_cluster_Filter_set_typed_config(envoy_api_v2_cluster_Filter *msg, struct google_protobuf_Any* value) { |
||||
UPB_FIELD_AT(msg, struct google_protobuf_Any*, UPB_SIZE(8, 16)) = value; |
||||
} |
||||
UPB_INLINE struct google_protobuf_Any* envoy_api_v2_cluster_Filter_mutable_typed_config(envoy_api_v2_cluster_Filter *msg, upb_arena *arena) { |
||||
struct google_protobuf_Any* sub = (struct google_protobuf_Any*)envoy_api_v2_cluster_Filter_typed_config(msg); |
||||
if (sub == NULL) { |
||||
sub = (struct google_protobuf_Any*)upb_msg_new(&google_protobuf_Any_msginit, arena); |
||||
if (!sub) return NULL; |
||||
envoy_api_v2_cluster_Filter_set_typed_config(msg, sub); |
||||
} |
||||
return sub; |
||||
} |
||||
|
||||
#ifdef __cplusplus |
||||
} /* extern "C" */ |
||||
#endif |
||||
|
||||
#include "upb/port_undef.inc" |
||||
|
||||
#endif /* ENVOY_API_V2_CLUSTER_FILTER_PROTO_UPB_H_ */ |
||||
@ -0,0 +1,35 @@ |
||||
/* This file was generated by upbc (the upb compiler) from the input
|
||||
* file: |
||||
* |
||||
* envoy/api/v2/core/http_uri.proto |
||||
* |
||||
* Do not edit -- your changes will be discarded when the file is |
||||
* regenerated. */ |
||||
|
||||
#include <stddef.h> |
||||
#include "upb/msg.h" |
||||
#include "envoy/api/v2/core/http_uri.upb.h" |
||||
#include "google/protobuf/duration.upb.h" |
||||
#include "gogoproto/gogo.upb.h" |
||||
#include "validate/validate.upb.h" |
||||
|
||||
#include "upb/port_def.inc" |
||||
|
||||
static const upb_msglayout *const envoy_api_v2_core_HttpUri_submsgs[1] = { |
||||
&google_protobuf_Duration_msginit, |
||||
}; |
||||
|
||||
static const upb_msglayout_field envoy_api_v2_core_HttpUri__fields[3] = { |
||||
{1, UPB_SIZE(0, 0), 0, 0, 9, 1}, |
||||
{2, UPB_SIZE(12, 24), UPB_SIZE(-21, -41), 0, 9, 1}, |
||||
{3, UPB_SIZE(8, 16), 0, 0, 11, 1}, |
||||
}; |
||||
|
||||
const upb_msglayout envoy_api_v2_core_HttpUri_msginit = { |
||||
&envoy_api_v2_core_HttpUri_submsgs[0], |
||||
&envoy_api_v2_core_HttpUri__fields[0], |
||||
UPB_SIZE(24, 48), 3, false, |
||||
}; |
||||
|
||||
#include "upb/port_undef.inc" |
||||
|
||||
@ -0,0 +1,80 @@ |
||||
/* This file was generated by upbc (the upb compiler) from the input
|
||||
* file: |
||||
* |
||||
* envoy/api/v2/core/http_uri.proto |
||||
* |
||||
* Do not edit -- your changes will be discarded when the file is |
||||
* regenerated. */ |
||||
|
||||
#ifndef ENVOY_API_V2_CORE_HTTP_URI_PROTO_UPB_H_ |
||||
#define ENVOY_API_V2_CORE_HTTP_URI_PROTO_UPB_H_ |
||||
|
||||
#include "upb/generated_util.h" |
||||
#include "upb/msg.h" |
||||
#include "upb/decode.h" |
||||
#include "upb/encode.h" |
||||
|
||||
#include "upb/port_def.inc" |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
struct envoy_api_v2_core_HttpUri; |
||||
typedef struct envoy_api_v2_core_HttpUri envoy_api_v2_core_HttpUri; |
||||
extern const upb_msglayout envoy_api_v2_core_HttpUri_msginit; |
||||
struct google_protobuf_Duration; |
||||
extern const upb_msglayout google_protobuf_Duration_msginit; |
||||
|
||||
|
||||
/* envoy.api.v2.core.HttpUri */ |
||||
|
||||
UPB_INLINE envoy_api_v2_core_HttpUri *envoy_api_v2_core_HttpUri_new(upb_arena *arena) { |
||||
return (envoy_api_v2_core_HttpUri *)upb_msg_new(&envoy_api_v2_core_HttpUri_msginit, arena); |
||||
} |
||||
UPB_INLINE envoy_api_v2_core_HttpUri *envoy_api_v2_core_HttpUri_parse(const char *buf, size_t size, |
||||
upb_arena *arena) { |
||||
envoy_api_v2_core_HttpUri *ret = envoy_api_v2_core_HttpUri_new(arena); |
||||
return (ret && upb_decode(buf, size, ret, &envoy_api_v2_core_HttpUri_msginit, arena)) ? ret : NULL; |
||||
} |
||||
UPB_INLINE char *envoy_api_v2_core_HttpUri_serialize(const envoy_api_v2_core_HttpUri *msg, upb_arena *arena, size_t *len) { |
||||
return upb_encode(msg, &envoy_api_v2_core_HttpUri_msginit, arena, len); |
||||
} |
||||
|
||||
typedef enum { |
||||
envoy_api_v2_core_HttpUri_http_upstream_type_cluster = 2, |
||||
envoy_api_v2_core_HttpUri_http_upstream_type_NOT_SET = 0 |
||||
} envoy_api_v2_core_HttpUri_http_upstream_type_oneofcases; |
||||
UPB_INLINE envoy_api_v2_core_HttpUri_http_upstream_type_oneofcases envoy_api_v2_core_HttpUri_http_upstream_type_case(const envoy_api_v2_core_HttpUri* msg) { return (envoy_api_v2_core_HttpUri_http_upstream_type_oneofcases)UPB_FIELD_AT(msg, int32_t, UPB_SIZE(20, 40)); } |
||||
|
||||
UPB_INLINE upb_strview envoy_api_v2_core_HttpUri_uri(const envoy_api_v2_core_HttpUri *msg) { return UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(0, 0)); } |
||||
UPB_INLINE bool envoy_api_v2_core_HttpUri_has_cluster(const envoy_api_v2_core_HttpUri *msg) { return _upb_has_oneof_field(msg, UPB_SIZE(20, 40), 2); } |
||||
UPB_INLINE upb_strview envoy_api_v2_core_HttpUri_cluster(const envoy_api_v2_core_HttpUri *msg) { return UPB_READ_ONEOF(msg, upb_strview, UPB_SIZE(12, 24), UPB_SIZE(20, 40), 2, upb_strview_make("", strlen(""))); } |
||||
UPB_INLINE const struct google_protobuf_Duration* envoy_api_v2_core_HttpUri_timeout(const envoy_api_v2_core_HttpUri *msg) { return UPB_FIELD_AT(msg, const struct google_protobuf_Duration*, UPB_SIZE(8, 16)); } |
||||
|
||||
UPB_INLINE void envoy_api_v2_core_HttpUri_set_uri(envoy_api_v2_core_HttpUri *msg, upb_strview value) { |
||||
UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(0, 0)) = value; |
||||
} |
||||
UPB_INLINE void envoy_api_v2_core_HttpUri_set_cluster(envoy_api_v2_core_HttpUri *msg, upb_strview value) { |
||||
UPB_WRITE_ONEOF(msg, upb_strview, UPB_SIZE(12, 24), value, UPB_SIZE(20, 40), 2); |
||||
} |
||||
UPB_INLINE void envoy_api_v2_core_HttpUri_set_timeout(envoy_api_v2_core_HttpUri *msg, struct google_protobuf_Duration* value) { |
||||
UPB_FIELD_AT(msg, struct google_protobuf_Duration*, UPB_SIZE(8, 16)) = value; |
||||
} |
||||
UPB_INLINE struct google_protobuf_Duration* envoy_api_v2_core_HttpUri_mutable_timeout(envoy_api_v2_core_HttpUri *msg, upb_arena *arena) { |
||||
struct google_protobuf_Duration* sub = (struct google_protobuf_Duration*)envoy_api_v2_core_HttpUri_timeout(msg); |
||||
if (sub == NULL) { |
||||
sub = (struct google_protobuf_Duration*)upb_msg_new(&google_protobuf_Duration_msginit, arena); |
||||
if (!sub) return NULL; |
||||
envoy_api_v2_core_HttpUri_set_timeout(msg, sub); |
||||
} |
||||
return sub; |
||||
} |
||||
|
||||
#ifdef __cplusplus |
||||
} /* extern "C" */ |
||||
#endif |
||||
|
||||
#include "upb/port_undef.inc" |
||||
|
||||
#endif /* ENVOY_API_V2_CORE_HTTP_URI_PROTO_UPB_H_ */ |
||||
@ -0,0 +1,58 @@ |
||||
/* This file was generated by upbc (the upb compiler) from the input
|
||||
* file: |
||||
* |
||||
* udpa/data/orca/v1/orca_load_report.proto |
||||
* |
||||
* Do not edit -- your changes will be discarded when the file is |
||||
* regenerated. */ |
||||
|
||||
#include <stddef.h> |
||||
#include "upb/msg.h" |
||||
#include "udpa/data/orca/v1/orca_load_report.upb.h" |
||||
#include "validate/validate.upb.h" |
||||
|
||||
#include "upb/port_def.inc" |
||||
|
||||
static const upb_msglayout *const udpa_data_orca_v1_OrcaLoadReport_submsgs[2] = { |
||||
&udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_msginit, |
||||
&udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_msginit, |
||||
}; |
||||
|
||||
static const upb_msglayout_field udpa_data_orca_v1_OrcaLoadReport__fields[5] = { |
||||
{1, UPB_SIZE(0, 0), 0, 0, 1, 1}, |
||||
{2, UPB_SIZE(8, 8), 0, 0, 1, 1}, |
||||
{3, UPB_SIZE(16, 16), 0, 0, 4, 1}, |
||||
{4, UPB_SIZE(24, 24), 0, 0, 11, 3}, |
||||
{5, UPB_SIZE(28, 32), 0, 1, 11, 3}, |
||||
}; |
||||
|
||||
const upb_msglayout udpa_data_orca_v1_OrcaLoadReport_msginit = { |
||||
&udpa_data_orca_v1_OrcaLoadReport_submsgs[0], |
||||
&udpa_data_orca_v1_OrcaLoadReport__fields[0], |
||||
UPB_SIZE(32, 40), 5, false, |
||||
}; |
||||
|
||||
static const upb_msglayout_field udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry__fields[2] = { |
||||
{1, UPB_SIZE(8, 8), 0, 0, 9, 1}, |
||||
{2, UPB_SIZE(0, 0), 0, 0, 1, 1}, |
||||
}; |
||||
|
||||
const upb_msglayout udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_msginit = { |
||||
NULL, |
||||
&udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry__fields[0], |
||||
UPB_SIZE(16, 32), 2, false, |
||||
}; |
||||
|
||||
static const upb_msglayout_field udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry__fields[2] = { |
||||
{1, UPB_SIZE(8, 8), 0, 0, 9, 1}, |
||||
{2, UPB_SIZE(0, 0), 0, 0, 1, 1}, |
||||
}; |
||||
|
||||
const upb_msglayout udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_msginit = { |
||||
NULL, |
||||
&udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry__fields[0], |
||||
UPB_SIZE(16, 32), 2, false, |
||||
}; |
||||
|
||||
#include "upb/port_undef.inc" |
||||
|
||||
@ -0,0 +1,144 @@ |
||||
/* This file was generated by upbc (the upb compiler) from the input
|
||||
* file: |
||||
* |
||||
* udpa/data/orca/v1/orca_load_report.proto |
||||
* |
||||
* Do not edit -- your changes will be discarded when the file is |
||||
* regenerated. */ |
||||
|
||||
#ifndef UDPA_DATA_ORCA_V1_ORCA_LOAD_REPORT_PROTO_UPB_H_ |
||||
#define UDPA_DATA_ORCA_V1_ORCA_LOAD_REPORT_PROTO_UPB_H_ |
||||
|
||||
#include "upb/generated_util.h" |
||||
#include "upb/msg.h" |
||||
#include "upb/decode.h" |
||||
#include "upb/encode.h" |
||||
|
||||
#include "upb/port_def.inc" |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
struct udpa_data_orca_v1_OrcaLoadReport; |
||||
struct udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry; |
||||
struct udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry; |
||||
typedef struct udpa_data_orca_v1_OrcaLoadReport udpa_data_orca_v1_OrcaLoadReport; |
||||
typedef struct udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry; |
||||
typedef struct udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry; |
||||
extern const upb_msglayout udpa_data_orca_v1_OrcaLoadReport_msginit; |
||||
extern const upb_msglayout udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_msginit; |
||||
extern const upb_msglayout udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_msginit; |
||||
|
||||
|
||||
/* udpa.data.orca.v1.OrcaLoadReport */ |
||||
|
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport *udpa_data_orca_v1_OrcaLoadReport_new(upb_arena *arena) { |
||||
return (udpa_data_orca_v1_OrcaLoadReport *)upb_msg_new(&udpa_data_orca_v1_OrcaLoadReport_msginit, arena); |
||||
} |
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport *udpa_data_orca_v1_OrcaLoadReport_parse(const char *buf, size_t size, |
||||
upb_arena *arena) { |
||||
udpa_data_orca_v1_OrcaLoadReport *ret = udpa_data_orca_v1_OrcaLoadReport_new(arena); |
||||
return (ret && upb_decode(buf, size, ret, &udpa_data_orca_v1_OrcaLoadReport_msginit, arena)) ? ret : NULL; |
||||
} |
||||
UPB_INLINE char *udpa_data_orca_v1_OrcaLoadReport_serialize(const udpa_data_orca_v1_OrcaLoadReport *msg, upb_arena *arena, size_t *len) { |
||||
return upb_encode(msg, &udpa_data_orca_v1_OrcaLoadReport_msginit, arena, len); |
||||
} |
||||
|
||||
UPB_INLINE double udpa_data_orca_v1_OrcaLoadReport_cpu_utilization(const udpa_data_orca_v1_OrcaLoadReport *msg) { return UPB_FIELD_AT(msg, double, UPB_SIZE(0, 0)); } |
||||
UPB_INLINE double udpa_data_orca_v1_OrcaLoadReport_mem_utilization(const udpa_data_orca_v1_OrcaLoadReport *msg) { return UPB_FIELD_AT(msg, double, UPB_SIZE(8, 8)); } |
||||
UPB_INLINE uint64_t udpa_data_orca_v1_OrcaLoadReport_rps(const udpa_data_orca_v1_OrcaLoadReport *msg) { return UPB_FIELD_AT(msg, uint64_t, UPB_SIZE(16, 16)); } |
||||
UPB_INLINE const udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry* const* udpa_data_orca_v1_OrcaLoadReport_request_cost(const udpa_data_orca_v1_OrcaLoadReport *msg, size_t *len) { return (const udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry* const*)_upb_array_accessor(msg, UPB_SIZE(24, 24), len); } |
||||
UPB_INLINE const udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry* const* udpa_data_orca_v1_OrcaLoadReport_utilization(const udpa_data_orca_v1_OrcaLoadReport *msg, size_t *len) { return (const udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry* const*)_upb_array_accessor(msg, UPB_SIZE(28, 32), len); } |
||||
|
||||
UPB_INLINE void udpa_data_orca_v1_OrcaLoadReport_set_cpu_utilization(udpa_data_orca_v1_OrcaLoadReport *msg, double value) { |
||||
UPB_FIELD_AT(msg, double, UPB_SIZE(0, 0)) = value; |
||||
} |
||||
UPB_INLINE void udpa_data_orca_v1_OrcaLoadReport_set_mem_utilization(udpa_data_orca_v1_OrcaLoadReport *msg, double value) { |
||||
UPB_FIELD_AT(msg, double, UPB_SIZE(8, 8)) = value; |
||||
} |
||||
UPB_INLINE void udpa_data_orca_v1_OrcaLoadReport_set_rps(udpa_data_orca_v1_OrcaLoadReport *msg, uint64_t value) { |
||||
UPB_FIELD_AT(msg, uint64_t, UPB_SIZE(16, 16)) = value; |
||||
} |
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry** udpa_data_orca_v1_OrcaLoadReport_mutable_request_cost(udpa_data_orca_v1_OrcaLoadReport *msg, size_t *len) { |
||||
return (udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry**)_upb_array_mutable_accessor(msg, UPB_SIZE(24, 24), len); |
||||
} |
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry** udpa_data_orca_v1_OrcaLoadReport_resize_request_cost(udpa_data_orca_v1_OrcaLoadReport *msg, size_t len, upb_arena *arena) { |
||||
return (udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry**)_upb_array_resize_accessor(msg, UPB_SIZE(24, 24), len, UPB_SIZE(4, 8), UPB_TYPE_MESSAGE, arena); |
||||
} |
||||
UPB_INLINE struct udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry* udpa_data_orca_v1_OrcaLoadReport_add_request_cost(udpa_data_orca_v1_OrcaLoadReport *msg, upb_arena *arena) { |
||||
struct udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry* sub = (struct udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry*)upb_msg_new(&udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_msginit, arena); |
||||
bool ok = _upb_array_append_accessor( |
||||
msg, UPB_SIZE(24, 24), UPB_SIZE(4, 8), UPB_TYPE_MESSAGE, &sub, arena); |
||||
if (!ok) return NULL; |
||||
return sub; |
||||
} |
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry** udpa_data_orca_v1_OrcaLoadReport_mutable_utilization(udpa_data_orca_v1_OrcaLoadReport *msg, size_t *len) { |
||||
return (udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry**)_upb_array_mutable_accessor(msg, UPB_SIZE(28, 32), len); |
||||
} |
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry** udpa_data_orca_v1_OrcaLoadReport_resize_utilization(udpa_data_orca_v1_OrcaLoadReport *msg, size_t len, upb_arena *arena) { |
||||
return (udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry**)_upb_array_resize_accessor(msg, UPB_SIZE(28, 32), len, UPB_SIZE(4, 8), UPB_TYPE_MESSAGE, arena); |
||||
} |
||||
UPB_INLINE struct udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry* udpa_data_orca_v1_OrcaLoadReport_add_utilization(udpa_data_orca_v1_OrcaLoadReport *msg, upb_arena *arena) { |
||||
struct udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry* sub = (struct udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry*)upb_msg_new(&udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_msginit, arena); |
||||
bool ok = _upb_array_append_accessor( |
||||
msg, UPB_SIZE(28, 32), UPB_SIZE(4, 8), UPB_TYPE_MESSAGE, &sub, arena); |
||||
if (!ok) return NULL; |
||||
return sub; |
||||
} |
||||
|
||||
/* udpa.data.orca.v1.OrcaLoadReport.RequestCostEntry */ |
||||
|
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_new(upb_arena *arena) { |
||||
return (udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *)upb_msg_new(&udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_msginit, arena); |
||||
} |
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_parse(const char *buf, size_t size, |
||||
upb_arena *arena) { |
||||
udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *ret = udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_new(arena); |
||||
return (ret && upb_decode(buf, size, ret, &udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_msginit, arena)) ? ret : NULL; |
||||
} |
||||
UPB_INLINE char *udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_serialize(const udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *msg, upb_arena *arena, size_t *len) { |
||||
return upb_encode(msg, &udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_msginit, arena, len); |
||||
} |
||||
|
||||
UPB_INLINE upb_strview udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_key(const udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *msg) { return UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(8, 8)); } |
||||
UPB_INLINE double udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_value(const udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *msg) { return UPB_FIELD_AT(msg, double, UPB_SIZE(0, 0)); } |
||||
|
||||
UPB_INLINE void udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_set_key(udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *msg, upb_strview value) { |
||||
UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(8, 8)) = value; |
||||
} |
||||
UPB_INLINE void udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry_set_value(udpa_data_orca_v1_OrcaLoadReport_RequestCostEntry *msg, double value) { |
||||
UPB_FIELD_AT(msg, double, UPB_SIZE(0, 0)) = value; |
||||
} |
||||
|
||||
/* udpa.data.orca.v1.OrcaLoadReport.UtilizationEntry */ |
||||
|
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_new(upb_arena *arena) { |
||||
return (udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *)upb_msg_new(&udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_msginit, arena); |
||||
} |
||||
UPB_INLINE udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_parse(const char *buf, size_t size, |
||||
upb_arena *arena) { |
||||
udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *ret = udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_new(arena); |
||||
return (ret && upb_decode(buf, size, ret, &udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_msginit, arena)) ? ret : NULL; |
||||
} |
||||
UPB_INLINE char *udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_serialize(const udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *msg, upb_arena *arena, size_t *len) { |
||||
return upb_encode(msg, &udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_msginit, arena, len); |
||||
} |
||||
|
||||
UPB_INLINE upb_strview udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_key(const udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *msg) { return UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(8, 8)); } |
||||
UPB_INLINE double udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_value(const udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *msg) { return UPB_FIELD_AT(msg, double, UPB_SIZE(0, 0)); } |
||||
|
||||
UPB_INLINE void udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_set_key(udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *msg, upb_strview value) { |
||||
UPB_FIELD_AT(msg, upb_strview, UPB_SIZE(8, 8)) = value; |
||||
} |
||||
UPB_INLINE void udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry_set_value(udpa_data_orca_v1_OrcaLoadReport_UtilizationEntry *msg, double value) { |
||||
UPB_FIELD_AT(msg, double, UPB_SIZE(0, 0)) = value; |
||||
} |
||||
|
||||
#ifdef __cplusplus |
||||
} /* extern "C" */ |
||||
#endif |
||||
|
||||
#include "upb/port_undef.inc" |
||||
|
||||
#endif /* UDPA_DATA_ORCA_V1_ORCA_LOAD_REPORT_PROTO_UPB_H_ */ |
||||
File diff suppressed because it is too large
Load Diff
@ -1,136 +0,0 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 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. |
||||
* |
||||
*/ |
||||
|
||||
/**
|
||||
* This is the legacy interface of this gRPC library. This API is deprecated and users should use |
||||
* the API in GRPCCall.h. This API exists solely for the purpose of backwards compatibility. |
||||
*/ |
||||
|
||||
#import <RxLibrary/GRXWriter.h> |
||||
#import "GRPCTypes.h" |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wnullability-completeness" |
||||
|
||||
/**
|
||||
* This interface is deprecated. Please use \a GRPCCall2. |
||||
* |
||||
* Represents a single gRPC remote call. |
||||
*/ |
||||
@interface GRPCCall : GRXWriter |
||||
|
||||
- (instancetype)init NS_UNAVAILABLE; |
||||
|
||||
/**
|
||||
* The container of the request headers of an RPC conforms to this protocol, which is a subset of |
||||
* NSMutableDictionary's interface. It will become a NSMutableDictionary later on. |
||||
* The keys of this container are the header names, which per the HTTP standard are case- |
||||
* insensitive. They are stored in lowercase (which is how HTTP/2 mandates them on the wire), and |
||||
* can only consist of ASCII characters. |
||||
* A header value is a NSString object (with only ASCII characters), unless the header name has the |
||||
* suffix "-bin", in which case the value has to be a NSData object. |
||||
*/ |
||||
/**
|
||||
* These HTTP headers will be passed to the server as part of this call. Each HTTP header is a |
||||
* name-value pair with string names and either string or binary values. |
||||
* |
||||
* The passed dictionary has to use NSString keys, corresponding to the header names. The value |
||||
* associated to each can be a NSString object or a NSData object. E.g.: |
||||
* |
||||
* call.requestHeaders = @{@"authorization": @"Bearer ..."}; |
||||
* |
||||
* call.requestHeaders[@"my-header-bin"] = someData; |
||||
* |
||||
* After the call is started, trying to modify this property is an error. |
||||
* |
||||
* The property is initialized to an empty NSMutableDictionary. |
||||
*/ |
||||
@property(atomic, readonly) NSMutableDictionary *requestHeaders; |
||||
|
||||
/**
|
||||
* This dictionary is populated with the HTTP headers received from the server. This happens before |
||||
* any response message is received from the server. It has the same structure as the request |
||||
* headers dictionary: Keys are NSString header names; names ending with the suffix "-bin" have a |
||||
* NSData value; the others have a NSString value. |
||||
* |
||||
* The value of this property is nil until all response headers are received, and will change before |
||||
* any of -writeValue: or -writesFinishedWithError: are sent to the writeable. |
||||
*/ |
||||
@property(atomic, readonly) NSDictionary *responseHeaders; |
||||
|
||||
/**
|
||||
* Same as responseHeaders, but populated with the HTTP trailers received from the server before the |
||||
* call finishes. |
||||
* |
||||
* The value of this property is nil until all response trailers are received, and will change |
||||
* before -writesFinishedWithError: is sent to the writeable. |
||||
*/ |
||||
@property(atomic, readonly) NSDictionary *responseTrailers; |
||||
|
||||
/**
|
||||
* The request writer has to write NSData objects into the provided Writeable. The server will |
||||
* receive each of those separately and in order as distinct messages. |
||||
* A gRPC call might not complete until the request writer finishes. On the other hand, the request |
||||
* finishing doesn't necessarily make the call to finish, as the server might continue sending |
||||
* messages to the response side of the call indefinitely (depending on the semantics of the |
||||
* specific remote method called). |
||||
* To finish a call right away, invoke cancel. |
||||
* host parameter should not contain the scheme (http:// or https://), only the name or IP addr
|
||||
* and the port number, for example @"localhost:5050". |
||||
*/ |
||||
- (instancetype)initWithHost:(NSString *)host |
||||
path:(NSString *)path |
||||
requestsWriter:(GRXWriter *)requestWriter; |
||||
|
||||
/**
|
||||
* Finishes the request side of this call, notifies the server that the RPC should be cancelled, and |
||||
* finishes the response side of the call with an error of code CANCELED. |
||||
*/ |
||||
- (void)cancel; |
||||
|
||||
/**
|
||||
* The following methods are deprecated. |
||||
*/ |
||||
+ (void)setCallSafety:(GRPCCallSafety)callSafety host:(NSString *)host path:(NSString *)path; |
||||
@property(atomic, copy, readwrite) NSString *serverName; |
||||
@property NSTimeInterval timeout; |
||||
- (void)setResponseDispatchQueue:(dispatch_queue_t)queue; |
||||
|
||||
@end |
||||
|
||||
#pragma mark Backwards compatibiity |
||||
|
||||
/** This protocol is kept for backwards compatibility with existing code. */ |
||||
DEPRECATED_MSG_ATTRIBUTE("Use NSDictionary or NSMutableDictionary instead.") |
||||
@protocol GRPCRequestHeaders<NSObject> |
||||
@property(nonatomic, readonly) NSUInteger count; |
||||
|
||||
- (id)objectForKeyedSubscript:(id)key; |
||||
- (void)setObject:(id)obj forKeyedSubscript:(id)key; |
||||
|
||||
- (void)removeAllObjects; |
||||
- (void)removeObjectForKey:(id)key; |
||||
@end |
||||
|
||||
#pragma clang diagnostic push |
||||
#pragma clang diagnostic ignored "-Wdeprecated" |
||||
/** This is only needed for backwards-compatibility. */ |
||||
@interface NSMutableDictionary (GRPCRequestHeaders)<GRPCRequestHeaders> |
||||
@end |
||||
#pragma clang diagnostic pop |
||||
#pragma clang diagnostic pop |
||||
@ -1,677 +0,0 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 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. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCCallLegacy.h" |
||||
|
||||
#import "GRPCCall+OAuth2.h" |
||||
#import "GRPCCallOptions.h" |
||||
#import "GRPCTypes.h" |
||||
|
||||
#import "private/GRPCCore/GRPCChannelPool.h" |
||||
#import "private/GRPCCore/GRPCCompletionQueue.h" |
||||
#import "private/GRPCCore/GRPCHost.h" |
||||
#import "private/GRPCCore/GRPCWrappedCall.h" |
||||
#import "private/GRPCCore/NSData+GRPC.h" |
||||
|
||||
#import <RxLibrary/GRXBufferedPipe.h> |
||||
#import <RxLibrary/GRXConcurrentWriteable.h> |
||||
#import <RxLibrary/GRXImmediateSingleWriter.h> |
||||
#import <RxLibrary/GRXWriter+Immediate.h> |
||||
|
||||
#include <grpc/grpc.h> |
||||
|
||||
const char *kCFStreamVarName = "grpc_cfstream"; |
||||
static NSMutableDictionary *callFlags; |
||||
|
||||
// At most 6 ops can be in an op batch for a client: SEND_INITIAL_METADATA, |
||||
// SEND_MESSAGE, SEND_CLOSE_FROM_CLIENT, RECV_INITIAL_METADATA, RECV_MESSAGE, |
||||
// and RECV_STATUS_ON_CLIENT. |
||||
NSInteger kMaxClientBatch = 6; |
||||
|
||||
static NSString *const kAuthorizationHeader = @"authorization"; |
||||
static NSString *const kBearerPrefix = @"Bearer "; |
||||
|
||||
@interface GRPCCall ()<GRXWriteable> |
||||
// Make them read-write. |
||||
@property(atomic, strong) NSDictionary *responseHeaders; |
||||
@property(atomic, strong) NSDictionary *responseTrailers; |
||||
|
||||
- (void)receiveNextMessages:(NSUInteger)numberOfMessages; |
||||
|
||||
@end |
||||
|
||||
// The following methods of a C gRPC call object aren't reentrant, and thus |
||||
// calls to them must be serialized: |
||||
// - start_batch |
||||
// - destroy |
||||
// |
||||
// start_batch with a SEND_MESSAGE argument can only be called after the |
||||
// OP_COMPLETE event for any previous write is received. This is achieved by |
||||
// pausing the requests writer immediately every time it writes a value, and |
||||
// resuming it again when OP_COMPLETE is received. |
||||
// |
||||
// Similarly, start_batch with a RECV_MESSAGE argument can only be called after |
||||
// the OP_COMPLETE event for any previous read is received.This is easier to |
||||
// enforce, as we're writing the received messages into the writeable: |
||||
// start_batch is enqueued once upon receiving the OP_COMPLETE event for the |
||||
// RECV_METADATA batch, and then once after receiving each OP_COMPLETE event for |
||||
// each RECV_MESSAGE batch. |
||||
@implementation GRPCCall { |
||||
dispatch_queue_t _callQueue; |
||||
|
||||
NSString *_host; |
||||
NSString *_path; |
||||
GRPCCallSafety _callSafety; |
||||
GRPCCallOptions *_callOptions; |
||||
GRPCWrappedCall *_wrappedCall; |
||||
|
||||
// The C gRPC library has less guarantees on the ordering of events than we |
||||
// do. Particularly, in the face of errors, there's no ordering guarantee at |
||||
// all. This wrapper over our actual writeable ensures thread-safety and |
||||
// correct ordering. |
||||
GRXConcurrentWriteable *_responseWriteable; |
||||
|
||||
// The network thread wants the requestWriter to resume (when the server is ready for more input), |
||||
// or to stop (on errors), concurrently with user threads that want to start it, pause it or stop |
||||
// it. Because a writer isn't thread-safe, we'll synchronize those operations on it. |
||||
// We don't use a dispatch queue for that purpose, because the writer can call writeValue: or |
||||
// writesFinishedWithError: on this GRPCCall as part of those operations. We want to be able to |
||||
// pause the writer immediately on writeValue:, so we need our locking to be recursive. |
||||
GRXWriter *_requestWriter; |
||||
|
||||
// To create a retain cycle when a call is started, up until it finishes. See |
||||
// |startWithWriteable:| and |finishWithError:|. This saves users from having to retain a |
||||
// reference to the call object if all they're interested in is the handler being executed when |
||||
// the response arrives. |
||||
GRPCCall *_retainSelf; |
||||
|
||||
GRPCRequestHeaders *_requestHeaders; |
||||
|
||||
// In the case that the call is a unary call (i.e. the writer to GRPCCall is of type |
||||
// GRXImmediateSingleWriter), GRPCCall will delay sending ops (not send them to C core |
||||
// immediately) and buffer them into a batch _unaryOpBatch. The batch is sent to C core when |
||||
// the SendClose op is added. |
||||
BOOL _unaryCall; |
||||
NSMutableArray *_unaryOpBatch; |
||||
|
||||
// The dispatch queue to be used for enqueuing responses to user. Defaulted to the main dispatch |
||||
// queue |
||||
dispatch_queue_t _responseQueue; |
||||
|
||||
// The OAuth2 token fetched from a token provider. |
||||
NSString *_fetchedOauth2AccessToken; |
||||
|
||||
// The callback to be called when a write message op is done. |
||||
void (^_writeDone)(void); |
||||
|
||||
// Indicate a read request to core is pending. |
||||
BOOL _pendingCoreRead; |
||||
|
||||
// Indicate pending read message request from user. |
||||
NSUInteger _pendingReceiveNextMessages; |
||||
} |
||||
|
||||
@synthesize state = _state; |
||||
|
||||
+ (void)initialize { |
||||
// Guarantees the code in {} block is invoked only once. See ref at: |
||||
// https://developer.apple.com/documentation/objectivec/nsobject/1418639-initialize?language=objc |
||||
if (self == [GRPCCall self]) { |
||||
grpc_init(); |
||||
callFlags = [NSMutableDictionary dictionary]; |
||||
} |
||||
} |
||||
|
||||
+ (void)setCallSafety:(GRPCCallSafety)callSafety host:(NSString *)host path:(NSString *)path { |
||||
if (host.length == 0 || path.length == 0) { |
||||
return; |
||||
} |
||||
NSString *hostAndPath = [NSString stringWithFormat:@"%@/%@", host, path]; |
||||
@synchronized(callFlags) { |
||||
switch (callSafety) { |
||||
case GRPCCallSafetyDefault: |
||||
callFlags[hostAndPath] = @0; |
||||
break; |
||||
case GRPCCallSafetyIdempotentRequest: |
||||
callFlags[hostAndPath] = @GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST; |
||||
break; |
||||
case GRPCCallSafetyCacheableRequest: |
||||
callFlags[hostAndPath] = @GRPC_INITIAL_METADATA_CACHEABLE_REQUEST; |
||||
break; |
||||
default: |
||||
break; |
||||
} |
||||
} |
||||
} |
||||
|
||||
+ (uint32_t)callFlagsForHost:(NSString *)host path:(NSString *)path { |
||||
NSString *hostAndPath = [NSString stringWithFormat:@"%@/%@", host, path]; |
||||
@synchronized(callFlags) { |
||||
return [callFlags[hostAndPath] intValue]; |
||||
} |
||||
} |
||||
|
||||
- (instancetype)initWithHost:(NSString *)host |
||||
path:(NSString *)path |
||||
requestsWriter:(GRXWriter *)requestWriter { |
||||
return [self initWithHost:host |
||||
path:path |
||||
callSafety:GRPCCallSafetyDefault |
||||
requestsWriter:requestWriter |
||||
callOptions:nil |
||||
writeDone:nil]; |
||||
} |
||||
|
||||
- (instancetype)initWithHost:(NSString *)host |
||||
path:(NSString *)path |
||||
callSafety:(GRPCCallSafety)safety |
||||
requestsWriter:(GRXWriter *)requestsWriter |
||||
callOptions:(GRPCCallOptions *)callOptions |
||||
writeDone:(void (^)(void))writeDone { |
||||
// Purposely using pointer rather than length (host.length == 0) for backwards compatibility. |
||||
NSAssert(host != nil && path != nil, @"Neither host nor path can be nil."); |
||||
NSAssert(safety <= GRPCCallSafetyCacheableRequest, @"Invalid call safety value."); |
||||
NSAssert(requestsWriter.state == GRXWriterStateNotStarted, |
||||
@"The requests writer can't be already started."); |
||||
if (!host || !path) { |
||||
return nil; |
||||
} |
||||
if (safety > GRPCCallSafetyCacheableRequest) { |
||||
return nil; |
||||
} |
||||
if (requestsWriter.state != GRXWriterStateNotStarted) { |
||||
return nil; |
||||
} |
||||
|
||||
if ((self = [super init])) { |
||||
_host = [host copy]; |
||||
_path = [path copy]; |
||||
_callSafety = safety; |
||||
_callOptions = [callOptions copy]; |
||||
|
||||
// Serial queue to invoke the non-reentrant methods of the grpc_call object. |
||||
_callQueue = dispatch_queue_create("io.grpc.call", DISPATCH_QUEUE_SERIAL); |
||||
|
||||
_requestWriter = requestsWriter; |
||||
_requestHeaders = [[GRPCRequestHeaders alloc] initWithCall:self]; |
||||
_writeDone = writeDone; |
||||
|
||||
if ([requestsWriter isKindOfClass:[GRXImmediateSingleWriter class]]) { |
||||
_unaryCall = YES; |
||||
_unaryOpBatch = [NSMutableArray arrayWithCapacity:kMaxClientBatch]; |
||||
} |
||||
|
||||
_responseQueue = dispatch_get_main_queue(); |
||||
|
||||
// do not start a read until initial metadata is received |
||||
_pendingReceiveNextMessages = 0; |
||||
_pendingCoreRead = YES; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)setResponseDispatchQueue:(dispatch_queue_t)queue { |
||||
@synchronized(self) { |
||||
if (_state != GRXWriterStateNotStarted) { |
||||
return; |
||||
} |
||||
_responseQueue = queue; |
||||
} |
||||
} |
||||
|
||||
#pragma mark Finish |
||||
|
||||
// This function should support being called within a @synchronized(self) block in another function |
||||
// Should not manipulate _requestWriter for deadlock prevention. |
||||
- (void)finishWithError:(NSError *)errorOrNil { |
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
_state = GRXWriterStateFinished; |
||||
|
||||
if (errorOrNil) { |
||||
[_responseWriteable cancelWithError:errorOrNil]; |
||||
} else { |
||||
[_responseWriteable enqueueSuccessfulCompletion]; |
||||
} |
||||
|
||||
// If the call isn't retained anywhere else, it can be deallocated now. |
||||
_retainSelf = nil; |
||||
} |
||||
} |
||||
|
||||
- (void)cancel { |
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
[self finishWithError:[NSError |
||||
errorWithDomain:kGRPCErrorDomain |
||||
code:GRPCErrorCodeCancelled |
||||
userInfo:@{NSLocalizedDescriptionKey : @"Canceled by app"}]]; |
||||
[_wrappedCall cancel]; |
||||
} |
||||
_requestWriter.state = GRXWriterStateFinished; |
||||
} |
||||
|
||||
- (void)dealloc { |
||||
__block GRPCWrappedCall *wrappedCall = _wrappedCall; |
||||
dispatch_async(_callQueue, ^{ |
||||
wrappedCall = nil; |
||||
}); |
||||
} |
||||
|
||||
#pragma mark Read messages |
||||
|
||||
// Only called from the call queue. |
||||
// The handler will be called from the network queue. |
||||
- (void)startReadWithHandler:(void (^)(grpc_byte_buffer *))handler { |
||||
// TODO(jcanizales): Add error handlers for async failures |
||||
[_wrappedCall startBatchWithOperations:@[ [[GRPCOpRecvMessage alloc] initWithHandler:handler] ]]; |
||||
} |
||||
|
||||
// Called initially from the network queue once response headers are received, |
||||
// then "recursively" from the responseWriteable queue after each response from the |
||||
// server has been written. |
||||
// If the call is currently paused, this is a noop. Restarting the call will invoke this |
||||
// method. |
||||
// TODO(jcanizales): Rename to readResponseIfNotPaused. |
||||
- (void)maybeStartNextRead { |
||||
@synchronized(self) { |
||||
if (_state != GRXWriterStateStarted) { |
||||
return; |
||||
} |
||||
if (_callOptions.flowControlEnabled && (_pendingCoreRead || _pendingReceiveNextMessages == 0)) { |
||||
return; |
||||
} |
||||
_pendingCoreRead = YES; |
||||
_pendingReceiveNextMessages--; |
||||
} |
||||
|
||||
dispatch_async(_callQueue, ^{ |
||||
__weak GRPCCall *weakSelf = self; |
||||
[self startReadWithHandler:^(grpc_byte_buffer *message) { |
||||
if (message == NULL) { |
||||
// No more messages from the server |
||||
return; |
||||
} |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf == nil) { |
||||
grpc_byte_buffer_destroy(message); |
||||
return; |
||||
} |
||||
NSData *data = [NSData grpc_dataWithByteBuffer:message]; |
||||
grpc_byte_buffer_destroy(message); |
||||
if (!data) { |
||||
// The app doesn't have enough memory to hold the server response. We |
||||
// don't want to throw, because the app shouldn't crash for a behavior |
||||
// that's on the hands of any server to have. Instead we finish and ask |
||||
// the server to cancel. |
||||
@synchronized(strongSelf) { |
||||
strongSelf->_pendingCoreRead = NO; |
||||
[strongSelf |
||||
finishWithError:[NSError errorWithDomain:kGRPCErrorDomain |
||||
code:GRPCErrorCodeResourceExhausted |
||||
userInfo:@{ |
||||
NSLocalizedDescriptionKey : |
||||
@"Client does not have enough memory to " |
||||
@"hold the server response." |
||||
}]]; |
||||
[strongSelf->_wrappedCall cancel]; |
||||
} |
||||
strongSelf->_requestWriter.state = GRXWriterStateFinished; |
||||
} else { |
||||
@synchronized(strongSelf) { |
||||
[strongSelf->_responseWriteable enqueueValue:data |
||||
completionHandler:^{ |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
@synchronized(strongSelf) { |
||||
strongSelf->_pendingCoreRead = NO; |
||||
[strongSelf maybeStartNextRead]; |
||||
} |
||||
} |
||||
}]; |
||||
} |
||||
} |
||||
}]; |
||||
}); |
||||
} |
||||
|
||||
#pragma mark Send headers |
||||
|
||||
- (void)sendHeaders { |
||||
// TODO (mxyan): Remove after deprecated methods are removed |
||||
uint32_t callSafetyFlags = 0; |
||||
switch (_callSafety) { |
||||
case GRPCCallSafetyDefault: |
||||
callSafetyFlags = 0; |
||||
break; |
||||
case GRPCCallSafetyIdempotentRequest: |
||||
callSafetyFlags = GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST; |
||||
break; |
||||
case GRPCCallSafetyCacheableRequest: |
||||
callSafetyFlags = GRPC_INITIAL_METADATA_CACHEABLE_REQUEST; |
||||
break; |
||||
} |
||||
|
||||
NSMutableDictionary *headers = [_requestHeaders mutableCopy]; |
||||
NSString *fetchedOauth2AccessToken; |
||||
@synchronized(self) { |
||||
fetchedOauth2AccessToken = _fetchedOauth2AccessToken; |
||||
} |
||||
if (fetchedOauth2AccessToken != nil) { |
||||
headers[@"authorization"] = [kBearerPrefix stringByAppendingString:fetchedOauth2AccessToken]; |
||||
} else if (_callOptions.oauth2AccessToken != nil) { |
||||
headers[@"authorization"] = |
||||
[kBearerPrefix stringByAppendingString:_callOptions.oauth2AccessToken]; |
||||
} |
||||
|
||||
// TODO(jcanizales): Add error handlers for async failures |
||||
GRPCOpSendMetadata *op = [[GRPCOpSendMetadata alloc] |
||||
initWithMetadata:headers |
||||
flags:callSafetyFlags |
||||
handler:nil]; // No clean-up needed after SEND_INITIAL_METADATA |
||||
dispatch_async(_callQueue, ^{ |
||||
if (!self->_unaryCall) { |
||||
[self->_wrappedCall startBatchWithOperations:@[ op ]]; |
||||
} else { |
||||
[self->_unaryOpBatch addObject:op]; |
||||
} |
||||
}); |
||||
} |
||||
|
||||
- (void)receiveNextMessages:(NSUInteger)numberOfMessages { |
||||
if (numberOfMessages == 0) { |
||||
return; |
||||
} |
||||
@synchronized(self) { |
||||
_pendingReceiveNextMessages += numberOfMessages; |
||||
|
||||
if (_state != GRXWriterStateStarted || !_callOptions.flowControlEnabled) { |
||||
return; |
||||
} |
||||
[self maybeStartNextRead]; |
||||
} |
||||
} |
||||
|
||||
#pragma mark GRXWriteable implementation |
||||
|
||||
// Only called from the call queue. The error handler will be called from the |
||||
// network queue if the write didn't succeed. |
||||
// If the call is a unary call, parameter \a errorHandler will be ignored and |
||||
// the error handler of GRPCOpSendClose will be executed in case of error. |
||||
- (void)writeMessage:(NSData *)message withErrorHandler:(void (^)(void))errorHandler { |
||||
__weak GRPCCall *weakSelf = self; |
||||
void (^resumingHandler)(void) = ^{ |
||||
// Resume the request writer. |
||||
GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
strongSelf->_requestWriter.state = GRXWriterStateStarted; |
||||
if (strongSelf->_writeDone) { |
||||
strongSelf->_writeDone(); |
||||
} |
||||
} |
||||
}; |
||||
GRPCOpSendMessage *op = |
||||
[[GRPCOpSendMessage alloc] initWithMessage:message handler:resumingHandler]; |
||||
if (!_unaryCall) { |
||||
[_wrappedCall startBatchWithOperations:@[ op ] errorHandler:errorHandler]; |
||||
} else { |
||||
// Ignored errorHandler since it is the same as the one for GRPCOpSendClose. |
||||
// TODO (mxyan): unify the error handlers of all Ops into a single closure. |
||||
[_unaryOpBatch addObject:op]; |
||||
} |
||||
} |
||||
|
||||
- (void)writeValue:(id)value { |
||||
NSAssert([value isKindOfClass:[NSData class]], @"value must be of type NSData"); |
||||
|
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
} |
||||
|
||||
// Pause the input and only resume it when the C layer notifies us that writes |
||||
// can proceed. |
||||
_requestWriter.state = GRXWriterStatePaused; |
||||
|
||||
dispatch_async(_callQueue, ^{ |
||||
// Write error is not processed here. It is handled by op batch of GRPC_OP_RECV_STATUS_ON_CLIENT |
||||
[self writeMessage:value withErrorHandler:nil]; |
||||
}); |
||||
} |
||||
|
||||
// Only called from the call queue. The error handler will be called from the |
||||
// network queue if the requests stream couldn't be closed successfully. |
||||
- (void)finishRequestWithErrorHandler:(void (^)(void))errorHandler { |
||||
if (!_unaryCall) { |
||||
[_wrappedCall startBatchWithOperations:@[ [[GRPCOpSendClose alloc] init] ] |
||||
errorHandler:errorHandler]; |
||||
} else { |
||||
[_unaryOpBatch addObject:[[GRPCOpSendClose alloc] init]]; |
||||
[_wrappedCall startBatchWithOperations:_unaryOpBatch errorHandler:errorHandler]; |
||||
} |
||||
} |
||||
|
||||
- (void)writesFinishedWithError:(NSError *)errorOrNil { |
||||
if (errorOrNil) { |
||||
[self cancel]; |
||||
} else { |
||||
dispatch_async(_callQueue, ^{ |
||||
// EOS error is not processed here. It is handled by op batch of GRPC_OP_RECV_STATUS_ON_CLIENT |
||||
[self finishRequestWithErrorHandler:nil]; |
||||
}); |
||||
} |
||||
} |
||||
|
||||
#pragma mark Invoke |
||||
|
||||
// Both handlers will eventually be called, from the network queue. Writes can start immediately |
||||
// after this. |
||||
// The first one (headersHandler), when the response headers are received. |
||||
// The second one (completionHandler), whenever the RPC finishes for any reason. |
||||
- (void)invokeCallWithHeadersHandler:(void (^)(NSDictionary *))headersHandler |
||||
completionHandler:(void (^)(NSError *, NSDictionary *))completionHandler { |
||||
dispatch_async(_callQueue, ^{ |
||||
// TODO(jcanizales): Add error handlers for async failures |
||||
[self->_wrappedCall |
||||
startBatchWithOperations:@[ [[GRPCOpRecvMetadata alloc] initWithHandler:headersHandler] ]]; |
||||
[self->_wrappedCall |
||||
startBatchWithOperations:@[ [[GRPCOpRecvStatus alloc] initWithHandler:completionHandler] ]]; |
||||
}); |
||||
} |
||||
|
||||
- (void)invokeCall { |
||||
__weak GRPCCall *weakSelf = self; |
||||
[self invokeCallWithHeadersHandler:^(NSDictionary *headers) { |
||||
// Response headers received. |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
@synchronized(strongSelf) { |
||||
// it is ok to set nil because headers are only received once |
||||
strongSelf.responseHeaders = nil; |
||||
// copy the header so that the GRPCOpRecvMetadata object may be dealloc'ed |
||||
NSDictionary *copiedHeaders = |
||||
[[NSDictionary alloc] initWithDictionary:headers copyItems:YES]; |
||||
strongSelf.responseHeaders = copiedHeaders; |
||||
strongSelf->_pendingCoreRead = NO; |
||||
[strongSelf maybeStartNextRead]; |
||||
} |
||||
} |
||||
} |
||||
completionHandler:^(NSError *error, NSDictionary *trailers) { |
||||
__strong GRPCCall *strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
strongSelf.responseTrailers = trailers; |
||||
|
||||
if (error) { |
||||
NSMutableDictionary *userInfo = [NSMutableDictionary dictionary]; |
||||
if (error.userInfo) { |
||||
[userInfo addEntriesFromDictionary:error.userInfo]; |
||||
} |
||||
userInfo[kGRPCTrailersKey] = strongSelf.responseTrailers; |
||||
// Since gRPC core does not guarantee the headers block being called before this block, |
||||
// responseHeaders might be nil. |
||||
userInfo[kGRPCHeadersKey] = strongSelf.responseHeaders; |
||||
error = [NSError errorWithDomain:error.domain code:error.code userInfo:userInfo]; |
||||
} |
||||
[strongSelf finishWithError:error]; |
||||
strongSelf->_requestWriter.state = GRXWriterStateFinished; |
||||
} |
||||
}]; |
||||
} |
||||
|
||||
#pragma mark GRXWriter implementation |
||||
|
||||
// Lock acquired inside startWithWriteable: |
||||
- (void)startCallWithWriteable:(id<GRXWriteable>)writeable { |
||||
@synchronized(self) { |
||||
if (_state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
|
||||
_responseWriteable = |
||||
[[GRXConcurrentWriteable alloc] initWithWriteable:writeable dispatchQueue:_responseQueue]; |
||||
|
||||
GRPCPooledChannel *channel = |
||||
[[GRPCChannelPool sharedInstance] channelWithHost:_host callOptions:_callOptions]; |
||||
_wrappedCall = [channel wrappedCallWithPath:_path |
||||
completionQueue:[GRPCCompletionQueue completionQueue] |
||||
callOptions:_callOptions]; |
||||
|
||||
if (_wrappedCall == nil) { |
||||
[self finishWithError:[NSError errorWithDomain:kGRPCErrorDomain |
||||
code:GRPCErrorCodeUnavailable |
||||
userInfo:@{ |
||||
NSLocalizedDescriptionKey : |
||||
@"Failed to create call or channel." |
||||
}]]; |
||||
return; |
||||
} |
||||
|
||||
[self sendHeaders]; |
||||
[self invokeCall]; |
||||
} |
||||
|
||||
// Now that the RPC has been initiated, request writes can start. |
||||
[_requestWriter startWithWriteable:self]; |
||||
} |
||||
|
||||
- (void)startWithWriteable:(id<GRXWriteable>)writeable { |
||||
id<GRPCAuthorizationProtocol> tokenProvider = nil; |
||||
@synchronized(self) { |
||||
_state = GRXWriterStateStarted; |
||||
|
||||
// Create a retain cycle so that this instance lives until the RPC finishes (or is cancelled). |
||||
// This makes RPCs in which the call isn't externally retained possible (as long as it is |
||||
// started before being autoreleased). Care is taken not to retain self strongly in any of the |
||||
// blocks used in this implementation, so that the life of the instance is determined by this |
||||
// retain cycle. |
||||
_retainSelf = self; |
||||
|
||||
// If _callOptions is nil, people must be using the deprecated v1 interface. In this case, |
||||
// generate the call options from the corresponding GRPCHost configs and apply other options |
||||
// that are not covered by GRPCHost. |
||||
if (_callOptions == nil) { |
||||
GRPCMutableCallOptions *callOptions = [[GRPCHost callOptionsForHost:_host] mutableCopy]; |
||||
if (_serverName.length != 0) { |
||||
callOptions.serverAuthority = _serverName; |
||||
} |
||||
if (_timeout > 0) { |
||||
callOptions.timeout = _timeout; |
||||
} |
||||
uint32_t callFlags = [GRPCCall callFlagsForHost:_host path:_path]; |
||||
if (callFlags != 0) { |
||||
if (callFlags == GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST) { |
||||
_callSafety = GRPCCallSafetyIdempotentRequest; |
||||
} else if (callFlags == GRPC_INITIAL_METADATA_CACHEABLE_REQUEST) { |
||||
_callSafety = GRPCCallSafetyCacheableRequest; |
||||
} |
||||
} |
||||
|
||||
id<GRPCAuthorizationProtocol> tokenProvider = self.tokenProvider; |
||||
if (tokenProvider != nil) { |
||||
callOptions.authTokenProvider = tokenProvider; |
||||
} |
||||
_callOptions = callOptions; |
||||
} |
||||
|
||||
NSAssert(_callOptions.authTokenProvider == nil || _callOptions.oauth2AccessToken == nil, |
||||
@"authTokenProvider and oauth2AccessToken cannot be set at the same time"); |
||||
|
||||
tokenProvider = _callOptions.authTokenProvider; |
||||
} |
||||
|
||||
if (tokenProvider != nil) { |
||||
__weak typeof(self) weakSelf = self; |
||||
[tokenProvider getTokenWithHandler:^(NSString *token) { |
||||
__strong typeof(self) strongSelf = weakSelf; |
||||
if (strongSelf) { |
||||
BOOL startCall = NO; |
||||
@synchronized(strongSelf) { |
||||
if (strongSelf->_state != GRXWriterStateFinished) { |
||||
startCall = YES; |
||||
if (token) { |
||||
strongSelf->_fetchedOauth2AccessToken = [token copy]; |
||||
} |
||||
} |
||||
} |
||||
if (startCall) { |
||||
[strongSelf startCallWithWriteable:writeable]; |
||||
} |
||||
} |
||||
}]; |
||||
} else { |
||||
[self startCallWithWriteable:writeable]; |
||||
} |
||||
} |
||||
|
||||
- (void)setState:(GRXWriterState)newState { |
||||
@synchronized(self) { |
||||
// Manual transitions are only allowed from the started or paused states. |
||||
if (_state == GRXWriterStateNotStarted || _state == GRXWriterStateFinished) { |
||||
return; |
||||
} |
||||
|
||||
switch (newState) { |
||||
case GRXWriterStateFinished: |
||||
_state = newState; |
||||
// Per GRXWriter's contract, setting the state to Finished manually |
||||
// means one doesn't wish the writeable to be messaged anymore. |
||||
[_responseWriteable cancelSilently]; |
||||
_responseWriteable = nil; |
||||
return; |
||||
case GRXWriterStatePaused: |
||||
_state = newState; |
||||
return; |
||||
case GRXWriterStateStarted: |
||||
if (_state == GRXWriterStatePaused) { |
||||
_state = newState; |
||||
[self maybeStartNextRead]; |
||||
} |
||||
return; |
||||
case GRXWriterStateNotStarted: |
||||
return; |
||||
} |
||||
} |
||||
} |
||||
|
||||
@end |
||||
@ -1,30 +0,0 @@ |
||||
|
||||
/*
|
||||
* |
||||
* Copyright 2019 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. |
||||
* |
||||
*/ |
||||
|
||||
/**
|
||||
* An object that processes its methods with a dispatch queue. |
||||
*/ |
||||
@protocol GRPCDispatchable |
||||
|
||||
/**
|
||||
* The dispatch queue where the object's methods should be run on. |
||||
*/ |
||||
@property(atomic, readonly) dispatch_queue_t dispatchQueue; |
||||
|
||||
@end |
||||
@ -1,82 +0,0 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 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. |
||||
* |
||||
*/ |
||||
|
||||
// The interface for a transport implementation
|
||||
|
||||
#import "GRPCInterceptor.h" |
||||
|
||||
NS_ASSUME_NONNULL_BEGIN |
||||
|
||||
#pragma mark Transport ID |
||||
|
||||
/**
|
||||
* The default transport implementations available in gRPC. These implementations will be provided |
||||
* by gRPC by default unless explicitly excluded. |
||||
*/ |
||||
extern const struct GRPCDefaultTransportImplList { |
||||
const GRPCTransportId core_secure; |
||||
const GRPCTransportId core_insecure; |
||||
} GRPCDefaultTransportImplList; |
||||
|
||||
/** Returns whether two transport id's are identical. */ |
||||
BOOL TransportIdIsEqual(GRPCTransportId lhs, GRPCTransportId rhs); |
||||
|
||||
/** Returns the hash value of a transport id. */ |
||||
NSUInteger TransportIdHash(GRPCTransportId); |
||||
|
||||
#pragma mark Transport and factory |
||||
|
||||
@protocol GRPCInterceptorInterface; |
||||
@protocol GRPCResponseHandler; |
||||
@class GRPCTransportManager; |
||||
@class GRPCRequestOptions; |
||||
@class GRPCCallOptions; |
||||
@class GRPCTransport; |
||||
|
||||
/** The factory method to create a transport. */ |
||||
@protocol GRPCTransportFactory<NSObject> |
||||
|
||||
- (GRPCTransport *)createTransportWithManager:(GRPCTransportManager *)transportManager; |
||||
|
||||
@end |
||||
|
||||
/** The registry of transport implementations. */ |
||||
@interface GRPCTransportRegistry : NSObject |
||||
|
||||
+ (instancetype)sharedInstance; |
||||
|
||||
/**
|
||||
* Register a transport implementation with the registry. All transport implementations to be used |
||||
* in a process must register with the registry on process start-up in its +load: class method. |
||||
* Parameter \a transportId is the identifier of the implementation, and \a factory is the factory |
||||
* object to create the corresponding transport instance. |
||||
*/ |
||||
- (void)registerTransportWithId:(GRPCTransportId)transportId |
||||
factory:(id<GRPCTransportFactory>)factory; |
||||
|
||||
@end |
||||
|
||||
/**
|
||||
* Base class for transport implementations. All transport implementation should inherit from this |
||||
* class. |
||||
*/ |
||||
@interface GRPCTransport : NSObject<GRPCInterceptorInterface> |
||||
|
||||
@end |
||||
|
||||
NS_ASSUME_NONNULL_END |
||||
@ -1,142 +0,0 @@ |
||||
/* |
||||
* |
||||
* Copyright 2019 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. |
||||
* |
||||
*/ |
||||
|
||||
#import "GRPCTransport.h" |
||||
|
||||
static const GRPCTransportId gGRPCCoreSecureId = "io.grpc.transport.core.secure"; |
||||
static const GRPCTransportId gGRPCCoreInsecureId = "io.grpc.transport.core.insecure"; |
||||
|
||||
const struct GRPCDefaultTransportImplList GRPCDefaultTransportImplList = { |
||||
.core_secure = gGRPCCoreSecureId, .core_insecure = gGRPCCoreInsecureId}; |
||||
|
||||
static const GRPCTransportId gDefaultTransportId = gGRPCCoreSecureId; |
||||
|
||||
static GRPCTransportRegistry *gTransportRegistry = nil; |
||||
static dispatch_once_t initTransportRegistry; |
||||
|
||||
BOOL TransportIdIsEqual(GRPCTransportId lhs, GRPCTransportId rhs) { |
||||
// Directly comparing pointers works because we require users to use the id provided by each |
||||
// implementation, not coming up with their own string. |
||||
return lhs == rhs; |
||||
} |
||||
|
||||
NSUInteger TransportIdHash(GRPCTransportId transportId) { |
||||
if (transportId == NULL) { |
||||
transportId = gDefaultTransportId; |
||||
} |
||||
return [NSString stringWithCString:transportId encoding:NSUTF8StringEncoding].hash; |
||||
} |
||||
|
||||
@implementation GRPCTransportRegistry { |
||||
NSMutableDictionary<NSString *, id<GRPCTransportFactory>> *_registry; |
||||
id<GRPCTransportFactory> _defaultFactory; |
||||
} |
||||
|
||||
+ (instancetype)sharedInstance { |
||||
dispatch_once(&initTransportRegistry, ^{ |
||||
gTransportRegistry = [[GRPCTransportRegistry alloc] init]; |
||||
NSAssert(gTransportRegistry != nil, @"Unable to initialize transport registry."); |
||||
if (gTransportRegistry == nil) { |
||||
NSLog(@"Unable to initialize transport registry."); |
||||
[NSException raise:NSGenericException format:@"Unable to initialize transport registry."]; |
||||
} |
||||
}); |
||||
return gTransportRegistry; |
||||
} |
||||
|
||||
- (instancetype)init { |
||||
if ((self = [super init])) { |
||||
_registry = [NSMutableDictionary dictionary]; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
- (void)registerTransportWithId:(GRPCTransportId)transportId |
||||
factory:(id<GRPCTransportFactory>)factory { |
||||
NSString *nsTransportId = [NSString stringWithCString:transportId encoding:NSUTF8StringEncoding]; |
||||
NSAssert(_registry[nsTransportId] == nil, @"The transport %@ has already been registered.", |
||||
nsTransportId); |
||||
if (_registry[nsTransportId] != nil) { |
||||
NSLog(@"The transport %@ has already been registered.", nsTransportId); |
||||
return; |
||||
} |
||||
_registry[nsTransportId] = factory; |
||||
|
||||
// if the default transport is registered, mark it. |
||||
if (0 == strcmp(transportId, gDefaultTransportId)) { |
||||
_defaultFactory = factory; |
||||
} |
||||
} |
||||
|
||||
- (id<GRPCTransportFactory>)getTransportFactoryWithId:(GRPCTransportId)transportId { |
||||
if (transportId == NULL) { |
||||
if (_defaultFactory == nil) { |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"Unable to get default transport factory"]; |
||||
return nil; |
||||
} |
||||
return _defaultFactory; |
||||
} |
||||
NSString *nsTransportId = [NSString stringWithCString:transportId encoding:NSUTF8StringEncoding]; |
||||
id<GRPCTransportFactory> transportFactory = _registry[nsTransportId]; |
||||
if (transportFactory == nil) { |
||||
// User named a transport id that was not registered with the registry. |
||||
[NSException raise:NSInvalidArgumentException |
||||
format:@"Unable to get transport factory with id %s", transportId]; |
||||
return nil; |
||||
} |
||||
return transportFactory; |
||||
} |
||||
|
||||
@end |
||||
|
||||
@implementation GRPCTransport |
||||
|
||||
- (dispatch_queue_t)dispatchQueue { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the dispatch queue"]; |
||||
return nil; |
||||
} |
||||
|
||||
- (void)startWithRequestOptions:(nonnull GRPCRequestOptions *)requestOptions |
||||
callOptions:(nonnull GRPCCallOptions *)callOptions { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)writeData:(nonnull id)data { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)cancel { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)finish { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
- (void)receiveNextMessages:(NSUInteger)numberOfMessages { |
||||
[NSException raise:NSGenericException |
||||
format:@"Implementations should override the methods of GRPCTransport"]; |
||||
} |
||||
|
||||
@end |
||||
@ -1,187 +0,0 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 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. |
||||
* |
||||
*/ |
||||
|
||||
/**
|
||||
* gRPC error codes. |
||||
* Note that a few of these are never produced by the gRPC libraries, but are of |
||||
* general utility for server applications to produce. |
||||
*/ |
||||
typedef NS_ENUM(NSUInteger, GRPCErrorCode) { |
||||
/** The operation was cancelled (typically by the caller). */ |
||||
GRPCErrorCodeCancelled = 1, |
||||
|
||||
/**
|
||||
* Unknown error. Errors raised by APIs that do not return enough error |
||||
* information may be converted to this error. |
||||
*/ |
||||
GRPCErrorCodeUnknown = 2, |
||||
|
||||
/**
|
||||
* The client specified an invalid argument. Note that this differs from |
||||
* FAILED_PRECONDITION. INVALID_ARGUMENT indicates arguments that are |
||||
* problematic regardless of the state of the server (e.g., a malformed file |
||||
* name). |
||||
*/ |
||||
GRPCErrorCodeInvalidArgument = 3, |
||||
|
||||
/**
|
||||
* Deadline expired before operation could complete. For operations that |
||||
* change the state of the server, this error may be returned even if the |
||||
* operation has completed successfully. For example, a successful response |
||||
* from the server could have been delayed long enough for the deadline to |
||||
* expire. |
||||
*/ |
||||
GRPCErrorCodeDeadlineExceeded = 4, |
||||
|
||||
/** Some requested entity (e.g., file or directory) was not found. */ |
||||
GRPCErrorCodeNotFound = 5, |
||||
|
||||
/** Some entity that we attempted to create (e.g., file or directory) already
|
||||
exists. */ |
||||
GRPCErrorCodeAlreadyExists = 6, |
||||
|
||||
/**
|
||||
* The caller does not have permission to execute the specified operation. |
||||
* PERMISSION_DENIED isn't used for rejections caused by exhausting some |
||||
* resource (RESOURCE_EXHAUSTED is used instead for those errors). |
||||
* PERMISSION_DENIED doesn't indicate a failure to identify the caller |
||||
* (UNAUTHENTICATED is used instead for those errors). |
||||
*/ |
||||
GRPCErrorCodePermissionDenied = 7, |
||||
|
||||
/**
|
||||
* The request does not have valid authentication credentials for the |
||||
* operation (e.g. the caller's identity can't be verified). |
||||
*/ |
||||
GRPCErrorCodeUnauthenticated = 16, |
||||
|
||||
/** Some resource has been exhausted, perhaps a per-user quota. */ |
||||
GRPCErrorCodeResourceExhausted = 8, |
||||
|
||||
/**
|
||||
* The RPC was rejected because the server is not in a state required for the |
||||
* procedure's execution. For example, a directory to be deleted may be |
||||
* non-empty, etc. The client should not retry until the server state has been |
||||
* explicitly fixed (e.g. by performing another RPC). The details depend on |
||||
* the service being called, and should be found in the NSError's userInfo. |
||||
*/ |
||||
GRPCErrorCodeFailedPrecondition = 9, |
||||
|
||||
/**
|
||||
* The RPC was aborted, typically due to a concurrency issue like sequencer |
||||
* check failures, transaction aborts, etc. The client should retry at a |
||||
* higher-level (e.g., restarting a read- modify-write sequence). |
||||
*/ |
||||
GRPCErrorCodeAborted = 10, |
||||
|
||||
/**
|
||||
* The RPC was attempted past the valid range. E.g., enumerating past the end |
||||
* of a list. Unlike INVALID_ARGUMENT, this error indicates a problem that may |
||||
* be fixed if the system state changes. For example, an RPC to get elements |
||||
* of a list will generate INVALID_ARGUMENT if asked to return the element at |
||||
* a negative index, but it will generate OUT_OF_RANGE if asked to return the |
||||
* element at an index past the current size of the list. |
||||
*/ |
||||
GRPCErrorCodeOutOfRange = 11, |
||||
|
||||
/** The procedure is not implemented or not supported/enabled in this server.
|
||||
*/ |
||||
GRPCErrorCodeUnimplemented = 12, |
||||
|
||||
/**
|
||||
* Internal error. Means some invariant expected by the server application or |
||||
* the gRPC library has been broken. |
||||
*/ |
||||
GRPCErrorCodeInternal = 13, |
||||
|
||||
/**
|
||||
* The server is currently unavailable. This is most likely a transient |
||||
* condition and may be corrected by retrying with a backoff. Note that it is |
||||
* not always safe to retry non-idempotent operations. |
||||
*/ |
||||
GRPCErrorCodeUnavailable = 14, |
||||
|
||||
/** Unrecoverable data loss or corruption. */ |
||||
GRPCErrorCodeDataLoss = 15, |
||||
}; |
||||
|
||||
/**
|
||||
* Safety remark of a gRPC method as defined in RFC 2616 Section 9.1 |
||||
*/ |
||||
typedef NS_ENUM(NSUInteger, GRPCCallSafety) { |
||||
/**
|
||||
* Signal that there is no guarantees on how the call affects the server |
||||
* state. |
||||
*/ |
||||
GRPCCallSafetyDefault = 0, |
||||
/** Signal that the call is idempotent. gRPC is free to use PUT verb. */ |
||||
GRPCCallSafetyIdempotentRequest = 1, |
||||
/**
|
||||
* Signal that the call is cacheable and will not affect server state. gRPC is |
||||
* free to use GET verb. |
||||
*/ |
||||
GRPCCallSafetyCacheableRequest = 2, |
||||
}; |
||||
|
||||
// Compression algorithm to be used by a gRPC call
|
||||
typedef NS_ENUM(NSUInteger, GRPCCompressionAlgorithm) { |
||||
GRPCCompressNone = 0, |
||||
GRPCCompressDeflate, |
||||
GRPCCompressGzip, |
||||
GRPCStreamCompressGzip, |
||||
}; |
||||
|
||||
// GRPCCompressAlgorithm is deprecated; use GRPCCompressionAlgorithm
|
||||
typedef GRPCCompressionAlgorithm GRPCCompressAlgorithm; |
||||
|
||||
/** The transport to be used by a gRPC call */ |
||||
typedef NS_ENUM(NSUInteger, GRPCTransportType) { |
||||
GRPCTransportTypeDefault = 0, |
||||
/** gRPC internal HTTP/2 stack with BoringSSL */ |
||||
GRPCTransportTypeChttp2BoringSSL = 0, |
||||
/** Cronet stack */ |
||||
GRPCTransportTypeCronet, |
||||
/** Insecure channel. FOR TEST ONLY! */ |
||||
GRPCTransportTypeInsecure, |
||||
}; |
||||
|
||||
/** Domain of NSError objects produced by gRPC. */ |
||||
extern NSString* _Nonnull const kGRPCErrorDomain; |
||||
|
||||
/**
|
||||
* Keys used in |NSError|'s |userInfo| dictionary to store the response headers |
||||
* and trailers sent by the server. |
||||
*/ |
||||
extern NSString* _Nonnull const kGRPCHeadersKey; |
||||
extern NSString* _Nonnull const kGRPCTrailersKey; |
||||
|
||||
/** The id of a transport implementation. */ |
||||
typedef char* _Nonnull GRPCTransportId; |
||||
|
||||
/**
|
||||
* Implement this protocol to provide a token to gRPC when a call is initiated. |
||||
*/ |
||||
@protocol GRPCAuthorizationProtocol |
||||
|
||||
/**
|
||||
* This method is called when gRPC is about to start the call. When OAuth token is acquired, |
||||
* \a handler is expected to be called with \a token being the new token to be used for this call. |
||||
*/ |
||||
- (void)getTokenWithHandler:(void (^_Nonnull)(NSString* _Nullable token))handler; |
||||
|
||||
@end |
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue