1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
/*
* Ceph - scalable distributed file system
*
* Copyright (C) 2014 Haomai Wang <haomaiwang@gmail.com>
*
* This is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License version 2.1, as published by the Free Software
* Foundation. See file COPYING.
*
*/
#include <random>
#include <sstream>
#include <iterator>
#include <algorithm>
#include "CompressionPlugin.h"
#include "Compressor.h"
#include "include/random.h"
#include "common/ceph_context.h"
#include "common/debug.h"
#include "common/dout.h"
const char* Compressor::get_comp_alg_name(int a) {
auto p = std::find_if(std::cbegin(compression_algorithms), std::cend(compression_algorithms),
[a](const auto& kv) { return kv.second == a; });
if (std::cend(compression_algorithms) == p)
return "???"; // It would be nice to revise this...
return p->first;
}
boost::optional<Compressor::CompressionAlgorithm> Compressor::get_comp_alg_type(const std::string &s) {
auto p = std::find_if(std::cbegin(compression_algorithms), std::cend(compression_algorithms),
[&s](const auto& kv) { return kv.first == s; });
if (std::cend(compression_algorithms) == p)
return {};
return p->second;
}
const char *Compressor::get_comp_mode_name(int m) {
switch (m) {
case COMP_NONE: return "none";
case COMP_PASSIVE: return "passive";
case COMP_AGGRESSIVE: return "aggressive";
case COMP_FORCE: return "force";
default: return "???";
}
}
boost::optional<Compressor::CompressionMode> Compressor::get_comp_mode_type(const std::string &s) {
if (s == "force")
return COMP_FORCE;
if (s == "aggressive")
return COMP_AGGRESSIVE;
if (s == "passive")
return COMP_PASSIVE;
if (s == "none")
return COMP_NONE;
return boost::optional<CompressionMode>();
}
CompressorRef Compressor::create(CephContext *cct, const std::string &type)
{
// support "random" for teuthology testing
if (type == "random") {
int alg = ceph::util::generate_random_number(0, COMP_ALG_LAST - 1);
if (alg == COMP_ALG_NONE) {
return nullptr;
}
return create(cct, alg);
}
CompressorRef cs_impl = NULL;
std::stringstream ss;
PluginRegistry *reg = cct->get_plugin_registry();
CompressionPlugin *factory = dynamic_cast<CompressionPlugin*>(reg->get_with_load("compressor", type));
if (factory == NULL) {
lderr(cct) << __func__ << " cannot load compressor of type " << type << dendl;
return NULL;
}
int err = factory->factory(&cs_impl, &ss);
if (err)
lderr(cct) << __func__ << " factory return error " << err << dendl;
return cs_impl;
}
CompressorRef Compressor::create(CephContext *cct, int alg)
{
if (alg < 0 || alg >= COMP_ALG_LAST) {
lderr(cct) << __func__ << " invalid algorithm value:" << alg << dendl;
return CompressorRef();
}
std::string type_name = get_comp_alg_name(alg);
return create(cct, type_name);
}
|
