큐 확장

큐를 확장해 보자. 어떤 경우에 사용한지 알아 보자.

다음과 같은 구조체가 있고 stl 큐를 사용하고 있다.
push , pop을 여러군데 사용하고 있다.
여기서 프로그램 유연성을 확보하기 위해 배열 대신 new, delete로 메모리 할당 방식으로 바꾸어 보자.

struct MemData
{
char data[100];
};

메모리 할당 방식으로 바꾸고 복사 생성자를 만든다.

#define BUFF_SIZE 4

struct MemData
{
    char* data;
    MemData& operator =(const MemData& v)
    {
        memcpy(this->data, v.data, BUFF_SIZE);
        return (*this);
    }
};

복사 생성자를 만드는 이유는 기존 코드에서 다음과 같이 사용하고 있기 때문이다.

MemData a;
MemData b;

a = b;

구조체에서 배열을 사용하고 있으면 알아서 복사 해준다.
메모리 방식으로 바꾸면 포인터만 전달 되기 때문에 복사 생성자를 만들어 버퍼를 복사한다.

stl queue의 push, pop일때는 메모리가 복사 안된다.
push를 여러번 하면 메모리 포인터만 전달된다.

push, pop 할때 메모리 생성, 제거, 복사를 구현해줘야 한다.
처음에는 함수를 만들어 해결할려고 했다. 기존에 push, pop이 너무 많고 QA 할양이 많고 쉽지도 않기 때문에
최소한으로 고칠수 있는 방법을 찾아 보았다.

MemData 전용 큐를 만들었다.
push 할때 메모리를 생성하고
"d = element"는 복사 생성자에 의해 메모리를 복사한다.
pop 할때 메모리를 해제한다.

struct QueueExt : std::queue<MemData>
{
    void push(const MemData& element)
    {
        MemData d;
        d.data = new char[BUFF_SIZE];
        d = element;
        std::queue<MemData>::push(d);
    }

    void pop()
    {
        MemData d = std::queue<MemData>::front();
        if(d.data)
            delete [] d.data;
        std::queue<MemData>::pop();
    }
};

QueueExt gQueue;

큐 선언만 수정하면 끝이다.

전체 예제 코드는 다음과 같다.

#include <stdio.h>
#include <queue>

#define BUFF_SIZE 4

struct MemData
{
    char* data;
    MemData& operator =(const MemData& v)
    {
        memcpy(this->data, v.data, BUFF_SIZE);
        return (*this);
    }
};

struct QueueExt : std::queue<MemData>
{
    void push(const MemData& element)
    {
        MemData d;
        d.data = new char[BUFF_SIZE];
        d = element;
        std::queue<MemData>::push(d);
    }

    void pop()
    {
        MemData d = std::queue<MemData>::front();
        if(d.data)
            delete [] d.data;
        std::queue<MemData>::pop();
    }
};

QueueExt gQueue;

int main(int argc, char* argv[])
{
    MemData a;
    MemData b;
    a.data = new char[4];
    b.data = new char[4];

    b.data[0] = 77;
    b.data[1] = 88;
    b.data[2] = 99;
    b.data[3] = 33;

    memset(a.data, 3, 4);
    gQueue.push(b);
    a = gQueue.front();
    gQueue.pop();

    delete a.data;
    delete b.data;
    return 0;
}

참고)
https://stackoverflow.com/questions/6806000/call-function-when-items-are-added-or-removed-from-a-vector