// 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 <string>

#include <unordered_map>

#include <unordered_set>

#include <vector>

#include "common/status.h"

#include "format/jni/jni_reader.h"

#include "storage/olap_scan_common.h"

namespace doris {

class RuntimeProfile;

class RuntimeState;

class SlotDescriptor;

class Block;

} // namespace doris

namespace doris {

#include "common/compile_check_begin.h"

/**

 * The demo usage of JniReader, showing how to read data from java scanner.

 * The java side is also a mock reader that provide values for each type.

 * This class will only be retained during the functional testing phase to verify that

 * the communication and data exchange with the jvm are correct.

 */

class PaimonJniReader : public JniReader {

    ENABLE_FACTORY_CREATOR(PaimonJniReader);

public:

    static const std::string PAIMON_OPTION_PREFIX;

    static const std::string HADOOP_OPTION_PREFIX;

    PaimonJniReader(const std::vector<SlotDescriptor*>& file_slot_descs, RuntimeState* state,

                    RuntimeProfile* profile, const TFileRangeDesc& range,

                    const TFileScanRangeParams* range_params);

    ~PaimonJniReader() override = default;

    Status get_next_block(Block* block, size_t* read_rows, bool* eof) override;

    Status init_reader();

private:

    int64_t _remaining_table_level_row_count;

};

#include "common/compile_check_end.h"

} // namespace doris