From f94e573477120d6226a11af667a998a487a7c1a8 Mon Sep 17 00:00:00 2001 From: vlad-gogov Date: Wed, 15 Jan 2025 12:06:05 +0000 Subject: [PATCH] resolve conversations --- ydb/core/base/appdata_fwd.h | 6 +++--- ydb/core/formats/arrow/serializer/native.h | 8 +++++--- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/ydb/core/base/appdata_fwd.h b/ydb/core/base/appdata_fwd.h index ae2bc9ab9fa5..09da416778ab 100644 --- a/ydb/core/base/appdata_fwd.h +++ b/ydb/core/base/appdata_fwd.h @@ -297,15 +297,15 @@ inline TAppData* AppData(NActors::TActorSystem* actorSystem) { } inline bool HasAppData() { - return !!NActors::TlsActivationContext; + return !!NActors::TlsActivationContext && NActors::TlsActivationContext->ExecutorThread.ActorSystem && + NActors::TlsActivationContext->ExecutorThread.ActorSystem->AppData(); } inline TAppData& AppDataVerified() { Y_ABORT_UNLESS(HasAppData()); auto& actorSystem = NActors::TlsActivationContext->ExecutorThread.ActorSystem; - Y_ABORT_UNLESS(actorSystem); TAppData* const x = actorSystem->AppData(); - Y_ABORT_UNLESS(x && x->Magic == TAppData::MagicTag); + Y_ABORT_UNLESS(x->Magic == TAppData::MagicTag); return *x; } diff --git a/ydb/core/formats/arrow/serializer/native.h b/ydb/core/formats/arrow/serializer/native.h index b76ad9577d98..c3cbad051c39 100644 --- a/ydb/core/formats/arrow/serializer/native.h +++ b/ydb/core/formats/arrow/serializer/native.h @@ -23,6 +23,10 @@ class TNativeSerializer: public ISerializer { private: arrow::ipc::IpcWriteOptions Options; + static inline std::shared_ptr DefaultCodec() { + return NArrow::TStatusValidator::GetValid(arrow::util::Codec::Create(arrow::Compression::type::ZSTD, 1)); + } + TConclusion> BuildCodec(const arrow::Compression::type& cType, const std::optional level) const; static const inline TFactory::TRegistrator Registrator = TFactory::TRegistrator(GetClassNameStatic()); protected: @@ -63,14 +67,12 @@ class TNativeSerializer: public ISerializer { static std::shared_ptr GetDefaultCodec() { if (!HasAppData() || (!AppData()->ColumnShardConfig.HasDefaultCompression() && !AppData()->ColumnShardConfig.HasDefaultCompressionLevel())) { - return NArrow::TStatusValidator::GetValid(arrow::util::Codec::Create(arrow::Compression::type::ZSTD, 1)); + return DefaultCodec(); } arrow::Compression::type codec = GetDefaultCompressionType(); if (AppData()->ColumnShardConfig.HasDefaultCompressionLevel()) { return NArrow::TStatusValidator::GetValid( arrow::util::Codec::Create(codec, AppData()->ColumnShardConfig.GetDefaultCompressionLevel())); - } else if (codec == arrow::Compression::ZSTD) { - return NArrow::TStatusValidator::GetValid(arrow::util::Codec::Create(codec, 1)); } return NArrow::TStatusValidator::GetValid(arrow::util::Codec::Create(codec)); }