ceph/src/jaegertracing/thrift/lib/py/src/ext/types.cpp

   1 /*
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements. See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership. The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License. You may obtain a copy of the License at
   9  *
  10  *   http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing,
  13  * software distributed under the License is distributed on an
  14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15  * KIND, either express or implied. See the License for the
  16  * specific language governing permissions and limitations
  17  * under the License.
  18  */
  19
  20 #include "ext/types.h"
  21 #include "ext/protocol.h"
  22
  23 namespace apache {
  24 namespace thrift {
  25 namespace py {
  26
  27 PyObject* ThriftModule = NULL;
  28
  29 #if PY_MAJOR_VERSION < 3
  30 char refill_signature[] = {'s', '#', 'i'};
  31 #else
  32 const char* refill_signature = "y#i";
  33 #endif
  34
  35 bool parse_struct_item_spec(StructItemSpec* dest, PyObject* spec_tuple) {
  36   // i'd like to use ParseArgs here, but it seems to be a bottleneck.
  37   if (PyTuple_Size(spec_tuple) != 5) {
  38     PyErr_Format(PyExc_TypeError, "expecting 5 arguments for spec tuple but got %d",
  39                  static_cast<int>(PyTuple_Size(spec_tuple)));
  40     return false;
  41   }
  42
  43   dest->tag = static_cast<TType>(PyInt_AsLong(PyTuple_GET_ITEM(spec_tuple, 0)));
  44   if (INT_CONV_ERROR_OCCURRED(dest->tag)) {
  45     return false;
  46   }
  47
  48   dest->type = static_cast<TType>(PyInt_AsLong(PyTuple_GET_ITEM(spec_tuple, 1)));
  49   if (INT_CONV_ERROR_OCCURRED(dest->type)) {
  50     return false;
  51   }
  52
  53   dest->attrname = PyTuple_GET_ITEM(spec_tuple, 2);
  54   dest->typeargs = PyTuple_GET_ITEM(spec_tuple, 3);
  55   dest->defval = PyTuple_GET_ITEM(spec_tuple, 4);
  56   return true;
  57 }
  58
  59 bool parse_set_list_args(SetListTypeArgs* dest, PyObject* typeargs) {
  60   if (PyTuple_Size(typeargs) != 3) {
  61     PyErr_SetString(PyExc_TypeError, "expecting tuple of size 3 for list/set type args");
  62     return false;
  63   }
  64
  65   dest->element_type = static_cast<TType>(PyInt_AsLong(PyTuple_GET_ITEM(typeargs, 0)));
  66   if (INT_CONV_ERROR_OCCURRED(dest->element_type)) {
  67     return false;
  68   }
  69
  70   dest->typeargs = PyTuple_GET_ITEM(typeargs, 1);
  71
  72   dest->immutable = Py_True == PyTuple_GET_ITEM(typeargs, 2);
  73
  74   return true;
  75 }
  76
  77 bool parse_map_args(MapTypeArgs* dest, PyObject* typeargs) {
  78   if (PyTuple_Size(typeargs) != 5) {
  79     PyErr_SetString(PyExc_TypeError, "expecting 5 arguments for typeargs to map");
  80     return false;
  81   }
  82
  83   dest->ktag = static_cast<TType>(PyInt_AsLong(PyTuple_GET_ITEM(typeargs, 0)));
  84   if (INT_CONV_ERROR_OCCURRED(dest->ktag)) {
  85     return false;
  86   }
  87
  88   dest->vtag = static_cast<TType>(PyInt_AsLong(PyTuple_GET_ITEM(typeargs, 2)));
  89   if (INT_CONV_ERROR_OCCURRED(dest->vtag)) {
  90     return false;
  91   }
  92
  93   dest->ktypeargs = PyTuple_GET_ITEM(typeargs, 1);
  94   dest->vtypeargs = PyTuple_GET_ITEM(typeargs, 3);
  95   dest->immutable = Py_True == PyTuple_GET_ITEM(typeargs, 4);
  96
  97   return true;
  98 }
  99
 100 bool parse_struct_args(StructTypeArgs* dest, PyObject* typeargs) {
 101   if (PyList_Size(typeargs) != 2) {
 102     PyErr_SetString(PyExc_TypeError, "expecting list of size 2 for struct args");
 103     return false;
 104   }
 105
 106   dest->klass = PyList_GET_ITEM(typeargs, 0);
 107   dest->spec = PyList_GET_ITEM(typeargs, 1);
 108
 109   return true;
 110 }
 111 }
 112 }
 113 }