#pragma once

#include "Internals.h"
#include "TaskTable.h"

namespace DB {


/// Just destination partition of a shard
/// I don't know what this comment means.
/// In short, when we discovered what shards contain currently processing partition,
/// This class describes a partition (name) that is stored on the shard (parent).
struct ShardPartition {
    ShardPartition(TaskShard &parent, String name_quoted_, size_t number_of_splits = 10)
            : task_shard(parent), name(std::move(name_quoted_)) { pieces.reserve(number_of_splits); }

    /*useful*/ String getPartitionPath() const;

    [[maybe_unused]] String getPartitionPiecePath(size_t current_piece_number) const;

    /*useful*/ String getPartitionCleanStartPath() const;

    [[maybe_unused]] String getPartitionPieceCleanStartPath(size_t current_piece_number) const;

    /*useful*/ String getCommonPartitionIsDirtyPath() const;

    /*useful*/ String getCommonPartitionIsCleanedPath() const;

    /*??????*/ String getPartitionActiveWorkersPath() const;

    /*??????*/ String getActiveWorkerPath() const;

    /*useful*/ String getPartitionShardsPath() const;

    /*useful*/ String getShardStatusPath() const;

    /// What partition pieces are present in current shard.
    /// FYI: Piece is a part of partition which has modulo equals to concrete constant (less than number_of_splits obliously)
    /// For example SELECT ... from ... WHERE partition=current_partition AND cityHash64(*) == const;
    /// Absent pieces have field is_absent_piece equals to true.
    PartitionPieces pieces;

    TaskShard & task_shard;
    String name;
};

String ShardPartition::getPartitionCleanStartPath() const {
    return getPartitionPath() + "/clean_start";
}

String ShardPartition::getPartitionPieceCleanStartPath(size_t current_piece_number) const {
    assert(current_piece_number < task_shard.task_table.number_of_splits);
    return getPartitionPiecePath(current_piece_number) + "/clean_start";
}

String ShardPartition::getPartitionPath() const {
    return task_shard.task_table.getPartitionPath(name);
}

String ShardPartition::getPartitionPiecePath(size_t current_piece_number) const {
    assert(current_piece_number < task_shard.task_table.number_of_splits);
    return task_shard.task_table.getPartitionPiecePath(name, current_piece_number);
}

String ShardPartition::getShardStatusPath() const {
    // schema: /<root...>/tables/<table>/<partition>/shards/<shard>
    // e.g. /root/table_test.hits/201701/shards/1
    return getPartitionShardsPath() + "/" + toString(task_shard.numberInCluster());
}

String ShardPartition::getPartitionShardsPath() const {
    return getPartitionPath() + "/shards";
}

String ShardPartition::getPartitionActiveWorkersPath() const {
    return getPartitionPath() + "/partition_active_workers";
}

String ShardPartition::getActiveWorkerPath() const {
    return getPartitionActiveWorkersPath() + "/" + toString(task_shard.numberInCluster());
}

String ShardPartition::getCommonPartitionIsDirtyPath() const {
    return getPartitionPath() + "/is_dirty";
}

String ShardPartition::getCommonPartitionIsCleanedPath() const {
    return getCommonPartitionIsDirtyPath() + "/cleaned";
}


}