1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989 |
- #include "yql_codec.h"
- #include "yql_codec_type_flags.h"
- #include <yql/essentials/core/yql_expr_type_annotation.h>
- #include <yql/essentials/public/decimal/yql_decimal.h>
- #include <yql/essentials/public/decimal/yql_decimal_serialize.h>
- #include <yql/essentials/minikql/mkql_node_cast.h>
- #include <yql/essentials/minikql/mkql_string_util.h>
- #include <yql/essentials/minikql/mkql_type_builder.h>
- #include <yql/essentials/minikql/computation/mkql_computation_node_pack.h>
- #include <yql/essentials/public/result_format/yql_restricted_yson.h>
- #include <yql/essentials/utils/yql_panic.h>
- #include <yql/essentials/utils/swap_bytes.h>
- #include <library/cpp/yson/node/node_io.h>
- #include <library/cpp/yson/writer.h>
- #include <library/cpp/string_utils/base64/base64.h>
- #include <library/cpp/yson/parser.h>
- #include <library/cpp/yson/detail.h>
- #include <util/string/cast.h>
- #include <util/generic/map.h>
- #include <yt/yt/library/decimal/decimal.h>
- namespace NYql {
- namespace NCommon {
- using namespace NKikimr;
- using namespace NKikimr::NMiniKQL;
- using namespace NYson::NDetail;
- void WriteYsonValueImpl(NResult::TYsonResultWriter& writer, const NUdf::TUnboxedValuePod& value, TType* type,
- const TVector<ui32>* structPositions) {
- // Result format
- switch (type->GetKind()) {
- case TType::EKind::Void:
- writer.OnVoid();
- return;
- case TType::EKind::Null:
- writer.OnNull();
- return;
- case TType::EKind::EmptyList:
- writer.OnEmptyList();
- return;
- case TType::EKind::EmptyDict:
- writer.OnEmptyDict();
- return;
- case TType::EKind::Data:
- {
- auto dataType = AS_TYPE(TDataType, type);
- switch (*dataType->GetDataSlot()) {
- case NUdf::EDataSlot::Bool:
- writer.OnBooleanScalar(value.Get<bool>());
- return;
- case NUdf::EDataSlot::Int32:
- writer.OnInt64Scalar(value.Get<i32>());
- return;
- case NUdf::EDataSlot::Uint32:
- writer.OnUint64Scalar(value.Get<ui32>());
- return;
- case NUdf::EDataSlot::Int64:
- writer.OnInt64Scalar(value.Get<i64>());
- return;
- case NUdf::EDataSlot::Uint64:
- writer.OnUint64Scalar(value.Get<ui64>());
- return;
- case NUdf::EDataSlot::Uint8:
- writer.OnUint64Scalar(value.Get<ui8>());
- return;
- case NUdf::EDataSlot::Int8:
- writer.OnInt64Scalar(value.Get<i8>());
- return;
- case NUdf::EDataSlot::Uint16:
- writer.OnUint64Scalar(value.Get<ui16>());
- return;
- case NUdf::EDataSlot::Int16:
- writer.OnInt64Scalar(value.Get<i16>());
- return;
- case NUdf::EDataSlot::Float:
- writer.OnFloatScalar(value.Get<float>());
- return;
- case NUdf::EDataSlot::Double:
- writer.OnDoubleScalar(value.Get<double>());
- return;
- case NUdf::EDataSlot::Json:
- case NUdf::EDataSlot::Utf8:
- // assume underlying string is utf8
- writer.OnUtf8StringScalar(value.AsStringRef());
- return;
- case NUdf::EDataSlot::String:
- case NUdf::EDataSlot::Uuid:
- case NUdf::EDataSlot::DyNumber:
- writer.OnStringScalar(value.AsStringRef());
- return;
- case NUdf::EDataSlot::Decimal: {
- const auto params = static_cast<TDataDecimalType*>(type)->GetParams();
- const auto str = NDecimal::ToString(value.GetInt128(), params.first, params.second);
- const auto size = str ? std::strlen(str) : 0;
- writer.OnUtf8StringScalar(TStringBuf(str, size));
- return;
- }
- case NUdf::EDataSlot::Yson:
- NResult::EncodeRestrictedYson(writer, value.AsStringRef());
- return;
- case NUdf::EDataSlot::Date:
- writer.OnUint64Scalar(value.Get<ui16>());
- return;
- case NUdf::EDataSlot::Datetime:
- writer.OnUint64Scalar(value.Get<ui32>());
- return;
- case NUdf::EDataSlot::Timestamp:
- writer.OnUint64Scalar(value.Get<ui64>());
- return;
- case NUdf::EDataSlot::Interval:
- writer.OnInt64Scalar(value.Get<i64>());
- return;
- case NUdf::EDataSlot::TzDate:
- case NUdf::EDataSlot::TzDatetime:
- case NUdf::EDataSlot::TzTimestamp:
- case NUdf::EDataSlot::TzDate32:
- case NUdf::EDataSlot::TzDatetime64:
- case NUdf::EDataSlot::TzTimestamp64:
- case NUdf::EDataSlot::JsonDocument: {
- const NUdf::TUnboxedValue out(ValueToString(*dataType->GetDataSlot(), value));
- writer.OnUtf8StringScalar(out.AsStringRef());
- return;
- }
- case NUdf::EDataSlot::Date32:
- writer.OnInt64Scalar(value.Get<i32>());
- return;
- case NUdf::EDataSlot::Datetime64:
- writer.OnInt64Scalar(value.Get<i64>());
- return;
- case NUdf::EDataSlot::Timestamp64:
- writer.OnInt64Scalar(value.Get<i64>());
- return;
- case NUdf::EDataSlot::Interval64:
- writer.OnInt64Scalar(value.Get<i64>());
- return;
- }
- }
- break;
- case TType::EKind::Pg:
- {
- auto pgType = AS_TYPE(TPgType, type);
- WriteYsonValuePg(writer, value, pgType, structPositions);
- return;
- }
- case TType::EKind::Struct:
- {
- writer.OnBeginList();
- auto structType = AS_TYPE(TStructType, type);
- if (structPositions && structPositions->size() != structType->GetMembersCount()) {
- YQL_ENSURE(false, "Invalid struct positions");
- }
- for (ui32 i = 0, e = structType->GetMembersCount(); i < e; ++i) {
- const ui32 pos = structPositions ? (*structPositions)[i] : i;
- if (pos < e) {
- writer.OnListItem();
- WriteYsonValueImpl(writer, value.GetElement(pos), structType->GetMemberType(pos), nullptr);
- }
- }
- writer.OnEndList();
- return;
- }
- case TType::EKind::List:
- {
- writer.OnBeginList();
- auto listType = AS_TYPE(TListType, type);
- const auto it = value.GetListIterator();
- for (NUdf::TUnboxedValue item; it.Next(item);) {
- writer.OnListItem();
- WriteYsonValueImpl(writer, item, listType->GetItemType(), nullptr);
- }
- writer.OnEndList();
- return;
- }
- case TType::EKind::Optional:
- {
- if (!value) {
- writer.OnEntity();
- } else {
- writer.OnBeginList();
- auto optionalType = AS_TYPE(TOptionalType, type);
- writer.OnListItem();
- WriteYsonValueImpl(writer, value.GetOptionalValue(), optionalType->GetItemType(), nullptr);
- writer.OnEndList();
- }
- return;
- }
- case TType::EKind::Dict:
- {
- writer.OnBeginList();
- auto dictType = AS_TYPE(TDictType, type);
- const auto it = value.GetDictIterator();
- for (NUdf::TUnboxedValue key, payload; it.NextPair(key, payload);) {
- writer.OnListItem();
- writer.OnBeginList();
- {
- writer.OnListItem();
- WriteYsonValueImpl(writer, key, dictType->GetKeyType(), nullptr);
- writer.OnListItem();
- WriteYsonValueImpl(writer, payload, dictType->GetPayloadType(), nullptr);
- }
- writer.OnEndList();
- }
- writer.OnEndList();
- }
- return;
- case TType::EKind::Tuple:
- {
- writer.OnBeginList();
- auto tupleType = AS_TYPE(TTupleType, type);
- for (ui32 i = 0, e = tupleType->GetElementsCount(); i < e; ++i) {
- writer.OnListItem();
- WriteYsonValueImpl(writer, value.GetElement(i), tupleType->GetElementType(i), nullptr);
- }
- writer.OnEndList();
- return;
- }
- case TType::EKind::Variant:
- {
- writer.OnBeginList();
- auto underlyingType = AS_TYPE(TVariantType, type)->GetUnderlyingType();
- writer.OnListItem();
- auto index = value.GetVariantIndex();
- writer.OnUint64Scalar(index);
- writer.OnListItem();
- if (underlyingType->IsTuple()) {
- WriteYsonValueImpl(writer, value.GetVariantItem(), AS_TYPE(TTupleType, underlyingType)->GetElementType(index), nullptr);
- } else {
- WriteYsonValueImpl(writer, value.GetVariantItem(), AS_TYPE(TStructType, underlyingType)->GetMemberType(index), nullptr);
- }
- writer.OnEndList();
- return;
- }
- case TType::EKind::Tagged:
- {
- auto underlyingType = AS_TYPE(TTaggedType, type)->GetBaseType();
- WriteYsonValueImpl(writer, value, underlyingType, structPositions);
- return;
- }
- default:
- YQL_ENSURE(false, "unknown type " << type->GetKindAsStr());
- }
- }
- void WriteYsonValue(NYson::TYsonConsumerBase& writer, const NUdf::TUnboxedValuePod& value, TType* type,
- const TVector<ui32>* structPositions)
- {
- NResult::TYsonResultWriter resultWriter(writer);
- WriteYsonValueImpl(resultWriter, value, type, structPositions);
- }
- TString WriteYsonValue(const NUdf::TUnboxedValuePod& value, TType* type, const TVector<ui32>* structPositions,
- NYson::EYsonFormat format) {
- TStringStream str;
- NYson::TYsonWriter writer(&str, format);
- WriteYsonValue(writer, value, type, structPositions);
- return str.Str();
- }
- TCodecContext::TCodecContext(
- const TTypeEnvironment& env,
- const IFunctionRegistry& functionRegistry,
- const NKikimr::NMiniKQL::THolderFactory* holderFactory /* = nullptr */
- )
- : Env(env)
- , Builder(Env, functionRegistry)
- , HolderFactory(holderFactory)
- {
- }
- TMaybe<TVector<ui32>> CreateStructPositions(TType* inputType, const TVector<TString>* columns) {
- if (inputType->GetKind() != TType::EKind::Struct) {
- return Nothing();
- }
-
- auto inputStruct = AS_TYPE(TStructType, inputType);
- TMap<TStringBuf, ui32> members;
- TVector<ui32> structPositions(inputStruct->GetMembersCount(), Max<ui32>());
- for (ui32 i = 0; i < inputStruct->GetMembersCount(); ++i) {
- if (columns) {
- members.insert(std::make_pair(inputStruct->GetMemberName(i), i));
- } else {
- structPositions[i] = i;
- }
- }
- if (columns) {
- TColumnOrder order(*columns);
- ui32 pos = 0;
- for (auto& [column, gen_column]: order) {
- const ui32* idx = members.FindPtr(gen_column);
- YQL_ENSURE(idx, "Unknown member: " << gen_column);
- structPositions[pos] = *idx;
- ++pos;
- }
- }
- return structPositions;
- }
- namespace {
- NYT::TNode DataValueToNode(const NKikimr::NUdf::TUnboxedValuePod& value, NKikimr::NMiniKQL::TType* type) {
- YQL_ENSURE(type->GetKind() == TType::EKind::Data);
- auto dataType = AS_TYPE(TDataType, type);
- switch (dataType->GetSchemeType()) {
- case NUdf::TDataType<i32>::Id:
- return NYT::TNode(value.Get<i32>());
- case NUdf::TDataType<i64>::Id:
- return NYT::TNode(value.Get<i64>());
- case NUdf::TDataType<ui32>::Id:
- return NYT::TNode(value.Get<ui32>());
- case NUdf::TDataType<ui64>::Id:
- return NYT::TNode(value.Get<ui64>());
- case NUdf::TDataType<float>::Id:
- return NYT::TNode(value.Get<float>());
- case NUdf::TDataType<double>::Id:
- return NYT::TNode(value.Get<double>());
- case NUdf::TDataType<bool>::Id:
- return NYT::TNode(value.Get<bool>());
- case NUdf::TDataType<ui8>::Id:
- return NYT::TNode((ui64)value.Get<ui8>());
- case NUdf::TDataType<i8>::Id:
- return NYT::TNode((i64)value.Get<i8>());
- case NUdf::TDataType<ui16>::Id:
- return NYT::TNode((ui64)value.Get<ui16>());
- case NUdf::TDataType<i16>::Id:
- return NYT::TNode((i64)value.Get<i16>());
- case NUdf::TDataType<char*>::Id:
- case NUdf::TDataType<NUdf::TUtf8>::Id:
- case NUdf::TDataType<NUdf::TJson>::Id:
- case NUdf::TDataType<NUdf::TUuid>::Id:
- return NYT::TNode(TString(value.AsStringRef()));
- case NUdf::TDataType<NUdf::TYson>::Id:
- return NYT::NodeFromYsonString(TString(value.AsStringRef()));
- case NUdf::TDataType<NUdf::TDate>::Id:
- return NYT::TNode((ui64)value.Get<ui16>());
- case NUdf::TDataType<NUdf::TDatetime>::Id:
- return NYT::TNode((ui64)value.Get<ui32>());
- case NUdf::TDataType<NUdf::TTimestamp>::Id:
- return NYT::TNode(value.Get<ui64>());
- case NUdf::TDataType<NUdf::TInterval>::Id:
- return NYT::TNode(value.Get<i64>());
- case NUdf::TDataType<NUdf::TTzDate>::Id: {
- TStringStream out;
- out << value.Get<ui16>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return NYT::TNode(out.Str());
- }
- case NUdf::TDataType<NUdf::TTzDatetime>::Id: {
- TStringStream out;
- out << value.Get<ui32>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return NYT::TNode(out.Str());
- }
- case NUdf::TDataType<NUdf::TTzTimestamp>::Id: {
- TStringStream out;
- out << value.Get<ui64>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return NYT::TNode(out.Str());
- }
- case NUdf::TDataType<NUdf::TDate32>::Id:
- return NYT::TNode((i64)value.Get<i32>());
- case NUdf::TDataType<NUdf::TDatetime64>::Id:
- return NYT::TNode(value.Get<i64>());
- case NUdf::TDataType<NUdf::TTimestamp64>::Id:
- return NYT::TNode(value.Get<i64>());
- case NUdf::TDataType<NUdf::TInterval64>::Id:
- return NYT::TNode(value.Get<i64>());
- case NUdf::TDataType<NUdf::TDecimal>::Id: {
- const auto params = static_cast<NKikimr::NMiniKQL::TDataDecimalType*>(type)->GetParams();
- return NYT::TNode(NDecimal::ToString(value.GetInt128(), params.first, params.second));
- }
- case NUdf::TDataType<NUdf::TJsonDocument>::Id: {
- NUdf::TUnboxedValue json = ValueToString(EDataSlot::JsonDocument, value);
- return NYT::TNode(ToString(TStringBuf(value.AsStringRef())));
- }
- case NUdf::TDataType<NUdf::TTzDate32>::Id: {
- TStringStream out;
- out << value.Get<i32>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return NYT::TNode(out.Str());
- }
- case NUdf::TDataType<NUdf::TTzDatetime64>::Id: {
- TStringStream out;
- out << value.Get<i64>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return NYT::TNode(out.Str());
- }
- case NUdf::TDataType<NUdf::TTzTimestamp64>::Id: {
- TStringStream out;
- out << value.Get<i64>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return NYT::TNode(out.Str());
- }
- }
- YQL_ENSURE(false, "Unsupported type: " << static_cast<int>(dataType->GetSchemeType()));
- }
- TExprNode::TPtr DataNodeToExprLiteral(TPositionHandle pos, const TTypeAnnotationNode& type, const NYT::TNode& node, TExprContext& ctx) {
- YQL_ENSURE(type.GetKind() == ETypeAnnotationKind::Data, "Expecting data type, got: " << type);
- TString strData;
- if (type.Cast<TDataExprType>()->GetSlot() == EDataSlot::Yson) {
- strData = NYT::NodeToYsonString(node);
- } else {
- switch (node.GetType()) {
- case NYT::TNode::String:
- strData = node.AsString();
- break;
- case NYT::TNode::Int64:
- strData = ToString(node.AsInt64());
- break;
- case NYT::TNode::Uint64:
- strData = ToString(node.AsUint64());
- break;
- case NYT::TNode::Double:
- strData = FloatToString(node.AsDouble());
- break;
- case NYT::TNode::Bool:
- strData = ToString(node.AsBool());
- break;
- default:
- YQL_ENSURE(false, "Unexpected Yson type: " << node.GetType() << " while deserializing literal of type " << type);
- }
- }
- return ctx.Builder(pos)
- .Callable(type.Cast<TDataExprType>()->GetName())
- .Do([&](TExprNodeBuilder& parent) -> TExprNodeBuilder& {
- parent.Atom(0, strData);
- if (IsDataTypeDecimal(type.Cast<TDataExprType>()->GetSlot())) {
- auto decimalType = type.Cast<TDataExprParamsType>();
- parent.Atom(1, decimalType->GetParamOne());
- parent.Atom(2, decimalType->GetParamTwo());
- }
- return parent;
- })
- .Seal()
- .Build();
- }
- TString DataValueToString(const NKikimr::NUdf::TUnboxedValuePod& value, const TDataExprType* type) {
- switch (type->GetSlot()) {
- case NUdf::EDataSlot::Int32:
- case NUdf::EDataSlot::Date32:
- return ToString(value.Get<i32>());
- case NUdf::EDataSlot::Int64:
- case NUdf::EDataSlot::Interval:
- case NUdf::EDataSlot::Datetime64:
- case NUdf::EDataSlot::Timestamp64:
- case NUdf::EDataSlot::Interval64:
- return ToString(value.Get<i64>());
- case NUdf::EDataSlot::Uint32:
- case NUdf::EDataSlot::Datetime:
- return ToString(value.Get<ui32>());
- case NUdf::EDataSlot::Uint64:
- case NUdf::EDataSlot::Timestamp:
- return ToString(value.Get<ui64>());
- case NUdf::EDataSlot::Float:
- return ::FloatToString(value.Get<float>());
- case NUdf::EDataSlot::Double:
- return ::FloatToString(value.Get<double>());
- case NUdf::EDataSlot::Bool:
- return ToString(value.Get<bool>());
- case NUdf::EDataSlot::Uint8:
- return ToString(static_cast<unsigned int>(value.Get<ui8>()));
- case NUdf::EDataSlot::Int8:
- return ToString(static_cast<int>(value.Get<i8>()));
- case NUdf::EDataSlot::Uint16:
- case NUdf::EDataSlot::Date:
- return ToString(static_cast<unsigned int>(value.Get<ui16>()));
- case NUdf::EDataSlot::Int16:
- return ToString(static_cast<int>(value.Get<i16>()));
- case NUdf::EDataSlot::String:
- case NUdf::EDataSlot::Utf8:
- case NUdf::EDataSlot::Json:
- case NUdf::EDataSlot::Uuid:
- case NUdf::EDataSlot::Yson:
- case NUdf::EDataSlot::DyNumber:
- return ToString((TStringBuf)value.AsStringRef());
- case NUdf::EDataSlot::Decimal:
- {
- const auto params = dynamic_cast<const TDataExprParamsType*>(type);
- YQL_ENSURE(params, "Unable to cast decimal params");
- return NDecimal::ToString(value.GetInt128(), FromString<ui8>(params->GetParamOne()), FromString<ui8>(params->GetParamTwo()));
- }
- case NUdf::EDataSlot::TzDate: {
- TStringStream out;
- out << value.Get<ui16>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return out.Str();
- }
- case NUdf::EDataSlot::TzDatetime: {
- TStringStream out;
- out << value.Get<ui32>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return out.Str();
- }
- case NUdf::EDataSlot::TzTimestamp: {
- TStringStream out;
- out << value.Get<ui64>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return out.Str();
- }
- case NUdf::EDataSlot::TzDate32: {
- TStringStream out;
- out << value.Get<i32>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return out.Str();
- }
- case NUdf::EDataSlot::TzDatetime64: {
- TStringStream out;
- out << value.Get<i64>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return out.Str();
- }
- case NUdf::EDataSlot::TzTimestamp64: {
- TStringStream out;
- out << value.Get<i64>() << "," << NKikimr::NMiniKQL::GetTimezoneIANAName(value.GetTimezoneId());
- return out.Str();
- }
- case NUdf::EDataSlot::JsonDocument: {
- NUdf::TUnboxedValue json = ValueToString(EDataSlot::JsonDocument, value);
- return ToString(TStringBuf(value.AsStringRef()));
- }
- }
- Y_ABORT("Unexpected");
- }
- } //namespace
- NYT::TNode ValueToNode(const NKikimr::NUdf::TUnboxedValuePod& value, NKikimr::NMiniKQL::TType* type) {
- NYT::TNode result;
- switch (type->GetKind()) {
- case TType::EKind::Optional: {
- result = NYT::TNode::CreateList();
- if (value) {
- result.Add(ValueToNode(value.GetOptionalValue(), AS_TYPE(TOptionalType, type)->GetItemType()));
- }
- break;
- }
- case TType::EKind::Tuple: {
- auto tupleType = AS_TYPE(TTupleType, type);
- result = NYT::TNode::CreateList();
- for (ui32 i = 0; i < tupleType->GetElementsCount(); ++i) {
- result.Add(ValueToNode(value.GetElement(i), tupleType->GetElementType(i)));
- }
- break;
- }
- case TType::EKind::List: {
- auto listType = AS_TYPE(TListType, type);
- result = NYT::TNode::CreateList();
- const auto iter = value.GetListIterator();
- for (NUdf::TUnboxedValue item; iter.Next(item); ) {
- result.Add(ValueToNode(item, listType->GetItemType()));
- }
- break;
- }
- default: {
- result = DataValueToNode(value, type);
- }
- }
- return result;
- }
- TExprNode::TPtr NodeToExprLiteral(TPositionHandle pos, const TTypeAnnotationNode& type, const NYT::TNode& node, TExprContext& ctx) {
- TExprNode::TPtr result;
- switch(type.GetKind()) {
- case ETypeAnnotationKind::Optional: {
- YQL_ENSURE(node.IsList() || node.IsNull());
- if (node.IsNull() || node.AsList().empty()) {
- return ctx.NewCallable(pos, "Nothing", { ExpandType(pos, type, ctx) });
- }
- YQL_ENSURE(node.AsList().size() == 1);
- result = ctx.NewCallable(pos, "Just", {
- NodeToExprLiteral(pos, *type.Cast<TOptionalExprType>()->GetItemType(), node.AsList().front(), ctx)
- });
- break;
- }
- case ETypeAnnotationKind::Tuple: {
- YQL_ENSURE(node.IsList());
- const TTypeAnnotationNode::TListType& itemTypes = type.Cast<TTupleExprType>()->GetItems();
- const auto& items = node.AsList();
- YQL_ENSURE(itemTypes.size() == items.size());
- TExprNodeList resultNodes;
- for (size_t i = 0; i < items.size(); ++i) {
- resultNodes.push_back(NodeToExprLiteral(pos, *itemTypes[i], items[i], ctx));
- }
- result = ctx.NewList(pos, std::move(resultNodes));
- break;
- }
- case ETypeAnnotationKind::List: {
- YQL_ENSURE(node.IsList());
- const TTypeAnnotationNode& itemType = *type.Cast<TListExprType>()->GetItemType();
- if (node.AsList().empty()) {
- return ctx.NewCallable(pos, "List", { ExpandType(pos, *ctx.MakeType<TListExprType>(&itemType), ctx) });
- }
- TExprNodeList children;
- for (auto& child : node.AsList()) {
- children.push_back(NodeToExprLiteral(pos, itemType, child, ctx));
- }
- result = ctx.NewCallable(pos, "AsList", std::move(children));
- break;
- }
- default: {
- result = DataNodeToExprLiteral(pos, type, node, ctx);
- }
- }
- return result;
- }
- void CopyYsonWithAttrs(char cmd, TInputBuf& buf, TVector<char>& yson) {
- if (cmd == BeginAttributesSymbol) {
- yson.push_back(cmd);
- cmd = buf.Read();
- for (;;) {
- if (cmd == EndAttributesSymbol) {
- yson.push_back(cmd);
- cmd = buf.Read();
- break;
- }
- CHECK_EXPECTED(cmd, StringMarker);
- yson.push_back(cmd);
- i32 length = buf.CopyVarI32(yson);
- CHECK_STRING_LENGTH(length);
- buf.CopyMany(length, yson);
- EXPECTED_COPY(buf, KeyValueSeparatorSymbol, yson);
- cmd = buf.Read();
- CopyYsonWithAttrs(cmd, buf, yson);
- cmd = buf.Read();
- if (cmd == KeyedItemSeparatorSymbol) {
- yson.push_back(cmd);
- cmd = buf.Read();
- }
- }
- }
- CopyYson(cmd, buf, yson);
- }
- void CopyYson(char cmd, TInputBuf& buf, TVector<char>& yson) {
- switch (cmd) {
- case EntitySymbol:
- case TrueMarker:
- case FalseMarker:
- yson.push_back(cmd);
- break;
- case Int64Marker:
- yson.push_back(cmd);
- buf.CopyVarI64(yson);
- break;
- case Uint64Marker:
- yson.push_back(cmd);
- buf.CopyVarUI64(yson);
- break;
- case DoubleMarker:
- yson.push_back(cmd);
- buf.CopyMany(8, yson);
- break;
- case StringMarker: {
- yson.push_back(cmd);
- i32 length = buf.CopyVarI32(yson);
- CHECK_STRING_LENGTH(length);
- buf.CopyMany(length, yson);
- break;
- }
- case BeginListSymbol: {
- yson.push_back(cmd);
- cmd = buf.Read();
- for (;;) {
- if (cmd == EndListSymbol) {
- yson.push_back(cmd);
- break;
- }
- CopyYsonWithAttrs(cmd, buf, yson);
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- yson.push_back(cmd);
- cmd = buf.Read();
- }
- }
- break;
- }
- case BeginMapSymbol: {
- yson.push_back(cmd);
- cmd = buf.Read();
- for (;;) {
- if (cmd == EndMapSymbol) {
- yson.push_back(cmd);
- break;
- }
- CHECK_EXPECTED(cmd, StringMarker);
- yson.push_back(cmd);
- i32 length = buf.CopyVarI32(yson);
- CHECK_STRING_LENGTH(length);
- buf.CopyMany(length, yson);
- EXPECTED_COPY(buf, KeyValueSeparatorSymbol, yson);
- cmd = buf.Read();
- CopyYsonWithAttrs(cmd, buf, yson);
- cmd = buf.Read();
- if (cmd == KeyedItemSeparatorSymbol) {
- yson.push_back(cmd);
- cmd = buf.Read();
- }
- }
- break;
- }
- default:
- YQL_ENSURE(false, "Unexpected yson character: " << cmd);
- }
- }
- void SkipYson(char cmd, TInputBuf& buf) {
- auto& yson = buf.YsonBuffer();
- yson.clear();
- CopyYsonWithAttrs(cmd, buf, yson);
- }
- NUdf::TUnboxedValue ReadYsonStringInResultFormat(char cmd, TInputBuf& buf) {
- NUdf::TUnboxedValue result;
- const bool needDecode = (cmd == BeginListSymbol);
- if (needDecode) {
- cmd = buf.Read();
- }
- CHECK_EXPECTED(cmd, StringMarker);
- const i32 length = buf.ReadVarI32();
- CHECK_STRING_LENGTH(length);
- TTempBuf tmpBuf(length);
- buf.ReadMany(tmpBuf.Data(), length);
- if (needDecode) {
- TString decoded = Base64Decode(TStringBuf(tmpBuf.Data(), length));
- result = NUdf::TUnboxedValue(MakeString(NUdf::TStringRef(decoded)));
- } else {
- result = NUdf::TUnboxedValue(MakeString(NUdf::TStringRef(tmpBuf.Data(), length)));
- }
- if (needDecode) {
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- }
- return result;
- }
- TStringBuf ReadNextString(char cmd, TInputBuf& buf) {
- CHECK_EXPECTED(cmd, StringMarker);
- return buf.ReadYtString();
- }
- template <typename T>
- T ReadNextSerializedNumber(char cmd, TInputBuf& buf) {
- auto nextString = ReadNextString(cmd, buf);
- if constexpr (!std::numeric_limits<T>::is_integer) {
- if (nextString == "inf" || nextString == "+inf") {
- return std::numeric_limits<T>::infinity();
- } else if (nextString == "-inf") {
- return -std::numeric_limits<T>::infinity();
- } else if (nextString == "nan") {
- return std::numeric_limits<T>::quiet_NaN();
- }
- }
- return FromString<T>(nextString);
- }
- template <typename T>
- T ReadYsonFloatNumber(char cmd, TInputBuf& buf, bool isTableFormat) {
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, DoubleMarker);
- double dbl;
- buf.ReadMany((char*)&dbl, sizeof(dbl));
- return dbl;
- }
- return ReadNextSerializedNumber<T>(cmd, buf);
- }
- NUdf::TUnboxedValue ReadYsonValue(TType* type, ui64 nativeYtTypeFlags,
- const NKikimr::NMiniKQL::THolderFactory& holderFactory, char cmd, TInputBuf& buf, bool isTableFormat) {
- switch (type->GetKind()) {
- case TType::EKind::Variant: {
- auto varType = static_cast<TVariantType*>(type);
- auto underlyingType = varType->GetUnderlyingType();
- if (isTableFormat && (nativeYtTypeFlags & NTCF_COMPLEX)) {
- CHECK_EXPECTED(cmd, BeginListSymbol);
- cmd = buf.Read();
- TType* type = nullptr;
- i64 index = 0;
- if (cmd == StringMarker) {
- YQL_ENSURE(underlyingType->IsStruct(), "Expected struct as underlying type");
- auto structType = static_cast<TStructType*>(underlyingType);
- auto nameBuffer = ReadNextString(cmd, buf);
- auto foundIndex = structType->FindMemberIndex(nameBuffer);
- YQL_ENSURE(foundIndex.Defined(), "Unexpected member: " << nameBuffer);
- index = *foundIndex;
- type = varType->GetAlternativeType(index);
- } else {
- YQL_ENSURE(cmd == Int64Marker || cmd == Uint64Marker);
- YQL_ENSURE(underlyingType->IsTuple(), "Expected tuple as underlying type");
- if (cmd == Uint64Marker) {
- index = buf.ReadVarUI64();
- } else {
- index = buf.ReadVarI64();
- }
- YQL_ENSURE(0 <= index && index < varType->GetAlternativesCount(), "Unexpected member index: " << index);
- type = varType->GetAlternativeType(index);
- }
- cmd = buf.Read();
- CHECK_EXPECTED(cmd, ListItemSeparatorSymbol);
- cmd = buf.Read();
- auto value = ReadYsonValue(type, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- cmd = buf.Read();
- if (cmd != EndListSymbol) {
- CHECK_EXPECTED(cmd, ListItemSeparatorSymbol);
- cmd = buf.Read();
- CHECK_EXPECTED(cmd, EndListSymbol);
- }
- return holderFactory.CreateVariantHolder(value.Release(), index);
- } else {
- if (cmd == StringMarker) {
- YQL_ENSURE(underlyingType->IsStruct(), "Expected struct as underlying type");
- auto name = ReadNextString(cmd, buf);
- auto index = static_cast<TStructType*>(underlyingType)->FindMemberIndex(name);
- YQL_ENSURE(index, "Unexpected member: " << name);
- YQL_ENSURE(static_cast<TStructType*>(underlyingType)->GetMemberType(*index)->IsVoid(), "Expected Void as underlying type");
- return holderFactory.CreateVariantHolder(NUdf::TUnboxedValuePod::Zero(), *index);
- }
- CHECK_EXPECTED(cmd, BeginListSymbol);
- cmd = buf.Read();
- i64 index = 0;
- if (isTableFormat) {
- YQL_ENSURE(cmd == Int64Marker || cmd == Uint64Marker);
- if (cmd == Uint64Marker) {
- index = buf.ReadVarUI64();
- } else {
- index = buf.ReadVarI64();
- }
- } else {
- if (cmd == BeginListSymbol) {
- cmd = buf.Read();
- YQL_ENSURE(underlyingType->IsStruct(), "Expected struct as underlying type");
- auto name = ReadNextString(cmd, buf);
- auto foundIndex = static_cast<TStructType*>(underlyingType)->FindMemberIndex(name);
- YQL_ENSURE(foundIndex, "Unexpected member: " << name);
- index = *foundIndex;
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- } else {
- index = ReadNextSerializedNumber<ui64>(cmd, buf);
- }
- }
- YQL_ENSURE(index < varType->GetAlternativesCount(), "Bad variant alternative: " << index << ", only " <<
- varType->GetAlternativesCount() << " are available");
- YQL_ENSURE(underlyingType->IsTuple() || underlyingType->IsStruct(), "Wrong underlying type");
- TType* itemType;
- if (underlyingType->IsTuple()) {
- itemType = static_cast<TTupleType*>(underlyingType)->GetElementType(index);
- }
- else {
- itemType = static_cast<TStructType*>(underlyingType)->GetMemberType(index);
- }
- EXPECTED(buf, ListItemSeparatorSymbol);
- cmd = buf.Read();
- auto value = ReadYsonValue(itemType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- return holderFactory.CreateVariantHolder(value.Release(), index);
- }
- }
- case TType::EKind::Data: {
- auto schemeType = static_cast<TDataType*>(type)->GetSchemeType();
- switch (schemeType) {
- case NUdf::TDataType<bool>::Id:
- YQL_ENSURE(cmd == FalseMarker || cmd == TrueMarker, "Expected either true or false, but got: " << TString(cmd).Quote());
- return NUdf::TUnboxedValuePod(cmd == TrueMarker);
- case NUdf::TDataType<ui8>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Uint64Marker);
- return NUdf::TUnboxedValuePod(ui8(buf.ReadVarUI64()));
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<ui8>(cmd, buf));
- case NUdf::TDataType<i8>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Int64Marker);
- return NUdf::TUnboxedValuePod(i8(buf.ReadVarI64()));
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<i8>(cmd, buf));
- case NUdf::TDataType<ui16>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Uint64Marker);
- return NUdf::TUnboxedValuePod(ui16(buf.ReadVarUI64()));
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<ui16>(cmd, buf));
- case NUdf::TDataType<i16>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Int64Marker);
- return NUdf::TUnboxedValuePod(i16(buf.ReadVarI64()));
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<i16>(cmd, buf));
- case NUdf::TDataType<i32>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Int64Marker);
- return NUdf::TUnboxedValuePod(i32(buf.ReadVarI64()));
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<i32>(cmd, buf));
- case NUdf::TDataType<ui32>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Uint64Marker);
- return NUdf::TUnboxedValuePod(ui32(buf.ReadVarUI64()));
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<ui32>(cmd, buf));
- case NUdf::TDataType<i64>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Int64Marker);
- return NUdf::TUnboxedValuePod(buf.ReadVarI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<i64>(cmd, buf));
- case NUdf::TDataType<ui64>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Uint64Marker);
- return NUdf::TUnboxedValuePod(buf.ReadVarUI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<ui64>(cmd, buf));
- case NUdf::TDataType<float>::Id:
- return NUdf::TUnboxedValuePod(ReadYsonFloatNumber<float>(cmd, buf, isTableFormat));
- case NUdf::TDataType<double>::Id:
- return NUdf::TUnboxedValuePod(ReadYsonFloatNumber<double>(cmd, buf, isTableFormat));
- case NUdf::TDataType<NUdf::TUtf8>::Id:
- case NUdf::TDataType<char*>::Id:
- case NUdf::TDataType<NUdf::TJson>::Id:
- case NUdf::TDataType<NUdf::TDyNumber>::Id:
- case NUdf::TDataType<NUdf::TUuid>::Id: {
- if (isTableFormat) {
- auto nextString = ReadNextString(cmd, buf);
- return NUdf::TUnboxedValue(MakeString(NUdf::TStringRef(nextString)));
- }
- return ReadYsonStringInResultFormat(cmd, buf);
- }
- case NUdf::TDataType<NUdf::TDecimal>::Id: {
- auto nextString = ReadNextString(cmd, buf);
- if (isTableFormat) {
- if (nativeYtTypeFlags & NTCF_DECIMAL) {
- auto const params = static_cast<TDataDecimalType*>(type)->GetParams();
- if (params.first < 10) {
- // The YQL format differs from the YT format in the inf/nan values. NDecimal::FromYtDecimal converts nan/inf
- NDecimal::TInt128 res = NDecimal::FromYtDecimal(NYT::NDecimal::TDecimal::ParseBinary32(params.first, nextString));
- YQL_ENSURE(!NDecimal::IsError(res));
- return NUdf::TUnboxedValuePod(res);
- } else if (params.first < 19) {
- NDecimal::TInt128 res = NDecimal::FromYtDecimal(NYT::NDecimal::TDecimal::ParseBinary64(params.first, nextString));
- YQL_ENSURE(!NDecimal::IsError(res));
- return NUdf::TUnboxedValuePod(res);
- } else {
- YQL_ENSURE(params.first < 36);
- NYT::NDecimal::TDecimal::TValue128 tmpRes = NYT::NDecimal::TDecimal::ParseBinary128(params.first, nextString);
- NDecimal::TInt128 res;
- static_assert(sizeof(NDecimal::TInt128) == sizeof(NYT::NDecimal::TDecimal::TValue128));
- memcpy(&res, &tmpRes, sizeof(NDecimal::TInt128));
- res = NDecimal::FromYtDecimal(res);
- YQL_ENSURE(!NDecimal::IsError(res));
- return NUdf::TUnboxedValuePod(res);
- }
- }
- else {
- const auto& des = NDecimal::Deserialize(nextString.data(), nextString.size());
- YQL_ENSURE(!NDecimal::IsError(des.first));
- YQL_ENSURE(nextString.size() == des.second);
- return NUdf::TUnboxedValuePod(des.first);
- }
- } else {
- const auto params = static_cast<TDataDecimalType*>(type)->GetParams();
- const auto val = NDecimal::FromString(nextString, params.first, params.second);
- YQL_ENSURE(!NDecimal::IsError(val));
- return NUdf::TUnboxedValuePod(val);
- }
- }
- case NUdf::TDataType<NUdf::TYson>::Id: {
- auto& yson = buf.YsonBuffer();
- yson.clear();
- CopyYsonWithAttrs(cmd, buf, yson);
- if (isTableFormat) {
- return NUdf::TUnboxedValue(MakeString(NUdf::TStringRef(yson)));
- }
- TString decodedYson = NResult::DecodeRestrictedYson(TStringBuf(yson.data(), yson.size()), NYson::EYsonFormat::Text);
- return NUdf::TUnboxedValue(MakeString(NUdf::TStringRef(decodedYson)));
- }
- case NUdf::TDataType<NUdf::TDate>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Uint64Marker);
- return NUdf::TUnboxedValuePod((ui16)buf.ReadVarUI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<ui16>(cmd, buf));
- case NUdf::TDataType<NUdf::TDatetime>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Uint64Marker);
- return NUdf::TUnboxedValuePod((ui32)buf.ReadVarUI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<ui32>(cmd, buf));
- case NUdf::TDataType<NUdf::TTimestamp>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Uint64Marker);
- return NUdf::TUnboxedValuePod(buf.ReadVarUI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<ui64>(cmd, buf));
- case NUdf::TDataType<NUdf::TInterval>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Int64Marker);
- return NUdf::TUnboxedValuePod(buf.ReadVarI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<i64>(cmd, buf));
- case NUdf::TDataType<NUdf::TTzDate>::Id: {
- auto nextString = ReadNextString(cmd, buf);
- NUdf::TUnboxedValuePod data;
- if (isTableFormat) {
- ui16 value;
- ui16 tzId = 0;
- YQL_ENSURE(DeserializeTzDate(nextString, value, tzId));
- data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- } else {
- data = ValueFromString(NUdf::EDataSlot::TzDate, nextString);
- YQL_ENSURE(data, "incorrect tz date format for value " << nextString);
- }
- return data;
- }
- case NUdf::TDataType<NUdf::TTzDatetime>::Id: {
- auto nextString = ReadNextString(cmd, buf);
- NUdf::TUnboxedValuePod data;
- if (isTableFormat) {
- ui32 value;
- ui16 tzId = 0;
- YQL_ENSURE(DeserializeTzDatetime(nextString, value, tzId));
- data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- } else {
- data = ValueFromString(NUdf::EDataSlot::TzDatetime, nextString);
- YQL_ENSURE(data, "incorrect tz datetime format for value " << nextString);
- }
- return data;
- }
- case NUdf::TDataType<NUdf::TTzTimestamp>::Id: {
- auto nextString = ReadNextString(cmd, buf);
- NUdf::TUnboxedValuePod data;
- if (isTableFormat) {
- ui64 value;
- ui16 tzId = 0;
- YQL_ENSURE(DeserializeTzTimestamp(nextString, value, tzId));
- data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- } else {
- data = ValueFromString(NUdf::EDataSlot::TzTimestamp, nextString);
- YQL_ENSURE(data, "incorrect tz timestamp format for value " << nextString);
- }
- return data;
- }
- case NUdf::TDataType<NUdf::TDate32>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Int64Marker);
- return NUdf::TUnboxedValuePod((i32)buf.ReadVarI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<i32>(cmd, buf));
- case NUdf::TDataType<NUdf::TDatetime64>::Id:
- case NUdf::TDataType<NUdf::TTimestamp64>::Id:
- case NUdf::TDataType<NUdf::TInterval64>::Id:
- if (isTableFormat) {
- CHECK_EXPECTED(cmd, Int64Marker);
- return NUdf::TUnboxedValuePod(buf.ReadVarI64());
- }
- return NUdf::TUnboxedValuePod(ReadNextSerializedNumber<i64>(cmd, buf));
- case NUdf::TDataType<NUdf::TJsonDocument>::Id: {
- if (isTableFormat) {
- return ValueFromString(EDataSlot::JsonDocument, ReadNextString(cmd, buf));
- }
- const auto json = ReadYsonStringInResultFormat(cmd, buf);
- return ValueFromString(EDataSlot::JsonDocument, json.AsStringRef());
- }
- case NUdf::TDataType<NUdf::TTzDate32>::Id: {
- auto nextString = ReadNextString(cmd, buf);
- NUdf::TUnboxedValuePod data;
- if (isTableFormat) {
- i32 value;
- ui16 tzId = 0;
- YQL_ENSURE(DeserializeTzDate32(nextString, value, tzId));
- data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- } else {
- data = ValueFromString(NUdf::EDataSlot::TzDate32, nextString);
- YQL_ENSURE(data, "incorrect tz date format for value " << nextString);
- }
- return data;
- }
- case NUdf::TDataType<NUdf::TTzDatetime64>::Id: {
- auto nextString = ReadNextString(cmd, buf);
- NUdf::TUnboxedValuePod data;
- if (isTableFormat) {
- i64 value;
- ui16 tzId = 0;
- YQL_ENSURE(DeserializeTzDatetime64(nextString, value, tzId));
- data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- } else {
- data = ValueFromString(NUdf::EDataSlot::TzDatetime64, nextString);
- YQL_ENSURE(data, "incorrect tz datetime format for value " << nextString);
- }
- return data;
- }
- case NUdf::TDataType<NUdf::TTzTimestamp64>::Id: {
- auto nextString = ReadNextString(cmd, buf);
- NUdf::TUnboxedValuePod data;
- if (isTableFormat) {
- i64 value;
- ui16 tzId = 0;
- YQL_ENSURE(DeserializeTzTimestamp64(nextString, value, tzId));
- data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- } else {
- data = ValueFromString(NUdf::EDataSlot::TzTimestamp64, nextString);
- YQL_ENSURE(data, "incorrect tz timestamp format for value " << nextString);
- }
- return data;
- }
- default:
- YQL_ENSURE(false, "Unsupported data type: " << schemeType);
- }
- }
- case TType::EKind::Struct: {
- YQL_ENSURE(cmd == BeginListSymbol || cmd == BeginMapSymbol);
- auto structType = static_cast<TStructType*>(type);
- NUdf::TUnboxedValue* items;
- NUdf::TUnboxedValue ret = holderFactory.CreateDirectArrayHolder(structType->GetMembersCount(), items);
- if (cmd == BeginListSymbol) {
- cmd = buf.Read();
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- items[i] = ReadYsonValue(structType->GetMemberType(i), nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- return ret;
- } else {
- cmd = buf.Read();
- for (;;) {
- if (cmd == EndMapSymbol) {
- break;
- }
- auto keyBuffer = ReadNextString(cmd, buf);
- auto pos = structType->FindMemberIndex(keyBuffer);
- EXPECTED(buf, KeyValueSeparatorSymbol);
- cmd = buf.Read();
- if (pos && cmd != '#') {
- auto memberType = structType->GetMemberType(*pos);
- auto unwrappedType = memberType;
- if (!(nativeYtTypeFlags & ENativeTypeCompatFlags::NTCF_COMPLEX) && isTableFormat && unwrappedType->IsOptional()) {
- unwrappedType = static_cast<TOptionalType*>(unwrappedType)->GetItemType();
- }
- items[*pos] = ReadYsonValue(unwrappedType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- } else {
- SkipYson(cmd, buf);
- }
- cmd = buf.Read();
- if (cmd == KeyedItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- }
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- if (items[i]) {
- continue;
- }
- YQL_ENSURE(structType->GetMemberType(i)->IsOptional(), "Missing required field: " << structType->GetMemberName(i));
- }
- return ret;
- }
- }
- case TType::EKind::List: {
- auto itemType = static_cast<TListType*>(type)->GetItemType();
- TDefaultListRepresentation items;
- CHECK_EXPECTED(cmd, BeginListSymbol);
- cmd = buf.Read();
- for (;;) {
- if (cmd == EndListSymbol) {
- break;
- }
- items = items.Append(ReadYsonValue(itemType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat));
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- }
- return holderFactory.CreateDirectListHolder(std::move(items));
- }
- case TType::EKind::Optional: {
- if (cmd == EntitySymbol) {
- return NUdf::TUnboxedValuePod();
- }
- auto itemType = static_cast<TOptionalType*>(type)->GetItemType();
- if (isTableFormat && (nativeYtTypeFlags & ENativeTypeCompatFlags::NTCF_COMPLEX)) {
- if (itemType->GetKind() == TType::EKind::Optional || itemType->GetKind() == TType::EKind::Pg) {
- CHECK_EXPECTED(cmd, BeginListSymbol);
- cmd = buf.Read();
- auto value = ReadYsonValue(itemType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- return value.Release().MakeOptional();
- } else {
- return ReadYsonValue(itemType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat).Release().MakeOptional();
- }
- } else {
- if (cmd != BeginListSymbol) {
- auto value = ReadYsonValue(itemType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- return value.Release().MakeOptional();
- }
- cmd = buf.Read();
- if (cmd == EndListSymbol) {
- return NUdf::TUnboxedValuePod();
- }
- auto value = ReadYsonValue(itemType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- return value.Release().MakeOptional();
- }
- }
- case TType::EKind::Dict: {
- auto dictType = static_cast<TDictType*>(type);
- auto keyType = dictType->GetKeyType();
- auto payloadType = dictType->GetPayloadType();
- TKeyTypes types;
- bool isTuple;
- bool encoded;
- bool useIHash;
- GetDictionaryKeyTypes(keyType, types, isTuple, encoded, useIHash);
- TMaybe<TValuePacker> packer;
- if (encoded) {
- packer.ConstructInPlace(true, keyType);
- }
- YQL_ENSURE(cmd == BeginListSymbol || cmd == BeginMapSymbol, "Expected '{' or '[', but read: " << TString(cmd).Quote());
- if (cmd == BeginMapSymbol) {
- bool unusedIsOptional;
- auto unpackedType = UnpackOptional(keyType, unusedIsOptional);
- YQL_ENSURE(unpackedType->IsData() &&
- (static_cast<TDataType*>(unpackedType)->GetSchemeType() == NUdf::TDataType<char*>::Id ||
- static_cast<TDataType*>(unpackedType)->GetSchemeType() == NUdf::TDataType<NUdf::TUtf8>::Id),
- "Expected String or Utf8 type as dictionary key type");
- auto filler = [&](TValuesDictHashMap& map) {
- cmd = buf.Read();
- for (;;) {
- if (cmd == EndMapSymbol) {
- break;
- }
- auto keyBuffer = ReadNextString(cmd, buf);
- auto keyStr = NUdf::TUnboxedValue(MakeString(keyBuffer));
- EXPECTED(buf, KeyValueSeparatorSymbol);
- cmd = buf.Read();
- auto payload = ReadYsonValue(payloadType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- map.emplace(std::move(keyStr), std::move(payload));
- cmd = buf.Read();
- if (cmd == KeyedItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- }
- };
- const NUdf::IHash* hash = holderFactory.GetHash(*keyType, useIHash);
- const NUdf::IEquate* equate = holderFactory.GetEquate(*keyType, useIHash);
- return holderFactory.CreateDirectHashedDictHolder(filler, types, isTuple, true, nullptr, hash, equate);
- }
- else {
- auto filler = [&](TValuesDictHashMap& map) {
- cmd = buf.Read();
- for (;;) {
- if (cmd == EndListSymbol) {
- break;
- }
- CHECK_EXPECTED(cmd, BeginListSymbol);
- cmd = buf.Read();
- auto key = ReadYsonValue(keyType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- EXPECTED(buf, ListItemSeparatorSymbol);
- cmd = buf.Read();
- auto payload = ReadYsonValue(payloadType, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- if (packer) {
- key = MakeString(packer->Pack(key));
- }
- map.emplace(std::move(key), std::move(payload));
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- }
- };
- const NUdf::IHash* hash = holderFactory.GetHash(*keyType, useIHash);
- const NUdf::IEquate* equate = holderFactory.GetEquate(*keyType, useIHash);
- return holderFactory.CreateDirectHashedDictHolder(filler, types, isTuple, true, encoded ? keyType : nullptr,
- hash, equate);
- }
- }
- case TType::EKind::Tuple: {
- auto tupleType = static_cast<TTupleType*>(type);
- NUdf::TUnboxedValue* items;
- NUdf::TUnboxedValue ret = holderFactory.CreateDirectArrayHolder(tupleType->GetElementsCount(), items);
- CHECK_EXPECTED(cmd, BeginListSymbol);
- cmd = buf.Read();
- for (ui32 i = 0; i < tupleType->GetElementsCount(); ++i) {
- items[i] = ReadYsonValue(tupleType->GetElementType(i), nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- cmd = buf.Read();
- if (cmd == ListItemSeparatorSymbol) {
- cmd = buf.Read();
- }
- }
- CHECK_EXPECTED(cmd, EndListSymbol);
- return ret;
- }
- case TType::EKind::Void: {
- if (cmd == EntitySymbol) {
- return NUdf::TUnboxedValuePod::Void();
- }
- auto nextString = ReadNextString(cmd, buf);
- YQL_ENSURE(nextString == NResult::TYsonResultWriter::VoidString, "Expected Void");
- return NUdf::TUnboxedValuePod::Void();
- }
- case TType::EKind::Null: {
- CHECK_EXPECTED(cmd, EntitySymbol);
- return NUdf::TUnboxedValuePod();
- }
- case TType::EKind::EmptyList: {
- CHECK_EXPECTED(cmd, BeginListSymbol);
- cmd = buf.Read();
- CHECK_EXPECTED(cmd, EndListSymbol);
- return holderFactory.GetEmptyContainerLazy();
- }
- case TType::EKind::EmptyDict: {
- YQL_ENSURE(cmd == BeginListSymbol || cmd == BeginMapSymbol, "Expected '{' or '[', but read: " << TString(cmd).Quote());
- if (cmd == BeginListSymbol) {
- cmd = buf.Read();
- CHECK_EXPECTED(cmd, EndListSymbol);
- } else {
- cmd = buf.Read();
- CHECK_EXPECTED(cmd, EndMapSymbol);
- }
- return holderFactory.GetEmptyContainerLazy();
- }
- case TType::EKind::Pg: {
- auto pgType = static_cast<TPgType*>(type);
- return isTableFormat ? ReadYsonValueInTableFormatPg(pgType, cmd, buf) : ReadYsonValuePg(pgType, cmd, buf);
- }
- case TType::EKind::Tagged: {
- auto taggedType = static_cast<TTaggedType*>(type);
- return ReadYsonValue(taggedType->GetBaseType(), nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- }
- default:
- YQL_ENSURE(false, "Unsupported type: " << type->GetKindAsStr());
- }
- }
- TMaybe<NUdf::TUnboxedValue> ParseYsonValue(const THolderFactory& holderFactory,
- const TStringBuf& yson, TType* type, ui64 nativeYtTypeFlags, IOutputStream* err, bool isTableFormat) {
- try {
- class TReader : public IBlockReader {
- public:
- TReader(const TStringBuf& yson)
- : Yson_(yson)
- {}
- void SetDeadline(TInstant deadline) override {
- Y_UNUSED(deadline);
- }
- std::pair<const char*, const char*> NextFilledBlock() override {
- if (FirstBuffer_) {
- FirstBuffer_ = false;
- return{ Yson_.begin(), Yson_.end() };
- }
- else {
- return{ nullptr, nullptr };
- }
- }
- void ReturnBlock() override {
- }
- bool Retry(const TMaybe<ui32>& rangeIndex, const TMaybe<ui64>& rowIndex, const std::exception_ptr& error) override {
- Y_UNUSED(rangeIndex);
- Y_UNUSED(rowIndex);
- Y_UNUSED(error);
- return false;
- }
- private:
- TStringBuf Yson_;
- bool FirstBuffer_ = true;
- };
- TReader reader(yson);
- TInputBuf buf(reader, nullptr);
- char cmd = buf.Read();
- return ReadYsonValue(type, nativeYtTypeFlags, holderFactory, cmd, buf, isTableFormat);
- }
- catch (const yexception& e) {
- if (err) {
- *err << "YSON parsing failed: " << e.what();
- }
- return Nothing();
- }
- }
- TMaybe<NUdf::TUnboxedValue> ParseYsonNode(const THolderFactory& holderFactory,
- const NYT::TNode& node, TType* type, ui64 nativeYtTypeFlags, IOutputStream* err) {
- return ParseYsonValue(holderFactory, NYT::NodeToYsonString(node, NYson::EYsonFormat::Binary), type, nativeYtTypeFlags, err, true);
- }
- TMaybe<NUdf::TUnboxedValue> ParseYsonNodeInResultFormat(const THolderFactory& holderFactory,
- const NYT::TNode& node, TType* type, IOutputStream* err) {
- return ParseYsonValue(holderFactory, NYT::NodeToYsonString(node, NYson::EYsonFormat::Binary), type, 0, err, false);
- }
- extern "C" void ReadYsonContainerValue(TType* type, ui64 nativeYtTypeFlags, const NKikimr::NMiniKQL::THolderFactory& holderFactory,
- NUdf::TUnboxedValue& value, TInputBuf& buf, bool wrapOptional) {
- // yson content
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- // parse binary yson...
- YQL_ENSURE(size > 0);
- char cmd = buf.Read();
- auto tmp = ReadYsonValue(type, nativeYtTypeFlags, holderFactory, cmd, buf, true);
- if (!wrapOptional) {
- value = std::move(tmp);
- }
- else {
- value = tmp.Release().MakeOptional();
- }
- }
- NUdf::TUnboxedValue ReadSkiffData(TType* type, ui64 nativeYtTypeFlags, TInputBuf& buf) {
- auto schemeType = static_cast<TDataType*>(type)->GetSchemeType();
- switch (schemeType) {
- case NUdf::TDataType<bool>::Id: {
- ui8 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(data != 0);
- }
- case NUdf::TDataType<ui8>::Id: {
- ui64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(ui8(data));
- }
- case NUdf::TDataType<i8>::Id: {
- i64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(i8(data));
- }
- case NUdf::TDataType<NUdf::TDate>::Id:
- case NUdf::TDataType<ui16>::Id: {
- ui64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(ui16(data));
- }
- case NUdf::TDataType<i16>::Id: {
- i64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(i16(data));
- }
- case NUdf::TDataType<NUdf::TDate32>::Id:
- case NUdf::TDataType<i32>::Id: {
- i64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(i32(data));
- }
- case NUdf::TDataType<NUdf::TDatetime>::Id:
- case NUdf::TDataType<ui32>::Id: {
- ui64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(ui32(data));
- }
- case NUdf::TDataType<NUdf::TInterval>::Id:
- case NUdf::TDataType<NUdf::TInterval64>::Id:
- case NUdf::TDataType<NUdf::TDatetime64>::Id:
- case NUdf::TDataType<NUdf::TTimestamp64>::Id:
- case NUdf::TDataType<i64>::Id: {
- i64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(data);
- }
- case NUdf::TDataType<NUdf::TTimestamp>::Id:
- case NUdf::TDataType<ui64>::Id: {
- ui64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(data);
- }
- case NUdf::TDataType<float>::Id: {
- double data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(float(data));
- }
- case NUdf::TDataType<double>::Id: {
- double data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(data);
- }
- case NUdf::TDataType<NUdf::TUtf8>::Id:
- case NUdf::TDataType<char*>::Id:
- case NUdf::TDataType<NUdf::TJson>::Id:
- case NUdf::TDataType<NUdf::TYson>::Id:
- case NUdf::TDataType<NUdf::TDyNumber>::Id:
- case NUdf::TDataType<NUdf::TUuid>::Id: {
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- auto str = NUdf::TUnboxedValue(MakeStringNotFilled(size));
- buf.ReadMany(str.AsStringRef().Data(), size);
- return str;
- }
- case NUdf::TDataType<NUdf::TDecimal>::Id: {
- if (nativeYtTypeFlags & NTCF_DECIMAL) {
- auto const params = static_cast<TDataDecimalType*>(type)->GetParams();
- if (params.first < 10) {
- i32 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(NDecimal::FromYtDecimal(data));
- } else if (params.first < 19) {
- i64 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(NDecimal::FromYtDecimal(data));
- } else {
- YQL_ENSURE(params.first < 36);
- NDecimal::TInt128 data;
- buf.ReadMany((char*)&data, sizeof(data));
- return NUdf::TUnboxedValuePod(NDecimal::FromYtDecimal(data));
- }
- } else {
- ui32 size;
- buf.ReadMany(reinterpret_cast<char*>(&size), sizeof(size));
- const auto maxSize = sizeof(NDecimal::TInt128);
- YQL_ENSURE(size > 0U && size <= maxSize, "Bad decimal field size: " << size);
- char data[maxSize];
- buf.ReadMany(data, size);
- const auto& v = NDecimal::Deserialize(data, size);
- YQL_ENSURE(!NDecimal::IsError(v.first), "Bad decimal field data: " << data);
- YQL_ENSURE(size == v.second, "Bad decimal field size: " << size);
- return NUdf::TUnboxedValuePod(v.first);
- }
- }
- case NUdf::TDataType<NUdf::TTzDate>::Id: {
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- auto& vec = buf.YsonBuffer();
- vec.resize(size);
- buf.ReadMany(vec.data(), size);
- ui16 value;
- ui16 tzId;
- YQL_ENSURE(DeserializeTzDate(TStringBuf(vec.begin(), vec.end()), value, tzId));
- auto data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- return data;
- }
- case NUdf::TDataType<NUdf::TTzDatetime>::Id: {
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- auto& vec = buf.YsonBuffer();
- vec.resize(size);
- buf.ReadMany(vec.data(), size);
- ui32 value;
- ui16 tzId;
- YQL_ENSURE(DeserializeTzDatetime(TStringBuf(vec.begin(), vec.end()), value, tzId));
- auto data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- return data;
- }
- case NUdf::TDataType<NUdf::TTzTimestamp>::Id: {
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- auto& vec = buf.YsonBuffer();
- vec.resize(size);
- buf.ReadMany(vec.data(), size);
- ui64 value;
- ui16 tzId;
- YQL_ENSURE(DeserializeTzTimestamp(TStringBuf(vec.begin(), vec.end()), value, tzId));
- auto data = NUdf::TUnboxedValuePod(value);
- data.SetTimezoneId(tzId);
- return data;
- }
- case NUdf::TDataType<NUdf::TJsonDocument>::Id: {
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- auto json = NUdf::TUnboxedValue(MakeStringNotFilled(size));
- buf.ReadMany(json.AsStringRef().Data(), size);
- return ValueFromString(EDataSlot::JsonDocument, json.AsStringRef());
- }
- default:
- YQL_ENSURE(false, "Unsupported data type: " << schemeType);
- }
- }
- void SkipSkiffField(NKikimr::NMiniKQL::TType* type, ui64 nativeYtTypeFlags, TInputBuf& buf) {
- const bool isOptional = type->IsOptional();
- if (isOptional) {
- // Unwrap optional
- type = static_cast<TOptionalType*>(type)->GetItemType();
- }
- if (isOptional) {
- auto marker = buf.Read();
- if (!marker) {
- return;
- }
- }
- if (type->IsData()) {
- auto schemeType = static_cast<TDataType*>(type)->GetSchemeType();
- switch (schemeType) {
- case NUdf::TDataType<bool>::Id:
- buf.SkipMany(sizeof(ui8));
- break;
- case NUdf::TDataType<ui8>::Id:
- case NUdf::TDataType<ui16>::Id:
- case NUdf::TDataType<ui32>::Id:
- case NUdf::TDataType<ui64>::Id:
- case NUdf::TDataType<NUdf::TDate>::Id:
- case NUdf::TDataType<NUdf::TDatetime>::Id:
- case NUdf::TDataType<NUdf::TTimestamp>::Id:
- buf.SkipMany(sizeof(ui64));
- break;
- case NUdf::TDataType<i8>::Id:
- case NUdf::TDataType<i16>::Id:
- case NUdf::TDataType<i32>::Id:
- case NUdf::TDataType<i64>::Id:
- case NUdf::TDataType<NUdf::TInterval>::Id:
- case NUdf::TDataType<NUdf::TDate32>::Id:
- case NUdf::TDataType<NUdf::TDatetime64>::Id:
- case NUdf::TDataType<NUdf::TTimestamp64>::Id:
- case NUdf::TDataType<NUdf::TInterval64>::Id:
- buf.SkipMany(sizeof(i64));
- break;
- case NUdf::TDataType<float>::Id:
- case NUdf::TDataType<double>::Id:
- buf.SkipMany(sizeof(double));
- break;
- case NUdf::TDataType<NUdf::TUtf8>::Id:
- case NUdf::TDataType<char*>::Id:
- case NUdf::TDataType<NUdf::TJson>::Id:
- case NUdf::TDataType<NUdf::TYson>::Id:
- case NUdf::TDataType<NUdf::TUuid>::Id:
- case NUdf::TDataType<NUdf::TDyNumber>::Id:
- case NUdf::TDataType<NUdf::TTzDate>::Id:
- case NUdf::TDataType<NUdf::TTzDatetime>::Id:
- case NUdf::TDataType<NUdf::TTzTimestamp>::Id:
- case NUdf::TDataType<NUdf::TJsonDocument>::Id: {
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- buf.SkipMany(size);
- break;
- }
- case NUdf::TDataType<NUdf::TDecimal>::Id: {
- if (nativeYtTypeFlags & NTCF_DECIMAL) {
- auto const params = static_cast<TDataDecimalType*>(type)->GetParams();
- if (params.first < 10) {
- buf.SkipMany(sizeof(i32));
- } else if (params.first < 19) {
- buf.SkipMany(sizeof(i64));
- } else {
- buf.SkipMany(sizeof(NDecimal::TInt128));
- }
- } else {
- ui32 size;
- buf.ReadMany((char*)&size, sizeof(size));
- CHECK_STRING_LENGTH_UNSIGNED(size);
- buf.SkipMany(size);
- }
- break;
- }
- default:
- YQL_ENSURE(false, "Unsupported data type: " << schemeType);
- }
- return;
- }
- if (type->IsPg()) {
- SkipSkiffPg(static_cast<TPgType*>(type), buf);
- return;
- }
- if (type->IsStruct()) {
- auto structType = static_cast<TStructType*>(type);
- const std::vector<size_t>* reorder = nullptr;
- if (auto cookie = structType->GetCookie()) {
- reorder = ((const std::vector<size_t>*)cookie);
- }
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- SkipSkiffField(structType->GetMemberType(reorder ? reorder->at(i) : i), nativeYtTypeFlags, buf);
- }
- return;
- }
- if (type->IsList()) {
- auto itemType = static_cast<TListType*>(type)->GetItemType();
- while (buf.Read() == '\0') {
- SkipSkiffField(itemType, nativeYtTypeFlags, buf);
- }
- return;
- }
- if (type->IsTuple()) {
- auto tupleType = static_cast<TTupleType*>(type);
- for (ui32 i = 0; i < tupleType->GetElementsCount(); ++i) {
- SkipSkiffField(tupleType->GetElementType(i), nativeYtTypeFlags, buf);
- }
- return;
- }
- if (type->IsVariant()) {
- auto varType = AS_TYPE(TVariantType, type);
- ui16 data = 0;
- if (varType->GetAlternativesCount() < 256) {
- buf.ReadMany((char*)&data, 1);
- } else {
- buf.ReadMany((char*)&data, sizeof(data));
- }
-
- if (varType->GetUnderlyingType()->IsTuple()) {
- auto tupleType = AS_TYPE(TTupleType, varType->GetUnderlyingType());
- YQL_ENSURE(data < tupleType->GetElementsCount());
- SkipSkiffField(tupleType->GetElementType(data), nativeYtTypeFlags, buf);
- } else {
- auto structType = AS_TYPE(TStructType, varType->GetUnderlyingType());
- if (auto cookie = structType->GetCookie()) {
- const std::vector<size_t>& reorder = *((const std::vector<size_t>*)cookie);
- data = reorder[data];
- }
- YQL_ENSURE(data < structType->GetMembersCount());
- SkipSkiffField(structType->GetMemberType(data), nativeYtTypeFlags, buf);
- }
- return;
- }
- if (type->IsVoid()) {
- return;
- }
- if (type->IsNull()) {
- return;
- }
- if (type->IsEmptyList() || type->IsEmptyDict()) {
- return;
- }
- if (type->IsDict()) {
- auto dictType = AS_TYPE(TDictType, type);
- auto keyType = dictType->GetKeyType();
- auto payloadType = dictType->GetPayloadType();
- while (buf.Read() == '\0') {
- SkipSkiffField(keyType, nativeYtTypeFlags, buf);
- SkipSkiffField(payloadType, nativeYtTypeFlags, buf);
- }
- return;
- }
- YQL_ENSURE(false, "Unsupported type for skip: " << type->GetKindAsStr());
- }
- NKikimr::NUdf::TUnboxedValue ReadSkiffNativeYtValue(NKikimr::NMiniKQL::TType* type, ui64 nativeYtTypeFlags,
- const NKikimr::NMiniKQL::THolderFactory& holderFactory, TInputBuf& buf)
- {
- if (type->IsData()) {
- return ReadSkiffData(type, nativeYtTypeFlags, buf);
- }
- if (type->IsPg()) {
- return ReadSkiffPg(static_cast<TPgType*>(type), buf);
- }
- if (type->IsOptional()) {
- auto marker = buf.Read();
- if (!marker) {
- return NUdf::TUnboxedValue();
- }
- auto value = ReadSkiffNativeYtValue(AS_TYPE(TOptionalType, type)->GetItemType(), nativeYtTypeFlags, holderFactory, buf);
- return value.Release().MakeOptional();
- }
- if (type->IsTuple()) {
- auto tupleType = AS_TYPE(TTupleType, type);
- NUdf::TUnboxedValue* items;
- auto value = holderFactory.CreateDirectArrayHolder(tupleType->GetElementsCount(), items);
- for (ui32 i = 0; i < tupleType->GetElementsCount(); ++i) {
- items[i] = ReadSkiffNativeYtValue(tupleType->GetElementType(i), nativeYtTypeFlags, holderFactory, buf);
- }
- return value;
- }
- if (type->IsStruct()) {
- auto structType = AS_TYPE(TStructType, type);
- NUdf::TUnboxedValue* items;
- auto value = holderFactory.CreateDirectArrayHolder(structType->GetMembersCount(), items);
- if (auto cookie = type->GetCookie()) {
- const std::vector<size_t>& reorder = *((const std::vector<size_t>*)cookie);
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- const auto ndx = reorder[i];
- items[ndx] = ReadSkiffNativeYtValue(structType->GetMemberType(ndx), nativeYtTypeFlags, holderFactory, buf);
- }
- } else {
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- items[i] = ReadSkiffNativeYtValue(structType->GetMemberType(i), nativeYtTypeFlags, holderFactory, buf);
- }
- }
- return value;
- }
- if (type->IsList()) {
- auto itemType = AS_TYPE(TListType, type)->GetItemType();
- TDefaultListRepresentation items;
- while (buf.Read() == '\0') {
- items = items.Append(ReadSkiffNativeYtValue(itemType, nativeYtTypeFlags, holderFactory, buf));
- }
- return holderFactory.CreateDirectListHolder(std::move(items));
- }
- if (type->IsVariant()) {
- auto varType = AS_TYPE(TVariantType, type);
- ui16 data = 0;
- if (varType->GetAlternativesCount() < 256) {
- buf.ReadMany((char*)&data, 1);
- } else {
- buf.ReadMany((char*)&data, sizeof(data));
- }
- if (varType->GetUnderlyingType()->IsTuple()) {
- auto tupleType = AS_TYPE(TTupleType, varType->GetUnderlyingType());
- YQL_ENSURE(data < tupleType->GetElementsCount());
- auto item = ReadSkiffNativeYtValue(tupleType->GetElementType(data), nativeYtTypeFlags, holderFactory, buf);
- return holderFactory.CreateVariantHolder(item.Release(), data);
- }
- else {
- auto structType = AS_TYPE(TStructType, varType->GetUnderlyingType());
- if (auto cookie = structType->GetCookie()) {
- const std::vector<size_t>& reorder = *((const std::vector<size_t>*)cookie);
- data = reorder[data];
- }
- YQL_ENSURE(data < structType->GetMembersCount());
- auto item = ReadSkiffNativeYtValue(structType->GetMemberType(data), nativeYtTypeFlags, holderFactory, buf);
- return holderFactory.CreateVariantHolder(item.Release(), data);
- }
- }
- if (type->IsVoid()) {
- return NUdf::TUnboxedValue::Zero();
- }
- if (type->IsNull()) {
- return NUdf::TUnboxedValue();
- }
- if (type->IsEmptyList() || type->IsEmptyDict()) {
- return holderFactory.GetEmptyContainerLazy();
- }
- if (type->IsDict()) {
- auto dictType = AS_TYPE(TDictType, type);
- auto keyType = dictType->GetKeyType();
- auto payloadType = dictType->GetPayloadType();
- auto builder = holderFactory.NewDict(dictType, NUdf::TDictFlags::EDictKind::Hashed);
- while (buf.Read() == '\0') {
- auto key = ReadSkiffNativeYtValue(keyType, nativeYtTypeFlags, holderFactory, buf);
- auto payload = ReadSkiffNativeYtValue(payloadType, nativeYtTypeFlags, holderFactory, buf);
- builder->Add(std::move(key), std::move(payload));
- }
- return builder->Build();
- }
- YQL_ENSURE(false, "Unsupported type: " << type->GetKindAsStr());
- }
- extern "C" void ReadContainerNativeYtValue(TType* type, ui64 nativeYtTypeFlags, const NKikimr::NMiniKQL::THolderFactory& holderFactory,
- NUdf::TUnboxedValue& value, TInputBuf& buf, bool wrapOptional) {
- auto tmp = ReadSkiffNativeYtValue(type, nativeYtTypeFlags, holderFactory, buf);
- if (!wrapOptional) {
- value = std::move(tmp);
- } else {
- value = tmp.Release().MakeOptional();
- }
- }
- ///////////////////////////////////////////
- //
- // Initial state first = last = &dummy
- //
- // +1 block first = &dummy, last = newPage, first.next = newPage, newPage.next= &dummy
- // +1 block first = &dummy, last = newPage2, first.next = newPage, newPage.next = newPage2, newPage2.next = &dummy
- //
- ///////////////////////////////////////////
- class TTempBlockWriter : public NCommon::IBlockWriter {
- public:
- TTempBlockWriter()
- : Pool_(*TlsAllocState)
- , Last_(&Dummy_)
- {
- Dummy_.Avail_ = 0;
- Dummy_.Next_ = &Dummy_;
- }
- ~TTempBlockWriter() {
- auto current = Dummy_.Next_; // skip dummy node
- while (current != &Dummy_) {
- auto next = current->Next_;
- Pool_.ReturnPage(current);
- current = next;
- }
- }
- void SetRecordBoundaryCallback(std::function<void()> callback) override {
- Y_UNUSED(callback);
- }
- void WriteBlocks(TOutputBuf& buf) const {
- auto current = Dummy_.Next_; // skip dummy node
- while (current != &Dummy_) {
- auto next = current->Next_;
- buf.WriteMany((const char*)(current + 1), current->Avail_);
- current = next;
- }
- }
- TTempBlockWriter(const TTempBlockWriter&) = delete;
- void operator=(const TTempBlockWriter&) = delete;
- std::pair<char*, char*> NextEmptyBlock() override {
- auto newPage = Pool_.GetPage();
- auto header = (TPageHeader*)newPage;
- header->Avail_ = 0;
- header->Next_ = &Dummy_;
- Last_->Next_ = header;
- Last_ = header;
- return std::make_pair((char*)(header + 1), (char*)newPage + TAlignedPagePool::POOL_PAGE_SIZE);
- }
- void ReturnBlock(size_t avail, std::optional<size_t> lastRecordBoundary) override {
- Y_UNUSED(lastRecordBoundary);
- YQL_ENSURE(avail <= TAlignedPagePool::POOL_PAGE_SIZE - sizeof(TPageHeader));
- Last_->Avail_ = avail;
- }
- void Finish() override {
- }
- private:
- struct TPageHeader {
- TPageHeader* Next_ = nullptr;
- ui32 Avail_ = 0;
- };
- NKikimr::TAlignedPagePool& Pool_;
- TPageHeader* Last_;
- TPageHeader Dummy_;
- };
- void WriteYsonValueInTableFormat(TOutputBuf& buf, TType* type, ui64 nativeYtTypeFlags, const NUdf::TUnboxedValuePod& value, bool topLevel) {
- // Table format, very compact
- switch (type->GetKind()) {
- case TType::EKind::Variant: {
- buf.Write(BeginListSymbol);
- auto varType = static_cast<TVariantType*>(type);
- auto underlyingType = varType->GetUnderlyingType();
- auto index = value.GetVariantIndex();
- YQL_ENSURE(index < varType->GetAlternativesCount(), "Bad variant alternative: " << index << ", only " << varType->GetAlternativesCount() << " are available");
- YQL_ENSURE(underlyingType->IsTuple() || underlyingType->IsStruct(), "Wrong underlying type");
- TType* itemType;
- if (underlyingType->IsTuple()) {
- itemType = static_cast<TTupleType*>(underlyingType)->GetElementType(index);
- }
- else {
- itemType = static_cast<TStructType*>(underlyingType)->GetMemberType(index);
- }
- if (!(nativeYtTypeFlags & NTCF_COMPLEX) || underlyingType->IsTuple()) {
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(index);
- } else {
- auto structType = static_cast<TStructType*>(underlyingType);
- auto varName = structType->GetMemberName(index);
- buf.Write(StringMarker);
- buf.WriteVarI32(varName.size());
- buf.WriteMany(varName);
- }
- buf.Write(ListItemSeparatorSymbol);
- WriteYsonValueInTableFormat(buf, itemType, nativeYtTypeFlags, value.GetVariantItem(), false);
- buf.Write(ListItemSeparatorSymbol);
- buf.Write(EndListSymbol);
- break;
- }
- case TType::EKind::Data: {
- auto schemeType = static_cast<TDataType*>(type)->GetSchemeType();
- switch (schemeType) {
- case NUdf::TDataType<bool>::Id: {
- buf.Write(value.Get<bool>() ? TrueMarker : FalseMarker);
- break;
- }
- case NUdf::TDataType<ui8>::Id:
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(value.Get<ui8>());
- break;
- case NUdf::TDataType<i8>::Id:
- buf.Write(Int64Marker);
- buf.WriteVarI64(value.Get<i8>());
- break;
- case NUdf::TDataType<ui16>::Id:
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(value.Get<ui16>());
- break;
- case NUdf::TDataType<i16>::Id:
- buf.Write(Int64Marker);
- buf.WriteVarI64(value.Get<i16>());
- break;
- case NUdf::TDataType<i32>::Id:
- buf.Write(Int64Marker);
- buf.WriteVarI64(value.Get<i32>());
- break;
- case NUdf::TDataType<ui32>::Id:
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(value.Get<ui32>());
- break;
- case NUdf::TDataType<i64>::Id:
- buf.Write(Int64Marker);
- buf.WriteVarI64(value.Get<i64>());
- break;
- case NUdf::TDataType<ui64>::Id:
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(value.Get<ui64>());
- break;
- case NUdf::TDataType<float>::Id: {
- buf.Write(DoubleMarker);
- double val = value.Get<float>();
- buf.WriteMany((const char*)&val, sizeof(val));
- break;
- }
- case NUdf::TDataType<double>::Id: {
- buf.Write(DoubleMarker);
- double val = value.Get<double>();
- buf.WriteMany((const char*)&val, sizeof(val));
- break;
- }
- case NUdf::TDataType<NUdf::TUtf8>::Id:
- case NUdf::TDataType<char*>::Id:
- case NUdf::TDataType<NUdf::TJson>::Id:
- case NUdf::TDataType<NUdf::TDyNumber>::Id:
- case NUdf::TDataType<NUdf::TUuid>::Id: {
- buf.Write(StringMarker);
- auto str = value.AsStringRef();
- buf.WriteVarI32(str.Size());
- buf.WriteMany(str);
- break;
- }
- case NUdf::TDataType<NUdf::TDecimal>::Id: {
- buf.Write(StringMarker);
- if (nativeYtTypeFlags & NTCF_DECIMAL){
- auto const params = static_cast<TDataDecimalType*>(type)->GetParams();
- const NDecimal::TInt128 data128 = value.GetInt128();
- char tmpBuf[NYT::NDecimal::TDecimal::MaxBinarySize];
- if (params.first < 10) {
- // The YQL format differs from the YT format in the inf/nan values. NDecimal::FromYtDecimal converts nan/inf
- TStringBuf resBuf = NYT::NDecimal::TDecimal::WriteBinary32(params.first, NDecimal::ToYtDecimal<i32>(data128), tmpBuf, NYT::NDecimal::TDecimal::MaxBinarySize);
- buf.WriteVarI32(resBuf.size());
- buf.WriteMany(resBuf.data(), resBuf.size());
- } else if (params.first < 19) {
- TStringBuf resBuf = NYT::NDecimal::TDecimal::WriteBinary64(params.first, NDecimal::ToYtDecimal<i64>(data128), tmpBuf, NYT::NDecimal::TDecimal::MaxBinarySize);
- buf.WriteVarI32(resBuf.size());
- buf.WriteMany(resBuf.data(), resBuf.size());
- } else {
- YQL_ENSURE(params.first < 36);
- NYT::NDecimal::TDecimal::TValue128 val;
- auto data128Converted = NDecimal::ToYtDecimal<NDecimal::TInt128>(data128);
- memcpy(&val, &data128Converted, sizeof(val));
- auto resBuf = NYT::NDecimal::TDecimal::WriteBinary128(params.first, val, tmpBuf, NYT::NDecimal::TDecimal::MaxBinarySize);
- buf.WriteVarI32(resBuf.size());
- buf.WriteMany(resBuf.data(), resBuf.size());
- }
- } else {
- char data[sizeof(NDecimal::TInt128)];
- const ui32 size = NDecimal::Serialize(value.GetInt128(), data);
- buf.WriteVarI32(size);
- buf.WriteMany(data, size);
- }
- break;
- }
- case NUdf::TDataType<NUdf::TYson>::Id: {
- // embed content
- buf.WriteMany(value.AsStringRef());
- break;
- }
- case NUdf::TDataType<NUdf::TDate>::Id:
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(value.Get<ui16>());
- break;
- case NUdf::TDataType<NUdf::TDatetime>::Id:
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(value.Get<ui32>());
- break;
- case NUdf::TDataType<NUdf::TTimestamp>::Id:
- buf.Write(Uint64Marker);
- buf.WriteVarUI64(value.Get<ui64>());
- break;
- case NUdf::TDataType<NUdf::TInterval>::Id:
- case NUdf::TDataType<NUdf::TInterval64>::Id:
- case NUdf::TDataType<NUdf::TDatetime64>::Id:
- case NUdf::TDataType<NUdf::TTimestamp64>::Id:
- buf.Write(Int64Marker);
- buf.WriteVarI64(value.Get<i64>());
- break;
- case NUdf::TDataType<NUdf::TDate32>::Id:
- buf.Write(Int64Marker);
- buf.WriteVarI64(value.Get<i32>());
- break;
- case NUdf::TDataType<NUdf::TTzDate>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui16 data = SwapBytes(value.Get<ui16>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.Write(StringMarker);
- buf.WriteVarI32(size);
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzDatetime>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui32 data = SwapBytes(value.Get<ui32>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.Write(StringMarker);
- buf.WriteVarI32(size);
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzTimestamp>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui64 data = SwapBytes(value.Get<ui64>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.Write(StringMarker);
- buf.WriteVarI32(size);
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzDate32>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui32 data = 0x80 ^ SwapBytes((ui32)value.Get<i32>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.Write(StringMarker);
- buf.WriteVarI32(size);
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzDatetime64>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui64 data = 0x80 ^ SwapBytes((ui64)value.Get<i64>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.Write(StringMarker);
- buf.WriteVarI32(size);
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzTimestamp64>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui64 data = 0x80 ^ SwapBytes((ui64)value.Get<i64>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.Write(StringMarker);
- buf.WriteVarI32(size);
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TJsonDocument>::Id: {
- buf.Write(StringMarker);
- NUdf::TUnboxedValue json = ValueToString(EDataSlot::JsonDocument, value);
- auto str = json.AsStringRef();
- buf.WriteVarI32(str.Size());
- buf.WriteMany(str);
- break;
- }
- default:
- YQL_ENSURE(false, "Unsupported data type: " << schemeType);
- }
- break;
- }
- case TType::EKind::Struct: {
- auto structType = static_cast<TStructType*>(type);
- if (nativeYtTypeFlags & ENativeTypeCompatFlags::NTCF_COMPLEX) {
- buf.Write(BeginMapSymbol);
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- buf.Write(StringMarker);
- auto key = structType->GetMemberName(i);
- buf.WriteVarI32(key.size());
- buf.WriteMany(key);
- buf.Write(KeyValueSeparatorSymbol);
- WriteYsonValueInTableFormat(buf, structType->GetMemberType(i), nativeYtTypeFlags, value.GetElement(i), false);
- buf.Write(KeyedItemSeparatorSymbol);
- }
- buf.Write(EndMapSymbol);
- } else {
- buf.Write(BeginListSymbol);
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- WriteYsonValueInTableFormat(buf, structType->GetMemberType(i), nativeYtTypeFlags, value.GetElement(i), false);
- buf.Write(ListItemSeparatorSymbol);
- }
- buf.Write(EndListSymbol);
- }
- break;
- }
- case TType::EKind::List: {
- auto itemType = static_cast<TListType*>(type)->GetItemType();
- const auto iter = value.GetListIterator();
- buf.Write(BeginListSymbol);
- for (NUdf::TUnboxedValue item; iter.Next(item); buf.Write(ListItemSeparatorSymbol)) {
- WriteYsonValueInTableFormat(buf, itemType, nativeYtTypeFlags, item, false);
- }
- buf.Write(EndListSymbol);
- break;
- }
- case TType::EKind::Optional: {
- auto itemType = static_cast<TOptionalType*>(type)->GetItemType();
- if (nativeYtTypeFlags & ENativeTypeCompatFlags::NTCF_COMPLEX) {
- if (value) {
- if (itemType->GetKind() == TType::EKind::Optional || itemType->GetKind() == TType::EKind::Pg) {
- buf.Write(BeginListSymbol);
- }
- WriteYsonValueInTableFormat(buf, itemType, nativeYtTypeFlags, value.GetOptionalValue(), false);
- if (itemType->GetKind() == TType::EKind::Optional || itemType->GetKind() == TType::EKind::Pg) {
- buf.Write(ListItemSeparatorSymbol);
- buf.Write(EndListSymbol);
- }
- } else {
- buf.Write(EntitySymbol);
- }
- } else {
- if (!value) {
- if (topLevel) {
- buf.Write(BeginListSymbol);
- buf.Write(EndListSymbol);
- }
- else {
- buf.Write(EntitySymbol);
- }
- }
- else {
- buf.Write(BeginListSymbol);
- WriteYsonValueInTableFormat(buf, itemType, nativeYtTypeFlags, value.GetOptionalValue(), false);
- buf.Write(ListItemSeparatorSymbol);
- buf.Write(EndListSymbol);
- }
- }
- break;
- }
- case TType::EKind::Dict: {
- auto dictType = static_cast<TDictType*>(type);
- const auto iter = value.GetDictIterator();
- buf.Write(BeginListSymbol);
- for (NUdf::TUnboxedValue key, payload; iter.NextPair(key, payload);) {
- buf.Write(BeginListSymbol);
- WriteYsonValueInTableFormat(buf, dictType->GetKeyType(), nativeYtTypeFlags, key, false);
- buf.Write(ListItemSeparatorSymbol);
- WriteYsonValueInTableFormat(buf, dictType->GetPayloadType(), nativeYtTypeFlags, payload, false);
- buf.Write(ListItemSeparatorSymbol);
- buf.Write(EndListSymbol);
- buf.Write(ListItemSeparatorSymbol);
- }
- buf.Write(EndListSymbol);
- break;
- }
- case TType::EKind::Tuple: {
- auto tupleType = static_cast<TTupleType*>(type);
- buf.Write(BeginListSymbol);
- for (ui32 i = 0; i < tupleType->GetElementsCount(); ++i) {
- WriteYsonValueInTableFormat(buf, tupleType->GetElementType(i), nativeYtTypeFlags, value.GetElement(i), false);
- buf.Write(ListItemSeparatorSymbol);
- }
- buf.Write(EndListSymbol);
- break;
- }
- case TType::EKind::Void: {
- buf.Write(EntitySymbol);
- break;
- }
- case TType::EKind::Null: {
- buf.Write(EntitySymbol);
- break;
- }
- case TType::EKind::EmptyList: {
- buf.Write(BeginListSymbol);
- buf.Write(EndListSymbol);
- break;
- }
- case TType::EKind::EmptyDict: {
- buf.Write(BeginListSymbol);
- buf.Write(EndListSymbol);
- break;
- }
- case TType::EKind::Pg: {
- auto pgType = static_cast<TPgType*>(type);
- WriteYsonValueInTableFormatPg(buf, pgType, value, topLevel);
- break;
- }
- default:
- YQL_ENSURE(false, "Unsupported type: " << type->GetKindAsStr());
- }
- }
- extern "C" void WriteYsonContainerValue(TType* type, ui64 nativeYtTypeFlags, const NUdf::TUnboxedValuePod& value, TOutputBuf& buf) {
- TTempBlockWriter blockWriter;
- TOutputBuf ysonBuf(blockWriter, nullptr);
- WriteYsonValueInTableFormat(ysonBuf, type, nativeYtTypeFlags, value, true);
- ysonBuf.Flush();
- ui32 size = ysonBuf.GetWrittenBytes();
- buf.WriteMany((const char*)&size, sizeof(size));
- blockWriter.WriteBlocks(buf);
- }
- extern "C" void WriteContainerNativeYtValue(TType* type, ui64 nativeYtTypeFlags, const NUdf::TUnboxedValuePod& value, TOutputBuf& buf) {
- WriteSkiffNativeYtValue(type, nativeYtTypeFlags, value, buf);
- }
- void WriteSkiffData(NKikimr::NMiniKQL::TType* type, ui64 nativeYtTypeFlags, const NKikimr::NUdf::TUnboxedValuePod& value, NCommon::TOutputBuf& buf) {
- auto schemeType = static_cast<TDataType*>(type)->GetSchemeType();
- switch (schemeType) {
- case NUdf::TDataType<bool>::Id: {
- ui8 data = value.Get<ui8>();
- buf.Write(data);
- break;
- }
- case NUdf::TDataType<ui8>::Id: {
- ui64 data = value.Get<ui8>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<i8>::Id: {
- i64 data = value.Get<i8>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<NUdf::TDate>::Id:
- case NUdf::TDataType<ui16>::Id: {
- ui64 data = value.Get<ui16>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<i16>::Id: {
- i64 data = value.Get<i16>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<NUdf::TDate32>::Id:
- case NUdf::TDataType<i32>::Id: {
- i64 data = value.Get<i32>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<NUdf::TDatetime>::Id:
- case NUdf::TDataType<ui32>::Id: {
- ui64 data = value.Get<ui32>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<NUdf::TInterval>::Id:
- case NUdf::TDataType<NUdf::TInterval64>::Id:
- case NUdf::TDataType<NUdf::TDatetime64>::Id:
- case NUdf::TDataType<NUdf::TTimestamp64>::Id:
- case NUdf::TDataType<i64>::Id: {
- i64 data = value.Get<i64>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<NUdf::TTimestamp>::Id:
- case NUdf::TDataType<ui64>::Id: {
- ui64 data = value.Get<ui64>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<float>::Id: {
- double data = value.Get<float>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<double>::Id: {
- double data = value.Get<double>();
- buf.WriteMany((const char*)&data, sizeof(data));
- break;
- }
- case NUdf::TDataType<NUdf::TUtf8>::Id:
- case NUdf::TDataType<char*>::Id:
- case NUdf::TDataType<NUdf::TJson>::Id:
- case NUdf::TDataType<NUdf::TYson>::Id:
- case NUdf::TDataType<NUdf::TDyNumber>::Id:
- case NUdf::TDataType<NUdf::TUuid>::Id: {
- auto str = value.AsStringRef();
- ui32 size = str.Size();
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany(str);
- break;
- }
- case NUdf::TDataType<NUdf::TDecimal>::Id: {
- if (nativeYtTypeFlags & NTCF_DECIMAL) {
- auto const params = static_cast<TDataDecimalType*>(type)->GetParams();
- const NDecimal::TInt128 data128 = value.GetInt128();
- if (params.first < 10) {
- auto data = NDecimal::ToYtDecimal<i32>(data128);
- buf.WriteMany((const char*)&data, sizeof(data));
- } else if (params.first < 19) {
- auto data = NDecimal::ToYtDecimal<i64>(data128);
- buf.WriteMany((const char*)&data, sizeof(data));
- } else {
- YQL_ENSURE(params.first < 36);
- auto data = NDecimal::ToYtDecimal<NDecimal::TInt128>(data128);
- buf.WriteMany((const char*)&data, sizeof(data));
- }
- } else {
- char data[sizeof(NDecimal::TInt128)];
- const ui32 size = NDecimal::Serialize(value.GetInt128(), data);
- buf.WriteMany(reinterpret_cast<const char*>(&size), sizeof(size));
- buf.WriteMany(data, size);
- }
- break;
- }
- case NUdf::TDataType<NUdf::TTzDate>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui16 data = SwapBytes(value.Get<ui16>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzDatetime>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui32 data = SwapBytes(value.Get<ui32>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzTimestamp>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui64 data = SwapBytes(value.Get<ui64>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzDate32>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui32 data = 0x80 ^ SwapBytes((ui32)value.Get<i32>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzDatetime64>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui64 data = 0x80 ^ SwapBytes((ui64)value.Get<i64>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TTzTimestamp64>::Id: {
- ui16 tzId = SwapBytes(value.GetTimezoneId());
- ui64 data = 0x80 ^ SwapBytes((ui64)value.Get<i64>());
- ui32 size = sizeof(data) + sizeof(tzId);
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany((const char*)&data, sizeof(data));
- buf.WriteMany((const char*)&tzId, sizeof(tzId));
- break;
- }
- case NUdf::TDataType<NUdf::TJsonDocument>::Id: {
- NUdf::TUnboxedValue json = ValueToString(EDataSlot::JsonDocument, value);
- auto str = json.AsStringRef();
- ui32 size = str.Size();
- buf.WriteMany((const char*)&size, sizeof(size));
- buf.WriteMany(str);
- break;
- }
- default:
- YQL_ENSURE(false, "Unsupported data type: " << schemeType);
- }
- }
- void WriteSkiffNativeYtValue(NKikimr::NMiniKQL::TType* type, ui64 nativeYtTypeFlags, const NKikimr::NUdf::TUnboxedValuePod& value, NCommon::TOutputBuf& buf) {
- if (type->IsData()) {
- WriteSkiffData(type, nativeYtTypeFlags, value, buf);
- } else if (type->IsPg()) {
- WriteSkiffPgValue(static_cast<TPgType*>(type), value, buf);
- } else if (type->IsOptional()) {
- if (!value) {
- buf.Write('\0');
- return;
- }
- buf.Write('\1');
- WriteSkiffNativeYtValue(AS_TYPE(TOptionalType, type)->GetItemType(), nativeYtTypeFlags, value.GetOptionalValue(), buf);
- } else if (type->IsList()) {
- auto itemType = AS_TYPE(TListType, type)->GetItemType();
- auto elements = value.GetElements();
- if (elements) {
- ui32 size = value.GetListLength();
- for (ui32 i = 0; i < size; ++i) {
- buf.Write('\0');
- WriteSkiffNativeYtValue(itemType, nativeYtTypeFlags, elements[i], buf);
- }
- } else {
- NUdf::TUnboxedValue item;
- for (auto iter = value.GetListIterator(); iter.Next(item); ) {
- buf.Write('\0');
- WriteSkiffNativeYtValue(itemType, nativeYtTypeFlags, item, buf);
- }
- }
- buf.Write('\xff');
- } else if (type->IsTuple()) {
- auto tupleType = AS_TYPE(TTupleType, type);
- auto elements = value.GetElements();
- if (elements) {
- for (ui32 i = 0; i < tupleType->GetElementsCount(); ++i) {
- WriteSkiffNativeYtValue(tupleType->GetElementType(i), nativeYtTypeFlags, elements[i], buf);
- }
- } else {
- for (ui32 i = 0; i < tupleType->GetElementsCount(); ++i) {
- WriteSkiffNativeYtValue(tupleType->GetElementType(i), nativeYtTypeFlags, value.GetElement(i), buf);
- }
- }
- } else if (type->IsStruct()) {
- auto structType = AS_TYPE(TStructType, type);
- auto elements = value.GetElements();
- if (auto cookie = type->GetCookie()) {
- const std::vector<size_t>& reorder = *((const std::vector<size_t>*)cookie);
- if (elements) {
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- const auto ndx = reorder[i];
- WriteSkiffNativeYtValue(structType->GetMemberType(ndx), nativeYtTypeFlags, elements[ndx], buf);
- }
- } else {
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- const auto ndx = reorder[i];
- WriteSkiffNativeYtValue(structType->GetMemberType(ndx), nativeYtTypeFlags, value.GetElement(ndx), buf);
- }
- }
- } else {
- if (elements) {
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- WriteSkiffNativeYtValue(structType->GetMemberType(i), nativeYtTypeFlags, elements[i], buf);
- }
- } else {
- for (ui32 i = 0; i < structType->GetMembersCount(); ++i) {
- WriteSkiffNativeYtValue(structType->GetMemberType(i), nativeYtTypeFlags, value.GetElement(i), buf);
- }
- }
- }
- } else if (type->IsVariant()) {
- auto varType = AS_TYPE(TVariantType, type);
- ui16 index = (ui16)value.GetVariantIndex();
- if (varType->GetAlternativesCount() < 256) {
- buf.WriteMany((const char*)&index, 1);
- } else {
- buf.WriteMany((const char*)&index, sizeof(index));
- }
- if (varType->GetUnderlyingType()->IsTuple()) {
- auto tupleType = AS_TYPE(TTupleType, varType->GetUnderlyingType());
- WriteSkiffNativeYtValue(tupleType->GetElementType(index), nativeYtTypeFlags, value.GetVariantItem(), buf);
- } else {
- auto structType = AS_TYPE(TStructType, varType->GetUnderlyingType());
- if (auto cookie = structType->GetCookie()) {
- const std::vector<size_t>& reorder = *((const std::vector<size_t>*)cookie);
- index = reorder[index];
- }
- YQL_ENSURE(index < structType->GetMembersCount());
- WriteSkiffNativeYtValue(structType->GetMemberType(index), nativeYtTypeFlags, value.GetVariantItem(), buf);
- }
- } else if (type->IsVoid() || type->IsNull() || type->IsEmptyList() || type->IsEmptyDict()) {
- } else if (type->IsDict()) {
- auto dictType = AS_TYPE(TDictType, type);
- auto keyType = dictType->GetKeyType();
- auto payloadType = dictType->GetPayloadType();
- NUdf::TUnboxedValue key, payload;
- for (auto iter = value.GetDictIterator(); iter.NextPair(key, payload); ) {
- buf.Write('\0');
- WriteSkiffNativeYtValue(keyType, nativeYtTypeFlags, key, buf);
- WriteSkiffNativeYtValue(payloadType, nativeYtTypeFlags, payload, buf);
- }
- buf.Write('\xff');
- } else {
- YQL_ENSURE(false, "Unsupported type: " << type->GetKindAsStr());
- }
- }
- TExprNode::TPtr ValueToExprLiteral(const TTypeAnnotationNode* type, const NKikimr::NUdf::TUnboxedValuePod& value, TExprContext& ctx,
- TPositionHandle pos) {
- switch (type->GetKind()) {
- case ETypeAnnotationKind::Variant: {
- auto variantType = type->Cast<TVariantExprType>();
- ui32 index = value.GetVariantIndex();
- const TTypeAnnotationNode* itemType;
- if (variantType->GetUnderlyingType()->GetKind() == ETypeAnnotationKind::Struct) {
- // struct
- const auto& items = variantType->GetUnderlyingType()->Cast<TStructExprType>()->GetItems();
- YQL_ENSURE(index < items.size());
- itemType = items[index]->GetItemType();
- } else if (variantType->GetUnderlyingType()->GetKind() == ETypeAnnotationKind::Tuple) {
- // tuple
- const auto& items = variantType->GetUnderlyingType()->Cast<TTupleExprType>()->GetItems();
- YQL_ENSURE(index < items.size());
- itemType = items[index];
- } else {
- YQL_ENSURE(false, "Unknown underlying type");
- }
- return ctx.NewCallable(pos, "Variant", {
- ValueToExprLiteral(itemType, value.GetVariantItem(), ctx, pos),
- ctx.NewAtom(pos, variantType->GetUnderlyingType()->GetKind() == ETypeAnnotationKind::Struct ?
- variantType->GetUnderlyingType()->Cast<TStructExprType>()->GetItems()[index]->GetName() : ToString(index)),
- ExpandType(pos, *type, ctx)
- });
- }
- case ETypeAnnotationKind::Data: {
- auto dataType = type->Cast<TDataExprType>();
- TVector<TExprNode::TPtr> args({ ctx.NewAtom(pos, DataValueToString(value, dataType)) });
- if (auto params = dynamic_cast<const TDataExprParamsType*>(dataType)) {
- args.reserve(3);
- args.push_back(ctx.NewAtom(pos, params->GetParamOne()));
- args.push_back(ctx.NewAtom(pos, params->GetParamTwo()));
- }
- return ctx.NewCallable(pos, dataType->GetName(), std::move(args));
- }
- case ETypeAnnotationKind::Struct: {
- auto structType = type->Cast<TStructExprType>();
- TExprNode::TListType items;
- items.reserve(1 + structType->GetSize());
- items.emplace_back(ExpandType(pos, *type, ctx));
- for (ui32 i = 0; i < structType->GetSize(); ++i) {
- auto pair = ctx.NewList(pos, {
- ctx.NewAtom(pos, structType->GetItems()[i]->GetName()),
- ValueToExprLiteral(structType->GetItems()[i]->GetItemType(), value.GetElement(i), ctx, pos)
- });
- items.emplace_back(std::move(pair));
- }
- return ctx.NewCallable(pos, "Struct", std::move(items));
- }
- case ETypeAnnotationKind::List: {
- auto listType = type->Cast<TListExprType>();
- auto itemType = listType->GetItemType();
- TExprNode::TListType items;
- items.emplace_back(ExpandType(pos, *type, ctx));
- NUdf::TUnboxedValue itemValue;
- for (auto iter = value.GetListIterator(); iter.Next(itemValue);) {
- items.emplace_back(ValueToExprLiteral(itemType, itemValue, ctx, pos));
- }
- if (items.size() > 1) {
- items.erase(items.begin());
- return ctx.NewCallable(pos, "AsList", std::move(items));
- }
- return ctx.NewCallable(pos, "List", std::move(items));
- }
- case ETypeAnnotationKind::Optional: {
- auto optionalType = type->Cast<TOptionalExprType>();
- auto itemType = optionalType->GetItemType();
- if (!value) {
- return ctx.NewCallable(pos, "Nothing", { ExpandType(pos, *type, ctx) });
- } else {
- return ctx.NewCallable(pos, "Just", { ValueToExprLiteral(itemType, value.GetOptionalValue(), ctx, pos)});
- }
- }
- case ETypeAnnotationKind::Dict: {
- auto dictType = type->Cast<TDictExprType>();
- auto keyType = dictType->GetKeyType();
- auto payloadType = dictType->GetPayloadType();
- TExprNode::TListType items;
- items.emplace_back(ExpandType(pos, *type, ctx));
- NUdf::TUnboxedValue keyValue, payloadValue;
- for (auto iter = value.GetDictIterator(); iter.NextPair(keyValue, payloadValue);) {
- auto pair = ctx.NewList(pos, {
- ValueToExprLiteral(keyType, keyValue, ctx, pos),
- ValueToExprLiteral(payloadType, payloadValue, ctx, pos)
- });
- items.emplace_back(std::move(pair));
- }
- return ctx.NewCallable(pos, "Dict", std::move(items));
- }
- case ETypeAnnotationKind::Tuple: {
- auto tupleType = type->Cast<TTupleExprType>();
- TExprNode::TListType items;
- items.reserve(tupleType->GetSize());
- for (ui32 i = 0; i < tupleType->GetSize(); ++i) {
- items.emplace_back(ValueToExprLiteral(tupleType->GetItems()[i], value.GetElement(i), ctx, pos));
- }
- return ctx.NewList(pos, std::move(items));
- }
- case ETypeAnnotationKind::Void: {
- return ctx.NewCallable(pos, "Void", {});
- }
- case ETypeAnnotationKind::Null: {
- return ctx.NewCallable(pos, "Null", {});
- }
- case ETypeAnnotationKind::EmptyList: {
- return ctx.NewCallable(pos, "AsList", {});
- }
- case ETypeAnnotationKind::EmptyDict: {
- return ctx.NewCallable(pos, "AsDict", {});
- }
- case ETypeAnnotationKind::Tagged: {
- auto taggedType = type->Cast<TTaggedExprType>();
- auto baseType = taggedType->GetBaseType();
- return ctx.NewCallable(pos, "AsTagged", {
- ValueToExprLiteral(baseType, value, ctx, pos),
- ctx.NewAtom(pos, taggedType->GetTag()),
- });
- }
- case ETypeAnnotationKind::Pg: {
- auto pgType = type->Cast<TPgExprType>();
- if (!value) {
- return ctx.NewCallable(pos, "Nothing", {
- ctx.NewCallable(pos, "PgType", {
- ctx.NewAtom(pos, pgType->GetName())
- })
- });
- } else {
- return ctx.NewCallable(pos, "PgConst", {
- ctx.NewAtom(pos, PgValueToString(value, pgType->GetId())),
- ctx.NewCallable(pos, "PgType", {
- ctx.NewAtom(pos, pgType->GetName())
- })
- });
- }
- }
- default:
- break;
- }
- YQL_ENSURE(false, "Unsupported type: " << type->GetKind());
- }
- } // namespace NCommon
- } // namespace NYql
|