cpp-ch/local-engine/Storages/SubstraitSource/Iceberg/EqualityDeleteFileReader.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 <Core/Block.h>
#include <Functions/FunctionFactory.h>
#include <Interpreters/Context_fwd.h>
#include <Interpreters/ExpressionActions.h>
#include <Storages/SubstraitSource/substrait_fwd.h>

namespace local_engine::iceberg
{

/// Visible for UT
///
class EqualityDeleteActionBuilder
{
public:
    static constexpr auto COLUMN_NAME = "__kept__";

private:
    DB::ActionsDAG actions;
    const DB::ContextPtr context;
    DB::ActionsDAG::NodeRawConstPtrs andArgs = {};
    UInt64 unique_name_counter = 0;

    const DB::ActionsDAG::Node & lastMerge();
    const DB::ActionsDAG::Node * Or(const DB::ActionsDAG::NodeRawConstPtrs & orArgs);

    std::string getUniqueName(const String & name = "_") { return name + "_" + std::to_string(unique_name_counter++); }
    const DB::ActionsDAG::Node & addFunction(const DB::FunctionOverloadResolverPtr & function, DB::ActionsDAG::NodeRawConstPtrs args);

public:
    explicit EqualityDeleteActionBuilder(const DB::ContextPtr & context_, const DB::NamesAndTypesList & inputs_)
        : actions(inputs_), context(context_)
    {
    }

    void notIn(DB::Block deleteBlock, const std::string & column_name = "");
    void notEquals(DB::Block deleteBlock, const DB::Names & column_names = {});
    DB::ExpressionActionsPtr finish();
};

class EqualityDeleteFileReader
{
    const DB::ContextPtr & context_;
    const SubstraitIcebergDeleteFile & deleteFile_;
    DB::Block data_file_schema_for_delete_;

public:
    static DB::ExpressionActionsPtr createDeleteExpr(
        const DB::ContextPtr & context,
        const DB::Block & data_file_header,
        const google::protobuf::RepeatedPtrField<SubstraitIcebergDeleteFile> & delete_files,
        const std::vector<int> & equality_delete_files,
        DB::Block & reader_header);

    explicit EqualityDeleteFileReader(
        const DB::ContextPtr & context, const DB::Block & read_header, const SubstraitIcebergDeleteFile & deleteFile);
    ~EqualityDeleteFileReader() = default;
    void readDeleteValues(EqualityDeleteActionBuilder & expressionInputs) const;
};

}