Skip to content

Commit 51041e3

Browse files
authored
[DQ] Fix YtDqProcessWrite with aux columns in YT emulation mode (#9400)
1 parent 4514278 commit 51041e3

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

ydb/library/yql/providers/yt/gateway/file/yql_yt_file.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -453,7 +453,7 @@ class TYtFileGateway : public IYtGateway {
453453
auto pos = options.Pos();
454454
try {
455455
TSession* session = GetSession(options);
456-
456+
457457
TSet<TString> uniqueTables;
458458
const auto fullPrefix = options.Prefix().Empty() ? TString() : (options.Prefix() + '/');
459459
const auto fullSuffix = options.Suffix().Empty() ? TString() : ('/' + options.Suffix());
@@ -776,7 +776,7 @@ class TYtFileGateway : public IYtGateway {
776776
writer.SetSpecs(spec);
777777

778778
TStringStream err;
779-
auto type = BuildType(*tableInfo.RowSpec->GetType(), typeBuilder, err);
779+
auto type = BuildType(*tableInfo.RowSpec->GetExtendedType(ctx), typeBuilder, err);
780780
TValuePacker packer(true, type);
781781
for (auto& c: content) {
782782
auto val = packer.Unpack(c, holderFactory);

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
#include <ydb/library/yql/providers/yt/provider/yql_yt_helpers.h>
1414
#include <ydb/library/yql/providers/common/provider/yql_provider.h>
1515
#include <ydb/library/yql/providers/common/transform/yql_exec.h>
16+
#include <ydb/library/yql/providers/common/schema/expr/yql_expr_schema.h>
1617
#include <ydb/library/yql/core/type_ann/type_ann_expr.h>
1718
#include <ydb/library/yql/core/yql_execution.h>
1819
#include <ydb/library/yql/core/yql_graph_transformer.h>
@@ -742,7 +743,7 @@ class TYtDataSinkExecTransformer : public TExecTransformerBase {
742743
NYT::TNode spec;
743744
rowSpec.FillCodecNode(spec[YqlRowSpecAttribute]);
744745
outSpec = NYT::TNode::CreateMap()(TString{YqlIOSpecTables}, NYT::TNode::CreateList().Add(spec));
745-
type = rowSpec.GetTypeNode();
746+
type = NCommon::TypeToYsonNode(rowSpec.GetExtendedType(ctx));
746747
}
747748

748749
// These settings will be passed to YT peephole callback from DQ

0 commit comments

Comments
 (0)