be/src/cloud/cloud_delta_writer.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 <bthread/mutex.h>

#include "load/delta_writer/delta_writer.h"
#include "runtime/workload_management/resource_context.h"

namespace doris {

class CloudStorageEngine;
class CloudRowsetBuilder;

class CloudDeltaWriter final : public BaseDeltaWriter {
public:
    CloudDeltaWriter(CloudStorageEngine& engine, const WriteRequest& req, RuntimeProfile* profile,
                     const UniqueId& load_id);
    ~CloudDeltaWriter() override;

    Status write(const Block* block, const DorisVector<uint32_t>& row_idxs) override;

    Status close() override;

    Status cancel_with_status(const Status& st) override;

    Status build_rowset() override;

    void update_tablet_stats();

    const RowsetMetaSharedPtr& rowset_meta();

    bool is_init() const { return _is_init; }

    static Status batch_init(std::vector<CloudDeltaWriter*> writers);

    Status commit_rowset();

    Status set_txn_related_info();
    std::shared_ptr<ResourceContext> resource_context() { return _resource_ctx; }

private:
    // Convert `_rowset_builder` from `BaseRowsetBuilder` to `CloudRowsetBuilder`
    CloudRowsetBuilder* rowset_builder();

    // Handle commit for empty rowset (when no data is written)
    Status _commit_empty_rowset();

    bthread::Mutex _mtx;
    CloudStorageEngine& _engine;
    std::shared_ptr<ResourceContext> _resource_ctx;
};

} // namespace doris