be/src/vec/jsonb/serialize.h

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

#pragma once

#include <cstddef>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#include "olap/tablet_schema.h"
#include "runtime/descriptors.h"
#include "vec/columns/column_string.h"
#include "vec/core/block.h"

namespace doris {
class TabletSchema;
class TupleDescriptor;
} // namespace doris

namespace doris::vectorized {
// use jsonb codec to store row format
class JsonbSerializeUtil {
public:
    // encode partial columns into jsonb
    // empty row_store_cids means encode full schema columns for compability
    static void block_to_jsonb(const TabletSchema& schema, const Block& block, ColumnString& dst,
                               int num_cols, const DataTypeSerDeSPtrs& serdes,
                               const std::unordered_set<int32_t>& row_store_cids);
    // batch rows
    static Status jsonb_to_block(const DataTypeSerDeSPtrs& serdes, const ColumnString& jsonb_column,
                                 const std::unordered_map<uint32_t, uint32_t>& col_id_to_idx,
                                 Block& dst, const std::vector<std::string>& default_values,
                                 const std::unordered_set<int>& include_cids);
    // single row
    static Status jsonb_to_block(const DataTypeSerDeSPtrs& serdes, const char* data, size_t size,
                                 const std::unordered_map<uint32_t, uint32_t>& col_id_to_idx,
                                 Block& dst, const std::vector<std::string>& default_values,
                                 const std::unordered_set<int>& include_cids);
};
} // namespace doris::vectorized