歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
Linux教程網 >> Linux編程 >> Linux編程 >> Linux C++循環緩沖區模板類

Linux C++循環緩沖區模板類

日期:2017/3/1 9:11:45   编辑:Linux編程

一:概述

實際學習和工作中,我們經常會遇到讀寫大量數據的情況,這個時候我們可能就用到了循環緩沖區。

循環緩沖區在處理大量數據的時候有很大的優點,循環緩沖區在一些競爭問題上提供了一種免鎖的機制,免鎖的前提是,生產者和消費都只有一個的情況下,否則也要加鎖。

二:循環緩沖區的實現理論如下圖

三:實現代碼如下所示:

//CRecycleQueue.h

#include<iostream>

//循環緩沖區類模板
template<class T>
class CRecycleQueue{

private:
//循環緩沖區地址指針
T **_queue;
//循環緩沖區讀游標 (讀的位置)
int _read;
//循環緩沖區寫游標 (寫的位置)
int _write;
//循環緩沖區的大小
int _size;
//我們姑且稱這個變量為掩碼,接下來用來作位&運算,從而實現循環緩沖
int _mask;

public:
CRecycleQueue(){
_queue = NULL;
_read = 0;
_write = 0;
_size = 0;
_mask = 0;
}
//初始化循環緩沖區
bool InitRecycleQueue(int exp){

if(0 > exp){

return false;
}

_read = 0;
_write = 0;
//傳進來一個整數,對1進行位移操作
//比如exp = 4
//_size的二進制表示:1000
_size = 1 << exp;
//_mask的二進制表示:0111
_mask = _size - 1;
//分配緩沖區空間
_queue = (T **)new char[sizeof (T *) * _size];

if(NULL == _queue){

return false;
}

return true;
}
/*
* size = 1000 mask = 0111
* write或read同mask 作&運算,可以實現循環緩沖區的功能
* 也許你會問這裡為什麼不使用 % 運算實現循環的循環功能呢?
* 答案是系統 & 運算效率要比 % 運算效率高
*
* Push:
* write = 0;
* 0000 & 0111 = 0; write++ (寫入緩沖隊列的第0位置)
* write = 1;
* 0001 & 0111 = 1; write++ (寫入緩沖隊列的第1位置)
* write = 2;
* 0010 & 0111 = 2; write++
* write = 3;
* 0011 & 0111 = 3; write++
* ...
* write = 8;
* 1000 & 0111 = 0; write++
* write = 9;
* 1001 & 0111 = 1; write++
* ...
*
* Pop:
* read = 0;
* 0000 & 0111 = 0; read++ (讀取緩沖隊列的第0位置的數據)
* read = 1;
* 0001 & 0111 = 1; read++ (讀取緩沖隊列的第1位置的數據)
* read = 2;
* 0010 & 0111 = 2; read++
* read = 3
* 0011 & 0111 = 3; read++
* ...
* read = 8;
* 1000 & 0111 = 0; read++
* ...
* */

bool Push(T *type){

if(NULL == type){

return false;
}
//當條件不滿足的時候,說明緩沖區已滿,Push進來的數據就會丟失
if(_write < _read + _size){
//我們這裡存入的是type指針,這個指針指向了一個我們分配的內存空間或者類
_queue[_write & _mask] = type;
_write++;
return true;
}

return false;
}

T *Pop(){

T *tmp = NULL;
//當條件不滿足的時候說明緩沖區已經沒有數據
if(_read < _write){
//取出隊列的數據
tmp = _queue[_read & _mask];
_read++;
}

return tmp;
}

int GetRemainSize(){

return (_write - _read);
}
};

下面是簡單的測試代碼:

//main.cpp

#include <iostream>
#include <pthread.h>
#include "CRecycleQueue.h"
using namespace std;

class UserInfo{

private :
int _num;

public:
UserInfo(int num){

_num = num;
}

int getUserNum(){

return _num;
}
};

CRecycleQueue<UserInfo> *queue = new CRecycleQueue<UserInfo>;

void *write_func(void *args){

int num = 0;
while(1){

//UserInfo裡可以封裝你自己想要的數據
//這裡僅僅是一個簡單的測試用例
UserInfo *info = new UserInfo(num++);

if(!queue->Push(info)){
//Push失敗 刪除手動分配的內存空間
delete info;
}
sleep(1);
}
}

void *read_func(void *args){

while(1){

UserInfo *info = NULL;
if(info = queue->Pop()){

cout<<info->getUserNum()<<endl;
delete info;
}
sleep(1);
}
}
int
main(){

queue->InitRecycleQueue(8);

pthread_t pid1;
pthread_t pid2;
//這種生產者和消費者都只有一個的情況下,這個循環緩沖區為競爭問題提供了免鎖,大大提高了程序的處理效率
pthread_create(&pid1,NULL,read_func,NULL);
pthread_create(&pid2,NULL,write_func,NULL);

pthread_join(pid1,NULL);
pthread_join(pid2,NULL);

return 0;
}

編譯:g++ main.cpp -lpthread -o test

這個循環緩沖隊列大體的功能已經實現,其中循環緩沖隊列一些其他操作並沒有去實現,只是描述了一些核心的操作!

如果有錯誤和其他意見,提出來大家一起相互討論和學習!

Copyright © Linux教程網 All Rights Reserved