1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
# Copyright 2017 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 collections
class InitialMetadataFlags:
used_mask = GRPC_INITIAL_METADATA_USED_MASK
wait_for_ready = GRPC_INITIAL_METADATA_WAIT_FOR_READY
wait_for_ready_explicitly_set = GRPC_INITIAL_METADATA_WAIT_FOR_READY_EXPLICITLY_SET
_Metadatum = collections.namedtuple('_Metadatum', ('key', 'value',))
cdef void _store_c_metadata(
metadata, grpc_metadata **c_metadata, size_t *c_count) except *:
if metadata is None:
c_count[0] = 0
c_metadata[0] = NULL
else:
metadatum_count = len(metadata)
if metadatum_count == 0:
c_count[0] = 0
c_metadata[0] = NULL
else:
c_count[0] = metadatum_count
c_metadata[0] = <grpc_metadata *>gpr_malloc(
metadatum_count * sizeof(grpc_metadata))
for index, (key, value) in enumerate(metadata):
encoded_key = _encode(key)
encoded_value = value if encoded_key[-4:] == b'-bin' else _encode(value)
if not isinstance(encoded_value, bytes):
raise TypeError('Binary metadata key="%s" expected bytes, got %s' % (
key,
type(encoded_value)
))
c_metadata[0][index].key = _slice_from_bytes(encoded_key)
c_metadata[0][index].value = _slice_from_bytes(encoded_value)
cdef void _release_c_metadata(grpc_metadata *c_metadata, int count) except *:
if 0 < count:
for index in range(count):
grpc_slice_unref(c_metadata[index].key)
grpc_slice_unref(c_metadata[index].value)
gpr_free(c_metadata)
cdef tuple _metadatum(grpc_slice key_slice, grpc_slice value_slice):
cdef bytes key = _slice_bytes(key_slice)
cdef bytes value = _slice_bytes(value_slice)
return <tuple>_Metadatum(
_decode(key), value if key[-4:] == b'-bin' else _decode(value))
cdef tuple _metadata(grpc_metadata_array *c_metadata_array):
return tuple(
_metadatum(
c_metadata_array.metadata[index].key,
c_metadata_array.metadata[index].value)
for index in range(c_metadata_array.count))
|
