Question

为了理解输入流是如何工作的，我设计了以下两个类：

#include <iostream>

class my_streambuf: public std::streambuf
{
private:
  std::streambuf* buffer;

  char ch;

protected:

  virtual std::streambuf::int_type underflow()
  {
    std::streambuf::int_type result = buffer->sbumpc();

    if (result != traits_type::eof())
    {
      ch = traits_type::to_char_type(result);

      setg(&ch, &ch, &ch + 1);
    }

    return result;
  }

public:
  my_streambuf(std::streambuf* buffer) : buffer(buffer) {};

  virtual ~my_streambuf() {};
};

class my_istream: public std::istream
{
public:
  my_istream(std::istream& stream) : std::istream(new my_streambuf(stream.rdbuf())) {};

  virtual ~my_istream()
  {
    delete rdbuf();
  }
};

int main()
{
  char s[32];
  my_istream is(std::cin);

  is >> s;
  std::cout << s;
  return 0;
}

哪个工作正常，直到我改变underflow方法的逻辑。主要目标是将数据保存在c-string valiable s中，这与用户输入不同。为了进行简单的测试，我将underflow方法更改为以下内容：

virtual std::streambuf::int_type underflow()
{
  std::streambuf::int_type result = buffer->sbumpc();

  if (result != traits_type::eof())
  {
    result = traits_type::to_int_type('+'); // <-- this was added

    ch = traits_type::to_char_type(result);

    setg(&ch, &ch, &ch + 1);
  }

  return result;
}

这个想法是让方法只返回+个符号而不是用户输入的字符。因此，例如，如果输入为123，我希望+++存储在变量s中。这不起作用。控制台挂起就像等待更多输入一样。只有一定数量的按键（或发送EOF）才有帮助。

我在这里缺少什么？

正如@ferosekhanj所指出的那样，问题是缺少换行符，修改后的underflow版本没有将其返回给调用者。因此，为了使代码正常工作，必须返回它。此版本的方法工作正常。

virtual std::streambuf::int_type underflow()
{
  std::streambuf::int_type result = buffer->sbumpc();

  if ((result != traits_type::eof()) && !traits_type::eq(traits_type::to_char_type(result), '\n'))
  {
     result = traits_type::to_int_type('+');

     ch = traits_type::to_char_type(result);

     setg(&ch, &ch, &ch + 1);
  }

  return result;
}

Answer 1

从我以前的旧C ++体验来看，流buf是流的底层缓冲区。当流需要更多数据时，它会调用下溢。在此方法中，您可以从源和setg中读取。当流有数据要写回源时，它会调用溢出。在此方法中，您从流中读取，写回源和setp。例如，如果您正在从streambuf中的套接字读取数据

socketbuf::int_type socketbuf::underflow(){
  int bytesRead = 0;
  try{
    bytesRead = soc->read(inbuffer,BUFFER_SIZE-1,0);
    if( bytesRead <= 0 ){
      return traits_type::eof();
    }
  }catch(IOException ioe){
    cout<<"Unable to read data"<<endl;
    return traits_type::eof();
  }
  setg(inbuffer,inbuffer,inbuffer+bytesRead);
  return traits_type::to_int_type(inbuffer[0]);
}

socketbuf::int_type socketbuf::overflow(socketbuf::int_type c){
  int bytesWritten = 0;
  try{
    if(pptr() - pbase() > 0){
      bytesWritten = soc->write(pbase(),(pptr() - pbase()),0);
      if( bytesWritten <= 0 )  return traits_type::not_eof(c);
    }
  }catch(IOException ioe){
    cout<<"Unable to write data"<<endl;
    return traits_type::eof();
  }
  outbuffer[0] = traits_type::to_char_type(c);
  setp(outbuffer,outbuffer+1,outbuffer+BUFFER_SIZE);
  return traits_type::not_eof(c);
}

现在来看你的代码，你添加了

result = traits_type::to_int_type('+'); // <-- this was added

流读取字符串，直到看到LF（换行符）。因此，当LF字符出现时，您将使用“+”编写，因此流将永远等待（对于LF）。通过添加此检查，您的代码应该按照您的预期进行操作。如果输入'abc'

，则输出'+++'

if (result != 10)// <-- add this in addition
    result = traits_type::to_int_type('+'); // <-- this was added

希望它对你有所帮助。

自定义输入流。流缓冲和下溢方法

1 个答案: