重写istream的get(char)：如何标识通道未关但暂无数据的状态？

核心思路 #

C++标准流的状态系统里，除了EOF，还能靠自定义错误状态位来区分“暂时无数据但通道有效”和真正的EOF。别直接返回EOF，不然会被get(char)这类标准操作误判成通道关闭。

具体实现步骤 #

1. 自定义状态位
   用std::ios_base::iostate的扩展位，避开标准的failbit、badbit、eofbit，自定义一个专属标识：

   // 自定义无数据状态位，确保不与标准位冲突
   const std::ios_base::iostate no_data_bit = std::ios_base::iostate(1 << 3);
   

2. 在underflow里设置状态
   检测到暂时无数据时，别返回traits_type::eof()，而是这么做：

   • 先清空输入缓冲区，避免旧数据干扰
   • 给关联的流设置自定义状态，同时返回一个非EOF的有效值：

     std::basic_streambuf<char>::int_type underflow() override {
         if (/* 判定当前无数据但通道未关闭 */) {
             // 清空缓冲区指针
             setg(nullptr, nullptr, nullptr);
             // 获取关联的流对象，设置自定义状态
             if (auto* ios = static_cast<std::basic_iostream<char>*>(this->getloc().get<std::ios_base*>())) {
                 ios->setstate(no_data_bit);
             }
             // 返回非EOF值，让get()不认为是结束
             return traits_type::not_eof(0);
         } else if (/* 通道真的关闭了 */) {
             return traits_type::eof();
         }
         // 正常有数据的处理逻辑
         // ...
     }
     

3. 上层判断状态
   调用get(char)后，别只查eof()，要同时检查自定义状态：

   char c;
   my_stream.get(c);
   if (my_stream.eof()) {
       // 通道彻底关闭，处理真正的EOF
   } else if (my_stream.rdstate() & no_data_bit) {
       // 暂时没数据，通道还活着，稍后重试就行
       my_stream.clear(no_data_bit); // 清除状态，方便后续操作
   } else if (my_stream.fail()) {
       // 其他错误情况，比如读取失败
   } else {
       // 成功读到数据
   }
   

4. 关键注意点

   • 别碰标准的failbit，不然标准流操作会直接判定为“操作失败”，而你这只是暂时没数据，操作本身是正常的。
   • 等后续有数据时，一定要先清除自定义状态位，不然会残留状态导致误判。
   • 如果是多线程场景，状态的设置和清除要加锁保证安全。

替代方案：封装等待逻辑 #

如果上层逻辑允许，也可以给你的流类加个wait_for_data()方法，在调用get()前先等数据，减少自定义状态的触发：

bool MyStream::wait_for_data(std::chrono::milliseconds timeout) {
    // 这里写轮询或阻塞等待数据的逻辑
    if (has_new_data()) {
        clear(no_data_bit);
        return true;
    }
    return false;
}

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