aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorgvit <gvit@ydb.tech>2023-01-22 22:56:11 +0300
committergvit <gvit@ydb.tech>2023-01-22 22:56:11 +0300
commit774dc92d50dd742e6bdfb3776096800ef6112fa3 (patch)
treeb59e2bbf0ed3078bf769e1f5fa37f50a80b725bd
parentc904a03f6f87cdb8db84e5e4729ac462278a4893 (diff)
downloadydb-774dc92d50dd742e6bdfb3776096800ef6112fa3.tar.gz
add retries to query execution
-rw-r--r--ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp17
1 files changed, 16 insertions, 1 deletions
diff --git a/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp b/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
index c671389d95..f86823c13b 100644
--- a/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
+++ b/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
@@ -11,7 +11,7 @@ using namespace NYdb;
namespace {
-std::vector<TString> ScanQuerySelect(
+std::vector<TString> ScanQuerySelectSimple(
NYdb::NTable::TTableClient client, const TString& tablePath,
const std::vector<std::pair<TString, NYdb::EPrimitiveType>>& ydbSchema = TTestOlap::PublicSchema())
{
@@ -90,6 +90,21 @@ std::vector<TString> ScanQuerySelect(
return out;
}
+std::vector<TString> ScanQuerySelect(
+ NYdb::NTable::TTableClient client, const TString& tablePath,
+ const std::vector<std::pair<TString, NYdb::EPrimitiveType>>& ydbSchema = TTestOlap::PublicSchema())
+{
+ for(ui32 iter = 0; iter < 3; iter++) {
+ try {
+ return ScanQuerySelectSimple(client, tablePath, ydbSchema);
+ } catch(...) {
+ /// o_O
+ }
+ }
+
+ return ScanQuerySelectSimple(client, tablePath, ydbSchema);
+}
+
}
Y_UNIT_TEST_SUITE(YdbTableBulkUpsertOlap) {