mirror of
https://github.com/JakeHillion/object-introspection.git
synced 2024-11-13 22:06:55 +00:00
40af807d8b
Support the capture-thrift-isset feature with TreeBuilder-v2. Fairly minor changes here except the type of the Enum in a template parameter now matters. We follow the previous behaviour of capturing a value for each field in a struct that has an `isset_bitset`. This value is a VarInt captured before the C++ contents of the member. It has 3 values: 0 (not set), 1 (set), and 2 (unavailable). These are handled by the processor and represented in the output as `false`, `true`, and `std::nullopt_t` respectively. Changes: - Add a simple Thrift isset processor before any fields that have Thrift isset. - Store the fully qualified names of enum types in DrgnParser - it already worked out this information anyway for naming the values and this is consistent with classes. - Forward all enum template parameters under their input name under the assumption that they will all be policy type things like `IssetBitsetOption`. This could turn out to be wrong. Test plan: - CI (doesn't test thrift changes but covers other regressions) - Updated Thrift enum tests for new format. - `FILTER='OilIntegration.*' make test` - Thrift tests failed before, succeed after.
314 lines
11 KiB
TOML
314 lines
11 KiB
TOML
thrift_definitions = '''
|
|
include "thrift/annotation/cpp.thrift"
|
|
include "thrift/annotation/thrift.thrift"
|
|
|
|
struct MyThriftStructUnpacked {
|
|
1: optional i32 a;
|
|
2: optional i32 b;
|
|
3: optional i32 c;
|
|
}
|
|
|
|
@cpp.PackIsset
|
|
struct MyThriftStructPacked {
|
|
1: optional i32 a;
|
|
2: optional i32 b;
|
|
3: optional i32 c;
|
|
4: optional i32 d;
|
|
5: optional i32 e;
|
|
6: optional i32 f;
|
|
7: optional i32 g;
|
|
8: optional i32 h;
|
|
9: optional i32 i;
|
|
10: optional i32 j;
|
|
}
|
|
|
|
@cpp.PackIsset{atomic = false}
|
|
struct MyThriftStructPackedNonAtomic {
|
|
1: optional i32 a;
|
|
2: optional i32 b;
|
|
3: optional i32 c;
|
|
4: optional i32 d;
|
|
}
|
|
|
|
struct MyThriftStructOutOfOrder {
|
|
3: i32 a;
|
|
1: i32 b;
|
|
2: i32 c;
|
|
}
|
|
|
|
struct MyThriftStructRequired {
|
|
1: required i32 a;
|
|
2: optional i32 b;
|
|
3: optional i32 c;
|
|
4: required i32 d;
|
|
5: required i32 e;
|
|
6: optional i32 f;
|
|
}
|
|
|
|
struct MyThriftStructBoxed {
|
|
@cpp.Ref{type = cpp.RefType.Unique}
|
|
1: optional i32 a;
|
|
@thrift.Box
|
|
2: optional i32 b;
|
|
3: optional i32 c;
|
|
4: optional i32 d;
|
|
5: optional i32 e;
|
|
}
|
|
'''
|
|
raw_definitions = '''
|
|
namespace cpp2 {
|
|
MyThriftStructBoxed::MyThriftStructBoxed() :
|
|
__fbthrift_field_b(),
|
|
__fbthrift_field_c(),
|
|
__fbthrift_field_d(),
|
|
__fbthrift_field_e() {
|
|
}
|
|
MyThriftStructBoxed::~MyThriftStructBoxed() {}
|
|
MyThriftStructBoxed::MyThriftStructBoxed(::cpp2::MyThriftStructBoxed&& other) noexcept :
|
|
__fbthrift_field_a(std::move(other.__fbthrift_field_a)),
|
|
__fbthrift_field_b(std::move(other.__fbthrift_field_b)),
|
|
__fbthrift_field_c(std::move(other.__fbthrift_field_c)),
|
|
__fbthrift_field_d(std::move(other.__fbthrift_field_d)),
|
|
__fbthrift_field_e(std::move(other.__fbthrift_field_e)),
|
|
__isset(other.__isset) {
|
|
}
|
|
} // namespace cpp2
|
|
'''
|
|
|
|
[cases]
|
|
[cases.unpacked]
|
|
param_types = ["const cpp2::MyThriftStructUnpacked&"]
|
|
setup = '''
|
|
cpp2::MyThriftStructUnpacked ret;
|
|
ret.a_ref() = 1;
|
|
ret.c_ref() = 1;
|
|
return ret;
|
|
'''
|
|
features = ["capture-thrift-isset"]
|
|
expect_json = '''[{
|
|
"staticSize":16,
|
|
"dynamicSize":0,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "isset":true},
|
|
{"name":"__isset", "staticSize":3}
|
|
]}]'''
|
|
expect_json_v2 = '''[{
|
|
"staticSize":16,
|
|
"exclusiveSize":1,
|
|
"size":16,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "is_set":true},
|
|
{"name":"__isset", "staticSize":3, "NOT":"is_set"}
|
|
]
|
|
}]'''
|
|
|
|
[cases.packed]
|
|
param_types = ["const cpp2::MyThriftStructPacked&"]
|
|
setup = '''
|
|
cpp2::MyThriftStructPacked ret;
|
|
ret.a_ref() = 1;
|
|
ret.c_ref() = 1;
|
|
ret.d_ref() = 1;
|
|
ret.g_ref() = 1;
|
|
ret.h_ref() = 1;
|
|
ret.j_ref() = 1;
|
|
return ret;
|
|
'''
|
|
features = ["capture-thrift-isset"]
|
|
expect_json = '''[{
|
|
"staticSize":44,
|
|
"dynamicSize":0,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_f", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_g", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_h", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_i", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_j", "staticSize":4, "isset":true},
|
|
{"name":"__isset", "staticSize":2}
|
|
]}]'''
|
|
expect_json_v2 = '''[{
|
|
"staticSize":44,
|
|
"exclusiveSize":2,
|
|
"size":44,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_f", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_g", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_h", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_i", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_j", "staticSize":4, "is_set":true},
|
|
{"name":"__isset", "staticSize":2, "NOT":"is_set"}
|
|
]}]'''
|
|
|
|
[cases.packed_non_atomic]
|
|
param_types = ["const cpp2::MyThriftStructPackedNonAtomic&"]
|
|
setup = '''
|
|
cpp2::MyThriftStructPackedNonAtomic ret;
|
|
ret.a_ref() = 1;
|
|
ret.c_ref() = 1;
|
|
return ret;
|
|
'''
|
|
features = ["capture-thrift-isset"]
|
|
expect_json = '''[{
|
|
"staticSize":20,
|
|
"dynamicSize":0,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "isset":false},
|
|
{"name":"__isset", "staticSize":1}
|
|
]}]'''
|
|
expect_json_v2 = '''[{
|
|
"staticSize":20,
|
|
"exclusiveSize":3,
|
|
"size":20,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "is_set":false},
|
|
{"name":"__isset", "staticSize":1, "NOT":"is_set"}
|
|
]}]'''
|
|
|
|
[cases.out_of_order]
|
|
param_types = ["const cpp2::MyThriftStructOutOfOrder&"]
|
|
setup = '''
|
|
cpp2::MyThriftStructOutOfOrder ret;
|
|
ret.b_ref() = 1;
|
|
return ret;
|
|
'''
|
|
features = ["capture-thrift-isset"]
|
|
expect_json = '''[{
|
|
"staticSize":16,
|
|
"dynamicSize":0,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "isset":false},
|
|
{"name":"__isset", "staticSize":3}
|
|
]}]'''
|
|
expect_json_v2 = '''[{
|
|
"staticSize":16,
|
|
"exclusiveSize":1,
|
|
"size":16,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "is_set":false},
|
|
{"name":"__isset", "staticSize":3, "NOT":"is_set"}
|
|
]}]'''
|
|
|
|
[cases.required]
|
|
param_types = ["const cpp2::MyThriftStructRequired&"]
|
|
setup = '''
|
|
cpp2::MyThriftStructRequired ret;
|
|
ret.b_ref() = 1;
|
|
ret.f_ref() = 1;
|
|
return ret;
|
|
'''
|
|
features = ["capture-thrift-isset"]
|
|
expect_json = '''[{
|
|
"staticSize":28,
|
|
"dynamicSize":0,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_f", "staticSize":4, "isset":true},
|
|
{"name":"__isset", "staticSize":3}
|
|
]}]'''
|
|
expect_json_v2 = '''[{
|
|
"staticSize":28,
|
|
"exclusiveSize":1,
|
|
"size":28,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":4, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_b", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_f", "staticSize":4, "is_set":true},
|
|
{"name":"__isset", "staticSize":3, "NOT":"is_set"}
|
|
]}]'''
|
|
|
|
[cases.box]
|
|
param_types = ["const cpp2::MyThriftStructBoxed&"]
|
|
setup = '''
|
|
cpp2::MyThriftStructBoxed ret;
|
|
ret.d_ref() = 1;
|
|
ret.e_ref() = 1;
|
|
return ret;
|
|
'''
|
|
features = ["capture-thrift-isset"]
|
|
expect_json = '''[{
|
|
"staticSize":32,
|
|
"dynamicSize":0,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":8, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_b", "staticSize":8, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "isset":false},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "isset":true},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "isset":true},
|
|
{"name":"__isset", "staticSize":3}
|
|
]}]'''
|
|
expect_json_v2 = '''[{
|
|
"staticSize":32,
|
|
"exclusiveSize":1,
|
|
"size":32,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":8, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_b", "staticSize":8, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "is_set":false},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "is_set":true},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "is_set":true},
|
|
{"name":"__isset", "staticSize":3, "NOT":"is_set"}
|
|
]}]'''
|
|
|
|
[cases.no_capture]
|
|
param_types = ["const cpp2::MyThriftStructBoxed&"]
|
|
setup = '''
|
|
cpp2::MyThriftStructBoxed ret;
|
|
ret.d_ref() = 1;
|
|
ret.e_ref() = 1;
|
|
return ret;
|
|
'''
|
|
expect_json = '''[{
|
|
"staticSize":32,
|
|
"dynamicSize":0,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":8, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_b", "staticSize":8, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "NOT":"isset"},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "NOT":"isset"},
|
|
{"name":"__isset", "staticSize":3}
|
|
]}]'''
|
|
expect_json_v2 = '''[{
|
|
"staticSize":32,
|
|
"exclusiveSize":1,
|
|
"size":32,
|
|
"members":[
|
|
{"name":"__fbthrift_field_a", "staticSize":8, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_b", "staticSize":8, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_c", "staticSize":4, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_d", "staticSize":4, "NOT":"is_set"},
|
|
{"name":"__fbthrift_field_e", "staticSize":4, "NOT":"is_set"},
|
|
{"name":"__isset", "staticSize":3, "NOT":"is_set"}
|
|
]}]'''
|