帮助自定义getline()函数

#include<iostream>
#include<string>

std::istream& custom_getline(std::istream& is, std::string& str)
{
    str.clear();

    // The input stream is in a bad state
    if (!is)
        return is;

    // Start reading characters
    char c = is.get();
    while (is && c != '\n')
    {
        str += c;
        c = is.get();
    }

    return is;
}

int main()
{
    std::string input;
    std::cout << "Enter some text: ";
    custom_getline(std::cin, input);
    std::cout << "You entered: "<< input<< std::endl;

    return 0;
}