实现指针计数

#pragma once

 

class UseCount

{

public:

UseCount(void);

~UseCount(void);

 

UseCount(const UseCount& useCount);

 

UseCount& operator= (const UseCount& useCount);

 

bool IsOnly();

 

bool reattach(const UseCount& useCount);

 

private:

int* count;

};

#include "StdAfx.h"

#include "UseCount.h"

UseCount::UseCount(void):

count(new int(1))

{

}

UseCount::~UseCount(void)

{

if(--*count == 0)

{

delete count;

}

}

UseCount::UseCount(const UseCount& useCount)

{

count = useCount.count;

++count;

}

UseCount& UseCount::operator= (const UseCount& useCount)

{

if(--*count == 0)

{

delete count;

}

count  = useCount.count;

return *this;

}

bool UseCount::IsOnly()

{

return *count == 1;

}

bool UseCount::reattach(const UseCount& useCount)

{

++*useCount.count;

if(--*count == 0)

{

delete count;

count  = useCount.count;

return true;

}

count  = useCount.count;

return false;

}

#pragma once

#include "Vector.h"

#include "UseCount.h"

class Handle

{

public:

Handle(void);

~Handle(void);

Handle::Handle(const Handle& handle);

Handle& operator= (const Handle& handle);

private:

Vector* vec;

//int* count;

UseCount useCount;

};

#include "StdAfx.h"

#include "Handle.h"

Handle::Handle(void):

vec(new Vector)

{

}

Handle::~Handle(void)

{

if(useCount.IsOnly())

{

delete vec;

}

}

Handle::Handle(const Handle& handle):

vec(handle.vec)

{

useCount.reattach(handle.useCount);

}

Handle& Handle::operator= (const Handle& handle)

{

if(useCount.reattach(handle.useCount))

{

delete vec;

};

vec = handle.vec;

return *this;

}