C++: streambuf

streambuf定義了對流緩沖區(qū)操作的接口炬转,主要分為輸出緩沖流和輸入緩沖流。streambuf有兩種用法,一是直接使用各個接口馆截,二是繼承并實現(xiàn)新的I/O channels。

1. 輸出緩沖流:
  • pbase(): put base或悲,輸出緩沖流的首指針孙咪。
  • pptr(): put pointer堪唐,當前可寫位置
  • epptr(): end put pointer巡语,輸出緩沖流的尾指針+1,即one past the last character淮菠。

函數(shù):
setp(char new_pbase, char new_epptr)**
注:設置pbase,pptr,epptr三個指針的位置:

  • pbase()=new_pbase;
  • pptr()=bew_pbase;
  • epptr()=new_epptr;

pbump(int offset)
注: 移動pptr offset個位置男公。當我們寫入num個數(shù)據(jù),pbump(num)合陵。

overflow
當緩沖區(qū)滿時的操作枢赔;
pptr() == epptr()

2. 輸入緩沖流:
  • eback(): 指向輸入緩沖流的首指針
  • gptr(): 當前輸入緩沖流的位置
  • egptr(): 輸入緩沖流的末尾
  • gbump(): 移動gptr
  • setg(): 設置eback/gptr/egptr的位置

underflow()
當無緩沖區(qū)數(shù)據(jù)可讀時。
cond: gptr()=egptr()

setg
setg (eback_base,new_gptr, new_egptr)
分別設置:

  • eback
  • gptr
  • egptr
3. 基于streambuf自定義一個輸入輸出緩沖流

prebufferstream.hpp:

/*  Author: zhangwen1@unionpay.com */
/*  Date: 2020-11-24               */

/*
 * Provide Custom Streambufeer Example 
*/ 

#ifndef PREBUFFERSTREAM_HPP
#define PREBUFFERSTREAM_HPP

#include <streambuf>
#include <cstdio>
#include <exception>

class prebuffer: public std::streambuf{

public:
  typedef std::char_traits<char> traits_ty;
  typedef traits_ty::int_type int_type;
  typedef traits_ty::pos_type pos_type;
  typedef traits_ty::off_type off_type;

private:
  int m_mode;
  int m_filedes;

  static const int s_buf_size = 4092;
  static const int s_pback_size = 4;
  char buffer[s_buf_size];

  void init(int fd, int om, bool throw_exception);

  int flushoutput();

public:

  prebuffer(int fd, int om, bool throw_exception=false);
  prebuffer(const prebuffer&)=delete;
  prebuffer& operator=(const prebuffer&)=delete;
  prebuffer(prebuffer&&)=delete;
  prebuffer& operator=(prebuffer&&)=delete;

  ~prebuffer(){close();}

  virtual int_type overflow(int_type c) override;
  virtual int_type underflow() override;

  bool is_open() const { return m_filedes >= 0; }

  void close();
  virtual int sync() override;

  std::string get_str(){
    return std::string(buffer,this->pptr()-buffer);
  }

};


class FileError: public std::exception{
  const char* what(void) const throw(){
    return "Invaild fd";
  }
};

#endif

prebufferstream.cc:

#include "prebufferstream.hpp"

#include <iostream>
#include <unistd.h>
#include <cstring>

void prebuffer::init(int fd, int om, bool throw_exception){
  m_filedes = fd;
  if (m_filedes > 0){
    m_mode = om;
  }
  if (throw_exception && !is_open()){
      throw FileError();
  }
  setg(buffer+s_pback_size,
       buffer+s_pback_size,
       buffer+s_pback_size);
  setp(buffer,
       buffer+s_buf_size-1);
}

prebuffer::prebuffer(int fd, int om, bool throw_exception){
  init(fd,om,throw_exception);
}

void prebuffer::close(){
  if (is_open()){
    this->sync();
    m_mode = 0;
    m_filedes = -1;
  }
}

prebuffer::int_type prebuffer::underflow(){

  if (gptr() < egptr()){
    return static_cast<unsigned char>(*gptr());
  }
  int num_putback=0;
  if (s_pback_size > 0){
    num_putback = gptr()-eback();
    if (num_putback > s_pback_size){
      num_putback = s_pback_size;
    }
    std::memcpy(buffer+s_pback_size-num_putback,gptr()-num_putback,num_putback);
  }

  const int num = ::read(m_filedes,buffer+s_pback_size,s_buf_size-s_pback_size);

  if (num <= 0){return EOF;}
  setg(buffer+s_pback_size-num_putback,buffer+s_pback_size,buffer+s_pback_size+num);

  return static_cast<unsigned char>(*gptr());

}

prebuffer::int_type prebuffer::overflow(int_type c){
  std::cout<<"Overflow Happend"<<std::endl;
  if (!(m_mode & std::ios::out) || !is_open()) return EOF;
  if (c != EOF){
    *pptr() = c;
    pbump(1);
  }
  if (flushoutput() == EOF)
    {
      return -1; 
    }

  return c;
}

int prebuffer::flushoutput(){
  if (!(m_mode & std::ios::out) || !is_open()) return EOF;
  size_t num = pptr()-pbase();
  if (::write(m_filedes,buffer,num) == -1){
    return EOF;
  }
  pbump(-num);
  return num;
}

int prebuffer::sync(){
  if (flushoutput() == EOF){
    return -1;
  }
  return 0;
}

測試, main.cc:

#include "prebufferstream.hpp"
#include <fcntl.h> 
#include <ostream>
#include <istream>
#include <unistd.h>
#include <iostream>

int main(int argc,char* argv[]){

  std::string file_name1 = "test";
  std::string file_name2 = "intest";
  int fdout;
  int fdin;
  if ((fdout=open(file_name1.c_str(),O_RDWR | O_CREAT | O_APPEND)) < 0){
    std::cout<<"ERROR Open File"<<std::endl;
  }
  if ((fdin=open(file_name2.c_str(),O_RDONLY | O_CREAT | O_APPEND)) < 0){
    std::cout<<"ERROR Open File"<<std::endl;
  }


  prebuffer ot{fdout,std::ios::out};
  prebuffer it{fdin,std::ios::in};
  std::ostream dout(&ot);
  std::istream dint(&it);
  dout<<1<<2<<3<<4<<5<<"\n";

  std::cout<<ot.get_str()<<std::endl;

  ot.sync();
  int line;
  while (dint>>line){
    std::cout<<line<<" ";
  }

  close(fdout);
  close(fdin);

  return 0;
}
?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末拥知,一起剝皮案震驚了整個濱河市踏拜,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌低剔,老刑警劉巖速梗,帶你破解...
    沈念sama閱讀 206,378評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件肮塞,死亡現(xiàn)場離奇詭異,居然都是意外死亡姻锁,警方通過查閱死者的電腦和手機枕赵,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,356評論 2 382
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來位隶,“玉大人拷窜,你說我怎么就攤上這事〗Щ疲” “怎么了篮昧?”我有些...
    開封第一講書人閱讀 152,702評論 0 342
  • 文/不壞的土叔 我叫張陵,是天一觀的道長笋妥。 經(jīng)常有香客問我恋谭,道長,這世上最難降的妖魔是什么挽鞠? 我笑而不...
    開封第一講書人閱讀 55,259評論 1 279
  • 正文 為了忘掉前任疚颊,我火速辦了婚禮,結果婚禮上信认,老公的妹妹穿的比我還像新娘材义。我一直安慰自己,他們只是感情好嫁赏,可當我...
    茶點故事閱讀 64,263評論 5 371
  • 文/花漫 我一把揭開白布其掂。 她就那樣靜靜地躺著,像睡著了一般潦蝇。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上贤牛,一...
    開封第一講書人閱讀 49,036評論 1 285
  • 那天,我揣著相機與錄音般卑,去河邊找鬼蝠检。 笑死叹谁,一個胖子當著我的面吹牛本慕,可吹牛的內(nèi)容都是我干的锅尘。 我是一名探鬼主播,決...
    沈念sama閱讀 38,349評論 3 400
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼议街!你這毒婦竟也來了特漩?” 一聲冷哼從身側響起涂身,我...
    開封第一講書人閱讀 36,979評論 0 259
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎悴能,沒想到半個月后漠酿,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體记靡,經(jīng)...
    沈念sama閱讀 43,469評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡空凸,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 35,938評論 2 323
  • 正文 我和宋清朗相戀三年紊选,在試婚紗的時候發(fā)現(xiàn)自己被綠了兵罢。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,059評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖即横,靈堂內(nèi)的尸體忽然破棺而出裆赵,到底是詐尸還是另有隱情战授,我是刑警寧澤植兰,帶...
    沈念sama閱讀 33,703評論 4 323
  • 正文 年R本政府宣布钉跷,位于F島的核電站爷辙,受9級特大地震影響膝晾,放射性物質(zhì)發(fā)生泄漏血当。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 39,257評論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望佳谦。 院中可真熱鬧钻蔑,春花似錦咪笑、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,262評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽剃法。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間钦椭,已是汗流浹背碑诉。 一陣腳步聲響...
    開封第一講書人閱讀 31,485評論 1 262
  • 我被黑心中介騙來泰國打工彪腔, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人进栽。 一個月前我還...
    沈念sama閱讀 45,501評論 2 354
  • 正文 我出身青樓德挣,卻偏偏與公主長得像,于是被迫代替她去往敵國和親快毛。 傳聞我的和親對象是個殘疾皇子格嗅,可洞房花燭夜當晚...
    茶點故事閱讀 42,792評論 2 345

推薦閱讀更多精彩內(nèi)容

  • [轉]C/C++ 文件讀寫操作總結 在編程的過程中,文件的操作是一個經(jīng)常用到的問題唠帝,在C++Builder中屯掖,可以...
    天之道天知道閱讀 5,187評論 0 7
  • I/O流的概念 程序的輸入指的是從輸入文件將數(shù)據(jù)傳送給程序,程序的輸出指的是從程序將數(shù)據(jù)傳送給輸出文件没隘。 C++輸...
    青春豬頭少年_閱讀 485評論 0 0
  • ??在c++中引入了stream懂扼,一開始接觸這個的時候感覺無法正確的理解stream的用法,在寫項目的時候要用到讀...
    UnderwoodY閱讀 12,235評論 0 5
  • 轉載(https://www.cnblogs.com/MrYuan/p/5383408.html) 基于C的文件操...
    送分童子笑嘻嘻閱讀 438評論 0 0
  • C/C++輸入輸出流總結 前兩天寫C++實習作業(yè)右蒲,突然發(fā)現(xiàn)I/O是那么的陌生阀湿,打了好長時間的文件都沒有打開,今天終...
    LuckTime閱讀 1,720評論 0 6