mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-21 15:12:02 +00:00
69 lines
2.3 KiB
C++
69 lines
2.3 KiB
C++
#include "createVolume.h"
|
|
|
|
#include <Disks/SingleDiskVolume.h>
|
|
#include <Disks/VolumeJBOD.h>
|
|
#include <Disks/VolumeRAID1.h>
|
|
|
|
#include <boost/algorithm/string.hpp>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int UNKNOWN_RAID_TYPE;
|
|
extern const int INVALID_RAID_TYPE;
|
|
}
|
|
|
|
VolumePtr createVolumeFromReservation(const ReservationPtr & reservation, VolumePtr other_volume)
|
|
{
|
|
if (other_volume->getType() == VolumeType::JBOD || other_volume->getType() == VolumeType::SINGLE_DISK)
|
|
{
|
|
/// Since reservation on JBOD chooses one of disks and makes reservation there, volume
|
|
/// for such type of reservation will be with one disk.
|
|
return std::make_shared<SingleDiskVolume>(other_volume->getName(), reservation->getDisk(), other_volume->max_data_part_size);
|
|
}
|
|
if (other_volume->getType() == VolumeType::RAID1)
|
|
{
|
|
auto volume = std::dynamic_pointer_cast<VolumeRAID1>(other_volume);
|
|
return std::make_shared<VolumeRAID1>(volume->getName(), reservation->getDisks(), volume->max_data_part_size, volume->are_merges_avoided);
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
VolumePtr createVolumeFromConfig(
|
|
String name,
|
|
const Poco::Util::AbstractConfiguration & config,
|
|
const String & config_prefix,
|
|
DiskSelectorPtr disk_selector
|
|
)
|
|
{
|
|
String raid_type = config.getString(config_prefix + ".raid_type", "JBOD");
|
|
if (raid_type == "JBOD")
|
|
{
|
|
return std::make_shared<VolumeJBOD>(name, config, config_prefix, disk_selector);
|
|
}
|
|
throw Exception("Unknown RAID type '" + raid_type + "'", ErrorCodes::UNKNOWN_RAID_TYPE);
|
|
}
|
|
|
|
VolumePtr updateVolumeFromConfig(
|
|
VolumePtr volume,
|
|
const Poco::Util::AbstractConfiguration & config,
|
|
const String & config_prefix,
|
|
DiskSelectorPtr & disk_selector
|
|
)
|
|
{
|
|
String raid_type = config.getString(config_prefix + ".raid_type", "JBOD");
|
|
if (raid_type == "JBOD")
|
|
{
|
|
VolumeJBODPtr volume_jbod = std::dynamic_pointer_cast<VolumeJBOD>(volume);
|
|
if (!volume_jbod)
|
|
throw Exception("Invalid RAID type '" + raid_type + "', shall be JBOD", ErrorCodes::INVALID_RAID_TYPE);
|
|
|
|
return std::make_shared<VolumeJBOD>(*volume_jbod, config, config_prefix, disk_selector);
|
|
}
|
|
throw Exception("Unknown RAID type '" + raid_type + "'", ErrorCodes::UNKNOWN_RAID_TYPE);
|
|
}
|
|
|
|
}
|