[yt provider] Fix YtCopy type annotation check for column groups (#11602)

rvu1024 · web-flow · commit 85bb2107ed7f · 2024-11-14T18:49:07.000+03:00
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp
@@ -965,6 +965,10 @@ class TYtDataSinkTypeAnnotationTransformer : public TVisitorTransformerBase {
             if (auto setting = NYql::GetSetting(GetOutputOp(out.Cast()).Output().Item(FromString<ui32>(out.Cast().OutIndex().Value())).Settings().Ref(), EYtSettingType::ColumnGroups)) {
                 inputColGroupSpec = setting->Tail().Content();
             }
+        } else if (auto outTable = path.Table().Maybe<TYtOutTable>()) {
+            if (auto setting = NYql::GetSetting(outTable.Cast().Settings().Ref(), EYtSettingType::ColumnGroups)) {
+                inputColGroupSpec = setting->Tail().Content();
+            }
         }
 
         if (outGroup != inputColGroupSpec) {

Original file line number	Diff line number	Diff line change
`@@ -965,6 +965,10 @@ class TYtDataSinkTypeAnnotationTransformer : public TVisitorTransformerBase {`
`965`	`965`	`if (auto setting = NYql::GetSetting(GetOutputOp(out.Cast()).Output().Item(FromString<ui32>(out.Cast().OutIndex().Value())).Settings().Ref(), EYtSettingType::ColumnGroups)) {`
`966`	`966`	`inputColGroupSpec = setting->Tail().Content();`
`967`	`967`	`}`
	`968`	`+ } else if (auto outTable = path.Table().Maybe<TYtOutTable>()) {`
	`969`	`+ if (auto setting = NYql::GetSetting(outTable.Cast().Settings().Ref(), EYtSettingType::ColumnGroups)) {`
	`970`	`+ inputColGroupSpec = setting->Tail().Content();`
	`971`	`+ }`
`968`	`972`	`}`
`969`	`973`
`970`	`974`	`if (outGroup != inputColGroupSpec) {`