c++ - 运算符 >> 用于字符指针包装器

Question

因为不允许我使用 std::string 进行赋值，所以我使用一个名为 CharString 的自定义类作为那些我无法自己初始化的情况的包装器。

这个类看起来像这样:

struct CharString {
    char* str;
    CharString() : str() {}  // Initialize NULL
    ~CharString() { free(str); }
    // Conversions to be usable with C functions
    operator char**() { return &str; }
    operator char*() { return str; }
};

但是当我想在上面使用 >> 时出现以下错误。

binary '>>': no operator found which takes a right-hand operand of type 'utils::CharString' (or there is no acceptable conversion).

如何重载运算符 >>>？

CharString operator>>(std::istream& is) {
    is >> str;
    return *this;
};

我尝试了上面的方法，但仍然出现同样的错误。

Answer 1

重载流提取运算符的一般方法是提供具有以下一般形式的友元函数:

istream& operator>> (istream& in, MyObjectType& object) {
    // read all the data you need to construct an object.
    //
    // Then:
    if (in) { // Read succeeded! Update object.
        // Build a new object of the appropriate type.
        MyObjectType theOneIRead(basedOnTheDataIRead);

        // Swap the object you were given as input for the one you
        // just read. This way, if the read completely succeeds, you
        // update the rhs, and if the read failed, nothing happens.
        std::swap(object, theOneIRead);
    }
    return in;
}

请注意，此函数的签名将与您编写的签名不同。

您需要处理与从流中一次读取一个字符相关的逻辑，并构建某种临时缓冲区来保存它们。这不是微不足道的，也是为什么我们有一个 std::string 类型与库一起打包的原因之一。但除此之外，遵循下面的模板应该可以满足您的需求。

独立地 - 您的结构当前有一个析构函数，但没有复制构造函数、移动构造函数或赋值运算符。通常，您还希望在析构函数的同时实现这些函数，否则复制您的对象最终会进行指针的浅拷贝，并在两个独立的对象试图释放同一个指针时导致内存问题。

此外，由于这是 C++，请考虑使用 new[] 和 delete[] 而不是 malloc 和 free.