cpp_src/core/cjson/baseencoder.h - master - uzer_007/reindexer

cpp_src/core/cjson/baseencoder.h
95 строк4 KB
madschemas
Update to version v5.10.0
29 дек 2025, 14:43
29 дек 2025, 14:4328ca522
#pragma once

#include "core/payload/payloadiface.h"
#include "fieldextractor.h"
#include "tagslengths.h"
#include "tools/serializer.h"

namespace reindexer {

class TagsMatcher;
class FieldsFilter;

namespace builders {
class CsvBuilder;
class ProtobufBuilder;
class CJsonBuilder;
class MsgPackBuilder;
}  // namespace builders
using builders::CsvBuilder;
using builders::ProtobufBuilder;
using builders::CJsonBuilder;
using builders::MsgPackBuilder;

class [[nodiscard]] IEncoderDatasourceWithJoins {
public:
	IEncoderDatasourceWithJoins() = default;
	virtual ~IEncoderDatasourceWithJoins() = default;

	virtual size_t GetJoinedRowsCount() const noexcept = 0;
	virtual size_t GetJoinedRowItemsCount(size_t rowId) const = 0;
	virtual ConstPayload GetJoinedItemPayload(size_t rowid, size_t plIndex) = 0;
	virtual const std::string& GetJoinedItemNamespace(size_t rowid) & noexcept = 0;
	virtual const TagsMatcher& GetJoinedItemTagsMatcher(size_t rowid) & noexcept = 0;
	virtual const FieldsFilter& GetJoinedItemFieldsFilter(size_t rowid) & noexcept = 0;

	auto GetJoinedItemNamespace(size_t) && = delete;
	auto GetJoinedItemTagsMatcher(size_t) && = delete;
	auto GetJoinedItemFieldsFilter(size_t) && = delete;
};

template <typename Builder>
class [[nodiscard]] IAdditionalDatasource {
public:
	virtual void PutAdditionalFields(Builder&) const = 0;
	virtual IEncoderDatasourceWithJoins* GetJoinsDatasource() noexcept = 0;
};

template <typename Builder>
class [[nodiscard]] BaseEncoder {
public:
	explicit BaseEncoder(const TagsMatcher* tagsMatcher, const FieldsFilter* filter);
	void Encode(ConstPayload& pl, Builder& builder,
				const h_vector<IAdditionalDatasource<Builder>*, 2>& dss = h_vector<IAdditionalDatasource<Builder>*, 2>());
	void Encode(std::string_view tuple, Builder& wrSer,
				const h_vector<IAdditionalDatasource<Builder>*, 2>& dss = h_vector<IAdditionalDatasource<Builder>*, 2>());

	const TagsLengths& GetTagsMeasures(ConstPayload& pl, IEncoderDatasourceWithJoins* ds = nullptr);

private:
	using IndexedTagsPathInternalT = IndexedTagsPathImpl<16>;
	constexpr static bool kWithTagsPathTracking = std::is_same_v<ProtobufBuilder, Builder>;
	constexpr static bool kWithFieldExtractor = std::is_same_v<FieldsExtractor, Builder>;

	struct [[nodiscard]] DummyTagsPathScope {
		DummyTagsPathScope(TagsPath&, TagName) noexcept {}
	};
	using PathScopeT = std::conditional_t<kWithTagsPathTracking, TagsPathScope<TagsPath>, DummyTagsPathScope>;

	template <concepts::TagNameOrIndex TagType, typename BuilderT>
	bool encode(ConstPayload* pl, Serializer& rdser, BuilderT&& builder, TagType indexedTag);
	template <concepts::TagNameOrIndex TagType, typename BuilderT>
	bool encodeImpl(ConstPayload* pl, ctag ctag, Serializer& rdser, BuilderT&& builder, TagType indexedTag);
	void encodeJoinedItems(Builder& builder, IEncoderDatasourceWithJoins* ds, size_t joinedIdx);
	bool collectTagsSizes(ConstPayload& pl, Serializer& rdser);
	void collectJoinedItemsTagsSizes(IEncoderDatasourceWithJoins* ds, size_t rowid);

	std::string_view getPlTuple(ConstPayload& pl);

	const TagsMatcher* tagsMatcher_{nullptr};
	std::array<int, kMaxIndexes> fieldsoutcnt_;
	const FieldsFilter* filter_{nullptr};
	WrSerializer tmpPlTuple_;
	TagsPath curTagsPath_;
	IndexedTagsPathInternalT indexedTagsPath_;
	TagsLengths tagsLengths_;
	ScalarIndexesSetT objectScalarIndexes_;
};

using JsonEncoder = BaseEncoder<JsonBuilder>;
using CJsonEncoder = BaseEncoder<CJsonBuilder>;
using MsgPackEncoder = BaseEncoder<MsgPackBuilder>;
using ProtobufEncoder = BaseEncoder<ProtobufBuilder>;
using CsvEncoder = BaseEncoder<CsvBuilder>;

}  // namespace reindexer