如何简化含双层循环的相似C++模板函数？哪种实现更合理？

原方案（两个独立模板函数） #

• 优点：循环内部无分支判断，性能拉满；编译器可针对每个函数的内部代码做充分优化（比如内联）。
• 缺点：双层循环代码完全重复，后续修改循环逻辑（比如调整边界、循环顺序）时，必须同时改动function_A和function_B，维护成本高，容易出现漏改。

你提出的循环外提+内部分支方案 #

• 优点：只写一次双层循环，维护性提升，改循环逻辑只需修改一处。
• 缺点：每次循环都要做functiontest的分支判断，如果循环次数极多（比如百万级以上），分支预测失败的开销会累积影响性能；另外，内部函数的调用如果无法被编译器内联，也会带来额外开销。

利用C++模板特性，把循环内部的逻辑作为模板参数传入，既消除重复代码，又避免运行时分支，兼顾性能和维护性。

示例代码：

// 通用循环框架模板函数
template<typename Func>
void generic_function(Func&& func, int h_max, int i_max, int* array_test) {
    for (int h = 0; h < h_max; ++h) {
        for (int i = 0; i < i_max; ++i) {
            // 直接执行传入的逻辑
            std::forward<Func>(func)(h, i, array_test);
        }
    }
}

// 原CODE1的逻辑封装
void code1(int h, int i, int* array_test) {
    // 这里写CODE1的内容
    array_test[h * 10 + i] = h + i;
}

// 原CODE2的逻辑封装
void code2(int h, int i, int* array_test) {
    // 这里写CODE2的内容
    array_test[h * 10 + i] = h * i;
}

// 调用逻辑
void caller(bool functiontest, int* array_test) {
    if (functiontest) {
        generic_function(code1, 10, 10, array_test);
    } else {
        generic_function(code2, 10, 10, array_test);
    }
}

该方案的优势：

• 循环逻辑仅需编写一次，维护成本低；
• 分支判断仅在循环外部执行一次，循环内部无额外开销；
• 编译器可轻松将code1/code2内联到generic_function中，性能和原方案几乎无差别；
• 扩展性强，后续新增循环内部逻辑时，只需新增对应的函数/lamda，无需修改通用循环框架。

如果functiontest是编译期常量（比如宏定义、constexpr变量），还可以用if constexpr进一步优化，完全消除运行时分支：

template<bool FunctionTest>
void caller(int* array_test) {
    generic_function([] (int h, int i, int* arr) {
        if constexpr (FunctionTest) {
            // CODE1内容
        } else {
            // CODE2内容
        }
    }, 10, 10, array_test);
}

• 如果循环次数少、对性能不敏感，你提出的循环外提方案完全够用，实现简单；
• 如果追求性能和维护性的平衡，优先选择模板参数化行为的方案；
• 如果原方案的循环代码极少改动，且性能是第一优先级，原方案也可接受，但长期维护成本更高。

内容的提问来源于stack exchange，提问作者chip