Skip to content

Commit bc29041

Browse files
nikvas0nikvas0
authored
Add bytes to cost test (#29777)
1 parent 585a708 commit bc29041

File tree

1 file changed

+88
-0
lines changed

1 file changed

+88
-0
lines changed

‎ydb/core/kqp/ut/cost/kqp_cost_ut.cpp‎

Lines changed: 88 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1101,6 +1101,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
11011101
size_t Writes = 0;
11021102
size_t Reads = 0;
11031103
size_t Deletes = 0;
1104+
1105+
size_t WriteBytes = 0;
1106+
size_t ReadBytes = 0;
1107+
size_t DeleteBytes = 0;
11041108
};
11051109

11061110
TTotalStats FromProto(const Ydb::TableStats::QueryStats& proto) {
@@ -1110,6 +1114,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
11101114
stats.Writes += proto.query_phases(phase).table_access(access).updates().rows();
11111115
stats.Reads += proto.query_phases(phase).table_access(access).reads().rows();
11121116
stats.Deletes += proto.query_phases(phase).table_access(access).deletes().rows();
1117+
1118+
stats.WriteBytes += proto.query_phases(phase).table_access(access).updates().bytes();
1119+
stats.ReadBytes += proto.query_phases(phase).table_access(access).reads().bytes();
1120+
stats.DeleteBytes += proto.query_phases(phase).table_access(access).deletes().bytes();
11131121
}
11141122
}
11151123
return stats;
@@ -1119,6 +1127,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
11191127
UNIT_ASSERT_VALUES_EQUAL(lhs.Writes, rhs.Writes);
11201128
UNIT_ASSERT_VALUES_EQUAL(lhs.Reads, rhs.Reads);
11211129
UNIT_ASSERT_VALUES_EQUAL(lhs.Deletes, rhs.Deletes);
1130+
1131+
UNIT_ASSERT_VALUES_EQUAL(lhs.WriteBytes, rhs.WriteBytes);
1132+
UNIT_ASSERT_VALUES_EQUAL(lhs.ReadBytes, rhs.ReadBytes);
1133+
UNIT_ASSERT_VALUES_EQUAL(lhs.DeleteBytes, rhs.DeleteBytes);
11221134
}
11231135

11241136

@@ -1164,6 +1176,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
11641176
.Writes = 1,
11651177
.Reads = 0,
11661178
.Deletes = 0,
1179+
1180+
.WriteBytes = 20,
1181+
.ReadBytes = 0,
1182+
.DeleteBytes = 0,
11671183
});
11681184
}
11691185

@@ -1197,6 +1213,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
11971213
.Writes = 1,
11981214
.Reads = 0,
11991215
.Deletes = 0,
1216+
1217+
.WriteBytes = 20,
1218+
.ReadBytes = 0,
1219+
.DeleteBytes = 0,
12001220
});
12011221
}
12021222

@@ -1225,6 +1245,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
12251245
.Writes = 0,
12261246
.Reads = 1,
12271247
.Deletes = 0,
1248+
1249+
.WriteBytes = 0,
1250+
.ReadBytes = 8,
1251+
.DeleteBytes = 0,
12281252
});
12291253
}
12301254

@@ -1259,6 +1283,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
12591283
.Writes = 1,
12601284
.Reads = 0,
12611285
.Deletes = 0,
1286+
1287+
.WriteBytes = 20,
1288+
.ReadBytes = 0,
1289+
.DeleteBytes = 0,
12621290
});
12631291
}
12641292

@@ -1291,6 +1319,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
12911319
.Writes = 0,
12921320
.Reads = 0,
12931321
.Deletes = 0,
1322+
1323+
.WriteBytes = 0,
1324+
.ReadBytes = 0,
1325+
.DeleteBytes = 0,
12941326
});
12951327
}
12961328

@@ -1320,6 +1352,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
13201352
.Writes = 1,
13211353
.Reads = 1,
13221354
.Deletes = 0,
1355+
1356+
.WriteBytes = 20,
1357+
.ReadBytes = 8,
1358+
.DeleteBytes = 0,
13231359
});
13241360
}
13251361

@@ -1354,6 +1390,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
13541390
.Writes = 0,
13551391
.Reads = 0,
13561392
.Deletes = 1,
1393+
1394+
.WriteBytes = 0,
1395+
.ReadBytes = 0,
1396+
.DeleteBytes = 0,
13571397
});
13581398
}
13591399

@@ -1388,6 +1428,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
13881428
.Writes = 0,
13891429
.Reads = 0,
13901430
.Deletes = 1,
1431+
1432+
.WriteBytes = 0,
1433+
.ReadBytes = 0,
1434+
.DeleteBytes = 0,
13911435
});
13921436
}
13931437
}
@@ -1443,6 +1487,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
14431487
.Writes = 1,
14441488
.Reads = 1,
14451489
.Deletes = 0,
1490+
1491+
.WriteBytes = 20,
1492+
.ReadBytes = 8,
1493+
.DeleteBytes = 0,
14461494
});
14471495
}
14481496

@@ -1483,6 +1531,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
14831531
.Writes = 1,
14841532
.Reads = 1,
14851533
.Deletes = 0,
1534+
1535+
.WriteBytes = 20,
1536+
.ReadBytes = 8,
1537+
.DeleteBytes = 0,
14861538
});
14871539
}
14881540

@@ -1518,6 +1570,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
15181570
.Writes = 0,
15191571
.Reads = 1,
15201572
.Deletes = 0,
1573+
1574+
.WriteBytes = 0,
1575+
.ReadBytes = 8,
1576+
.DeleteBytes = 0,
15211577
});
15221578
}
15231579

@@ -1555,6 +1611,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
15551611
.Writes = 0,
15561612
.Reads = 1,
15571613
.Deletes = 0,
1614+
1615+
.WriteBytes = 0,
1616+
.ReadBytes = 8,
1617+
.DeleteBytes = 0,
15581618
});
15591619
}
15601620

@@ -1592,6 +1652,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
15921652
.Writes = 0,
15931653
.Reads = 1,
15941654
.Deletes = 0,
1655+
1656+
.WriteBytes = 0,
1657+
.ReadBytes = 8,
1658+
.DeleteBytes = 0,
15951659
});
15961660
}
15971661

@@ -1631,6 +1695,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
16311695
.Writes = isSink ? 1 : 0, // EvWrite writes before next read
16321696
.Reads = 1,
16331697
.Deletes = 0,
1698+
1699+
.WriteBytes = isSink ? 20 : 0,
1700+
.ReadBytes = 8,
1701+
.DeleteBytes = 0,
16341702
});
16351703
}
16361704

@@ -1670,6 +1738,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
16701738
.Writes = isSink ? 1 : 0, // EvWrite writes before next read
16711739
.Reads = 1,
16721740
.Deletes = 0,
1741+
1742+
.WriteBytes = isSink ? 20 : 0,
1743+
.ReadBytes = 8,
1744+
.DeleteBytes = 0,
16731745
});
16741746
}
16751747

@@ -1709,6 +1781,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
17091781
.Writes = isSink ? 2 : 0, // EvWrite writes before next read
17101782
.Reads = 1,
17111783
.Deletes = 0,
1784+
1785+
.WriteBytes = isSink ? 40 : 0,
1786+
.ReadBytes = 8,
1787+
.DeleteBytes = 0,
17121788
});
17131789
}
17141790

@@ -1749,6 +1825,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
17491825
.Writes = isSink ? 3 : 0, // EvWrite writes before next read
17501826
.Reads = isSink ? 1 : 0,
17511827
.Deletes = 0,
1828+
1829+
.WriteBytes = isSink ? 60 : 0,
1830+
.ReadBytes = isSink ? 8 : 0,
1831+
.DeleteBytes = 0,
17521832
});
17531833
}
17541834
}
@@ -1788,6 +1868,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
17881868
.Writes = 4,
17891869
.Reads = 4,
17901870
.Deletes = 0,
1871+
1872+
.WriteBytes = isOlap ? 1472 : 80,
1873+
.ReadBytes = isOlap ? 144 : 80,
1874+
.DeleteBytes = 0,
17911875
});
17921876
}
17931877
}
@@ -1854,6 +1938,10 @@ Y_UNIT_TEST_SUITE(KqpCost) {
18541938
.Writes = 4,
18551939
.Reads = 4,
18561940
.Deletes = 0,
1941+
1942+
.WriteBytes = isOlap ? 1472 : 80,
1943+
.ReadBytes = isOlap ? 144 : 80,
1944+
.DeleteBytes = 0,
18571945
});
18581946
}
18591947

0 commit comments

Comments
 (0)