Fix bug in merger

ydb-platform · Dec 27, 2024 · 09dca29 · 09dca29
1 parent 0fbaf0d
commit 09dca29
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...tory-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/merge/ByEntityYqlQueriesMerger.java b/...tory-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/merge/ByEntityYqlQueriesMerger.java
@@ -164,7 +164,7 @@ private MergingState doTransition(MergingState state, Statement.QueryType nextQu
     private static YdbRepository.Query convertInsertToUpsert(YdbRepository.Query<?> query) {
         var type = getEntityClass(query);
         var schema = EntitySchema.of(type);
-        var tableDescriptor = TableDescriptor.from(schema);
+        var tableDescriptor = convertQueryToYqlStatement(query).getTableDescriptor();
         var statement = new UpsertYqlStatement<>(tableDescriptor, schema);
         return new YdbRepository.Query<>(statement, query.getValues().get(0));
     }
@@ -173,7 +173,7 @@ private static YdbRepository.Query convertInsertToUpsert(YdbRepository.Query<?>
     private static YdbRepository.Query convertInsertToDelete(YdbRepository.Query<?> query) {
         var type = getEntityClass(query);
         var schema = EntitySchema.of(type);
-        var tableDescriptor = TableDescriptor.from(schema);
+        var tableDescriptor = convertQueryToYqlStatement(query).getTableDescriptor();
         var statement = new DeleteByIdStatement<>(tableDescriptor, schema);
         return new YdbRepository.Query<>(statement, getEntityId(query));
     }