mirror of
https://github.com/protocolbuffers/protobuf
synced 2026-08-26 02:23:14 -04:00
976029283f
This change implements maps with keys and values of type string e.g. Map<ProtoStr, i32> and Map<ProtoStr, ProtoStr>. Implementing the Map type for ProtoStr has been different from scalar types because ProtoStr is an unsized type i.e. its size is not known at compile time. The existing Map implementation assumed sized types in many places. To make unsized types fit into the existing code architecture I have added an associated type 'Value' to the MapWith*KeyOps traits. The associated type needs to be sized and is the type returned by the Map::get(self, key) method e.g. for aProtoStr, the `type Value = &ProtoStr`. PiperOrigin-RevId: 588783751
117 lines
7.6 KiB
C++
117 lines
7.6 KiB
C++
#include "rust/cpp_kernel/cpp_api.h"
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
|
|
#include "google/protobuf/map.h"
|
|
#include "google/protobuf/repeated_field.h"
|
|
|
|
extern "C" {
|
|
|
|
#define expose_repeated_field_methods(ty, rust_ty) \
|
|
google::protobuf::RepeatedField<ty>* __pb_rust_RepeatedField_##rust_ty##_new() { \
|
|
return new google::protobuf::RepeatedField<ty>(); \
|
|
} \
|
|
void __pb_rust_RepeatedField_##rust_ty##_add(google::protobuf::RepeatedField<ty>* r, \
|
|
ty val) { \
|
|
r->Add(val); \
|
|
} \
|
|
size_t __pb_rust_RepeatedField_##rust_ty##_size( \
|
|
google::protobuf::RepeatedField<ty>* r) { \
|
|
return r->size(); \
|
|
} \
|
|
ty __pb_rust_RepeatedField_##rust_ty##_get(google::protobuf::RepeatedField<ty>* r, \
|
|
size_t index) { \
|
|
return r->Get(index); \
|
|
} \
|
|
void __pb_rust_RepeatedField_##rust_ty##_set(google::protobuf::RepeatedField<ty>* r, \
|
|
size_t index, ty val) { \
|
|
return r->Set(index, val); \
|
|
} \
|
|
void __pb_rust_RepeatedField_##rust_ty##_copy_from( \
|
|
google::protobuf::RepeatedField<ty> const& src, google::protobuf::RepeatedField<ty>& dst) { \
|
|
dst.CopyFrom(src); \
|
|
}
|
|
|
|
expose_repeated_field_methods(int32_t, i32);
|
|
expose_repeated_field_methods(uint32_t, u32);
|
|
expose_repeated_field_methods(float, f32);
|
|
expose_repeated_field_methods(double, f64);
|
|
expose_repeated_field_methods(bool, bool);
|
|
expose_repeated_field_methods(uint64_t, u64);
|
|
expose_repeated_field_methods(int64_t, i64);
|
|
|
|
#undef expose_repeated_field_methods
|
|
|
|
#define expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
|
|
value_ty, rust_value_ty, ffi_value_ty, \
|
|
to_cpp_value, to_ffi_value) \
|
|
google::protobuf::Map<key_ty, value_ty>* \
|
|
__pb_rust_Map_##rust_key_ty##_##rust_value_ty##_new() { \
|
|
return new google::protobuf::Map<key_ty, value_ty>(); \
|
|
} \
|
|
void __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_clear( \
|
|
google::protobuf::Map<key_ty, value_ty>* m) { \
|
|
m->clear(); \
|
|
} \
|
|
size_t __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_size( \
|
|
google::protobuf::Map<key_ty, value_ty>* m) { \
|
|
return m->size(); \
|
|
} \
|
|
void __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_insert( \
|
|
google::protobuf::Map<key_ty, value_ty>* m, ffi_key_ty key, ffi_value_ty value) { \
|
|
auto cpp_key = to_cpp_key; \
|
|
auto cpp_value = to_cpp_value; \
|
|
(*m)[cpp_key] = cpp_value; \
|
|
} \
|
|
bool __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_get( \
|
|
google::protobuf::Map<key_ty, value_ty>* m, ffi_key_ty key, ffi_value_ty* value) { \
|
|
auto cpp_key = to_cpp_key; \
|
|
auto it = m->find(cpp_key); \
|
|
if (it == m->end()) { \
|
|
return false; \
|
|
} \
|
|
auto& cpp_value = it->second; \
|
|
*value = to_ffi_value; \
|
|
return true; \
|
|
} \
|
|
bool __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_remove( \
|
|
google::protobuf::Map<key_ty, value_ty>* m, ffi_key_ty key, ffi_value_ty* value) { \
|
|
auto cpp_key = to_cpp_key; \
|
|
auto num_removed = m->erase(cpp_key); \
|
|
return num_removed > 0; \
|
|
}
|
|
|
|
#define expose_scalar_map_methods_for_key_type(key_ty, rust_key_ty, \
|
|
ffi_key_ty, to_cpp_key) \
|
|
expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
|
|
int32_t, i32, int32_t, value, cpp_value); \
|
|
expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
|
|
uint32_t, u32, uint32_t, value, cpp_value); \
|
|
expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
|
|
float, f32, float, value, cpp_value); \
|
|
expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
|
|
double, f64, double, value, cpp_value); \
|
|
expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, bool, \
|
|
bool, bool, value, cpp_value); \
|
|
expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
|
|
uint64_t, u64, uint64_t, value, cpp_value); \
|
|
expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
|
|
int64_t, i64, int64_t, value, cpp_value); \
|
|
expose_scalar_map_methods( \
|
|
key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, std::string, ProtoStr, \
|
|
google::protobuf::rust_internal::PtrAndLen, std::string(value.ptr, value.len), \
|
|
google::protobuf::rust_internal::PtrAndLen(cpp_value.data(), cpp_value.size()));
|
|
|
|
expose_scalar_map_methods_for_key_type(int32_t, i32, int32_t, key);
|
|
expose_scalar_map_methods_for_key_type(uint32_t, u32, uint32_t, key);
|
|
expose_scalar_map_methods_for_key_type(bool, bool, bool, key);
|
|
expose_scalar_map_methods_for_key_type(uint64_t, u64, uint64_t, key);
|
|
expose_scalar_map_methods_for_key_type(int64_t, i64, int64_t, key);
|
|
expose_scalar_map_methods_for_key_type(std::string, ProtoStr,
|
|
google::protobuf::rust_internal::PtrAndLen,
|
|
std::string(key.ptr, key.len));
|
|
|
|
#undef expose_scalar_map_methods
|
|
#undef expose_map_methods
|
|
}
|