如何修改CStringA函数以兼容CString、CStringA、CStringW？

问题背景 #

项目里有一批仅支持CStringA的字符串工具函数，比如：

int CountSubstring(CStringA longStr, CStringA substr);
bool StartsWith(CStringA longStr, CStringA head);
bool EndsWith(CStringA longStr, CStringA tail);

现在要让这些函数支持CStringW调用，但之前写的简单模板会允许传入std::string或char*这类不兼容类型，导致编译报错晦涩难懂，需要做类型约束。

解决方案 #

方案1：用CStringT限定模板参数（最直接） #

CStringA和CStringW本质都是CStringT模板的特化：CStringA是CStringT<char, StrTraitMFC<char>>，CStringW是CStringT<wchar_t, StrTraitMFC<wchar_t>>。我们可以直接把模板参数限定为CStringT的特化类型，这样就只允许传入MFC/ATL的CString系列类型。

修改后的函数定义：

template<typename CharType>
int CountSubstring(CStringT<CharType, StrTraitMFC<CharType>> longStr, 
                   CStringT<CharType, StrTraitMFC<CharType>> substr);

template<typename CharType>
bool StartsWith(CStringT<CharType, StrTraitMFC<CharType>> longStr, 
                CStringT<CharType, StrTraitMFC<CharType>> head);

template<typename CharType>
bool EndsWith(CStringT<CharType, StrTraitMFC<CharType>> longStr, 
              CStringT<CharType, StrTraitMFC<CharType>> tail);

这样一来，只有CStringA、CStringW和CString（编译时自动对应宽窄字符）能传入，其他类型会直接触发明确的类型不匹配错误，比之前的晦涩报错友好很多。

方案2：用静态断言做类型检查（错误提示更直白） #

如果觉得上面的模板参数写起来太长，也可以保留简单模板，同时用static_assert在函数内部做类型校验，直接告诉调用者允许的类型：

#include <type_traits>

template<typename T>
int CountSubstring(T longStr, T substr) {
    // 强制限定只能传CStringA或CStringW
    static_assert(std::is_same_v<T, CStringA> || std::is_same_v<T, CStringW>, 
                  "CountSubstring仅支持CStringA或CStringW类型参数");
    
    // 原函数逻辑修改：把针对char的API换成CStringT通用成员函数
    int count = 0;
    int pos = 0;
    while ((pos = longStr.Find(substr, pos)) != -1) {
        count++;
        pos += substr.GetLength();
    }
    return count;
}

// StartsWith和EndsWith同理
template<typename T>
bool StartsWith(T longStr, T head) {
    static_assert(std::is_same_v<T, CStringA> || std::is_same_v<T, CStringW>, 
                  "StartsWith仅支持CStringA或CStringW类型参数");
    return longStr.Left(head.GetLength()) == head;
}

template<typename T>
bool EndsWith(T longStr, T tail) {
    static_assert(std::is_same_v<T, CStringA> || std::is_same_v<T, CStringW>, 
                  "EndsWith仅支持CStringA或CStringW类型参数");
    return longStr.Right(tail.GetLength()) == tail;
}

这种方式的错误提示非常直观，调用者一眼就能明白是参数类型不对，模板定义也更简洁。

关于CString/CStringT的简单说明 #

• CString是MFC的typedef，Unicode环境下等价于CStringW，多字节环境下等价于CStringA
• CStringT是ATL/MFC的通用字符串模板类，CStringA和CStringW都是它的特化版本，分别对应char和wchar_t类型
• 用CStringT做模板参数，能同时兼容宽窄字符的CString类型

迁移注意事项 #

1. 原函数里的字符串操作要改成通用方式：比如把strstr换成_tcsstr，或者直接用CStringT自带的成员函数（比如Find、Left、Right，CStringA和CStringW都支持）
2. 原有上百处CStringA的调用不需要修改，会自动匹配模板的CharType=char版本
3. 新的CStringW调用直接传参即可，模板会自动实例化对应的版本

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