蘭陵N梓記

一指流沙,程序年华


  • 首页

  • 归档

  • 关于

  • 搜索
close

飞哥讲代码29:C++可变参数模板应用

时间: 2022-04-04   |   分类: 技术     |   阅读: 2409 字 ~5分钟

案例

在我司新的C++融合编程规范中,提到避免定义C风格的变参数函数:

为了避免类型错误,应当使用可变参数模板等其它的方式来代替va_arg可变参数。

规范中也给出了一个简单的示例,像我这种有使用过的经验感觉示例有点意犹未尽。去年年底我使用C++写一个小工具,其中就使用可变参数模板封装了一个日志接口,同时参考了Java slf4j日志用法,支持占位符。正好借这个代码的来温习与讲解一下可变参数模板,感受一下C++模板编程的魅力。

使用示例(如果您可以使用C++20,建议参考std::format):

INFO("Hell world");
INFO("Hi {}", "xiao ming");
INFO("Welcome {} join {}", "to", "conference");

核心函数如下所示:

// 函数1:递归终止函数
void write_log(int log_level, const char *code_file, int code_line, string &msg) 
{}

// 函数2:展开模板函数
template<typename First, typename ...Rest>
void write_log(int log_level, const char *code_file, int code_line, string &fmt,
    const First &first, const Rest &...rest) 
{
        auto idx = fmt.find("{}");
        if (idx == string::npos) { // 如果没有占位符,则忽略可变参数
            write_log(log_level, code_file, code_line, fmt);
        } else {
            ostringstream oss;
            oss << first;
            fmt.replace(idx, PLACEHOLDER_SIZE, oss.str()); // constepxr int PLACEHOLDER_SIZE=2
            write_log(log_level, code_file, code_line, fmt, rest...);
        }
}

// 函数3:可变参数模板函数
template<typename ...Args>
void write_log(int log_level, const char *code_file, int code_line, const string &fmt,
    const Args &...args) 
{
    string tmp(fmt);
    write_log(log_level, code_file, code_line, tmp, args...);
}

#define INFO(fmt, ...) \
write_log(LogLevel::INFO, __FILE__, __LINE__, fmt, ##__VA_ARGS__)

代码讲解

可变参数模板是C++11新增的特性,允许模板定义中包含0到任意个模板参数。声明可变参数模板时,需要在typename或class后面加上省略号"…"。 省略号的作用有两个:

  • 声明一个参数包,这个参数包中可以包含0到任意个模板参数
  • 在模板定义的右边,可以将参数包展开成一个一个独立的参数

结合上面的案例:

  • 函数3:是可变参数模板函数。参数包可以包含0个或者多个参数,如果需要用参数包中的参数,则一定要将参数包展开。
  • 函数2:最展开模板函数,它是采用递归函数方式展开参数包。逐渐展开First参数,剩余的参数Rest参数再次传给展开函数,即逐渐吃掉展开的First参数。
  • 函数1:是递归终止函数。递归终止函数的展开参数可以为0个或1个,当参数包中剩余的参数个数等于递归终止函数的参数个数时,就调用递归终止函数,则函数终止。案例代码中函数终止的展开可变参数是0(log_level, code_file, code_line, fmt并非可变参数)。

案例代码中的三个函数是采用同名函数(最后一个函数fmt参数多一个const),排版上存在前后顺序,递归终止函数是放在最前面,再次是展开函数,最后是入口函数。

C++11也提供sizeof...来得到可变参数的个数,那我们是否可以通过判断个数来直接调用,省掉终止函数?

template<typename First, typename ...Rest>
void write_log(int log_level, const char *code_file, int code_line, string &fmt,
    const First &first, const Rest &...rest) 
{
    // ...
    if (sizeof...(rest) > 0) { // 增加参数个数
        write_log(log_level, code_file, code_line, fmt, rest...);
    } else {
        // 输出日志
    }   
}

这个是编译通不过的,因为模板在编译阶段也会将if的所有代码都进行编译,而不会去根据if的条件去进行选择性的编译,选择运行if的哪个分支是在运行期间做的。C++17引入了编译期的if(Compile-Time If),所以上面的代码可以这么写:

template<typename First, typename ...Rest>
void write_log(int log_level, const char *code_file, int code_line, string &fmt,
    const First &first, const Rest &...rest) 
{
    // ...
    if constexpr (sizeof...(rest) > 0) { // 增加参数个数
        write_log(log_level, code_file, code_line, fmt, rest...);
    } else {
        // 输出日志
    }   
}

if constexpr表示编译期if的语法,可以进行在编译期决定编译if条件的哪个分支。当然if中的求值要求在编译期可以计算,否则也会报错。侧面也说明sizeof...对可变参数求个数是编译期计算的。

在c++11还可以使用enable_if,它用于判断当某个条件成立时是否匹配哪个函数调用,简介可参见飞哥讲代码20:窥探C++的模板

扩展阅读

初始化列表

对于可变参数还可以使用初始化列表initializer_list变通方式来支持可变参数,如:

const string join(const initializer_list<string> &list, const string_view &sep) 
{}

join({"1"});
join({"1", "2"});
join({"1", "2", "3"});

但是initializer_list<T>需要指定固定类型,则不能像上述案例可变参数模板函数支持可变参数可以是不同的类型,如:

INFO("len of {} is {}", "xiao ming", 9);

另外,C++11还提供tuple,是一个可变参数模板类,可以携带任意类型任意个数的模板参数。

初始化列表方式展开

前面对于可变参数模板函数展开参数包是采用递归的方式,也支持采用初始化列表initializer_list方式展开。

template<typename T>
void print(T a)
{
    cout << a << endl;
}
 
template<typename ...Args>
void expand(Args... args)
{
    auto a = {(print(args), 0)...};
    // 或者 std::initializer_list<int>{(print(args), 0)...};
}

// {(print(args), 0)...}会被展开为{(print(arg1), 0), (print(arg2), 0), (print(arg3), 0)}
expand("hello", "word", 1, 2);

这里利用了逗号表达式技巧,由于initializer_list<T>必须固定类型,而(print(args), 0) 逗号表达式的结果是0,所以能赋值给initializer_list<int>,则Args可以是不同的类型不受initializer_list类型的影响。

print(args)其中的args是取到第一个参数,...则是展开,initializer_list初始化支持这种方式。为了减少干扰,去掉函数调用如下所示,其中具名的args...,或者匿名的...代表所有的可变参数集合,可以将args和...分开,此时args表示...中每一个参数。

template<typename... Args>
void print(Args... args)
{
    auto a = { (args, 0)... }; 
}

上面的示例可以进一步简化,采用lambda表达式替换print(T a)。

template<typename... Args>
void expand(Args... args)
{
    std::initializer_list<int>{([&]{cout << args << endl;}(), 0)...};
}

折叠表达式

从C++17开始,折叠表达式可以将二元运算符作用于所有参数展开上:

Fold Expression Evaluation
( … op pack ) ((( pack1 op pack2 ) op pack3 ) … op packN )
( pack op … ) ( pack1 op ( … ( packN-1 op packN )))
( init op … op pack ) ((( init op pack1 ) op pack2 ) … op packN )
( pack op … op init ) ( pack1 op ( … ( packN op init )))
template<typename... T>
auto foldSum (T... s) 
{
  return (... + s);           // ((s1 + s2) + s3) ..., 注意有()
}

foldSum(1, 2, 3, 4);

采用折叠表达式,对前面的expand改进:

template<typename... T>
void expand(T const&... args) 
{
  (std::cout << ... << args) << '';  // 注意有()
}

如果想要在每个参数中间输出空格,可以配合lambda:

template <typename First, typename... Args>
void print(First first, Args... args) 
{
  std::cout << first;
  auto printWhiteSpace = [](const auto &arg) { std::cout << " " << arg; };
  (..., printWhiteSpace(args)); // op是,
}

结语

现代C++引入不少特征,可变参数模板与折叠表达式在模板编程上带来一些新的检验。我司新的编程规范也紧跟语言发展脚步,对新的特征也有不少的描述。作为一名写了近20年代码的老兵,唯有不断学习才能跟上编程语言的发展。在工作中用好这些特性,让代码更现代化,享受语言新特性带来红利。

#软件开发# #c++#
飞哥讲代码30:C++简单依赖注入
软件开发漫谈7:开发协作
微信扫一扫交流

标题:飞哥讲代码29:C++可变参数模板应用
作者:兰陵子
关注:lanlingthink(览聆时刻)
声明:自由转载-非商用-非衍生-保持署名(创作共享3.0许可证)

  • 文章目录
  • 站点概览
兰陵子

兰陵子

Programmer & Architect

164 日志
4 分类
57 标签
GitHub 知乎
  • 案例
  • 代码讲解
  • 扩展阅读
    • 初始化列表
    • 初始化列表方式展开
    • 折叠表达式
  • 结语
© 2009 - 2022 蘭陵N梓記
Powered by - Hugo v0.101.0
Theme by - NexT
0%