Skip to content

Commit 1663995

Browse files
committed
Use table columns only in case of PG's INSERT with no columns specified
1 parent 99dda8a commit 1663995

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

ydb/library/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -445,8 +445,9 @@ class TYtDataSinkTypeAnnotationTransformer : public TVisitorTransformerBase {
445445
}
446446
}
447447

448-
const TMaybe<TColumnOrder> contentColumnOrder = outTableInfo.RowSpec->GetColumnOrder();
448+
TMaybe<TColumnOrder> contentColumnOrder;
449449
if (content) {
450+
contentColumnOrder = State_->Types->LookupColumnOrder(*content);
450451
if (content->IsCallable("AssumeColumnOrder")) {
451452
YQL_ENSURE(contentColumnOrder);
452453
YQL_CLOG(INFO, ProviderYt) << "Dropping top level " << content->Content() << " from WriteTable input";
@@ -458,7 +459,9 @@ class TYtDataSinkTypeAnnotationTransformer : public TVisitorTransformerBase {
458459
auto pgSelect = TCoPgSelect(content);
459460
if (NCommon::NeedToRenamePgSelectColumns(pgSelect)) {
460461
TExprNode::TPtr output;
461-
bool result = NCommon::RenamePgSelectColumns(pgSelect, output, contentColumnOrder, ctx, *State_->Types);
462+
463+
Y_ENSURE(outTableInfo.RowSpec);
464+
bool result = NCommon::RenamePgSelectColumns(pgSelect, output, outTableInfo.RowSpec->GetColumnOrder(), ctx, *State_->Types);
462465
if (!result) {
463466
return TStatus::Error;
464467
}

0 commit comments

Comments
 (0)