Function write_primitive_item

python/pyfory/cpp/pyfory.cc:783–931

}

static int write_primitive_item(Buffer *buffer, PyObject *value,
                                uint8_t type_id) {
  switch (static_cast<TypeId>(type_id)) {
  case TypeId::STRING:
    return write_python_string(buffer, value);
  case TypeId::VARINT64:
  case TypeId::INT64: {
    int64_t v = 0;
    if (FORY_PREDICT_FALSE(!py_long_to_int64(value, &v))) {
      return -1;
    }
    if (static_cast<TypeId>(type_id) == TypeId::INT64) {
      buffer->write_int64(v);
    } else {
      buffer->write_var_int64(v);
    }
    return 0;
  }
  case TypeId::VARINT32:
  case TypeId::INT32: {
    int32_t v = 0;
    if (FORY_PREDICT_FALSE(
            !py_long_to_integral_range<int32_t>(value, "int32", &v))) {
      return -1;
    }
    if (static_cast<TypeId>(type_id) == TypeId::INT32) {
      buffer->write_int32(v);
    } else {
      buffer->write_var_int32(v);
    }
    return 0;
  }
  case TypeId::VAR_UINT64:
  case TypeId::UINT64:
  case TypeId::TAGGED_UINT64: {
    uint64_t v = 0;
    if (FORY_PREDICT_FALSE(
            !py_long_to_unsigned_range<uint64_t>(value, "uint64", &v))) {
      return -1;
    }
    if (static_cast<TypeId>(type_id) == TypeId::VAR_UINT64) {
      buffer->write_var_uint64(v);
    } else if (static_cast<TypeId>(type_id) == TypeId::TAGGED_UINT64) {
      buffer->write_tagged_uint64(v);
    } else {
      buffer->write_int64(static_cast<int64_t>(v));
    }
    return 0;
  }
  case TypeId::VAR_UINT32:
  case TypeId::UINT32: {
    uint32_t v = 0;
    if (FORY_PREDICT_FALSE(
            !py_long_to_unsigned_range<uint32_t>(value, "uint32", &v))) {
      return -1;
    }
    if (static_cast<TypeId>(type_id) == TypeId::VAR_UINT32) {
      buffer->write_var_uint32(v);