C++11自定义非线程安全的std::shared_ptr

C++ Templates一书中的源代码,这个作者自己写了一个类似std::shared_ptr的非原子的引用计数实现。
当然没有std::shared_ptr牛笔,但是可以看出所有权转让和引用计数的原理。

CMakeLists.txt

cmake_minimum_required(VERSION 2.6)

if(APPLE)
    message(STATUS "This is Apple, do nothing.")
elseif(UNIX)
    message(STATUS "This is linux, set CMAKE_PREFIX_PATH.")
    set(CMAKE_PREFIX_PATH /vcpkg/ports/cppwork/vcpkg_installed/x64-linux/share)
endif(APPLE)

project(count_ptr)

add_definitions(-std=c++17)

add_definitions(-g)

find_package(ZLIB)

find_package(glog REQUIRED)

find_package(OpenCV REQUIRED )

find_package(Boost REQUIRED COMPONENTS
    system
    filesystem
    serialization
    program_options
    thread
    )

find_package(DataFrame REQUIRED)

if(APPLE)
    MESSAGE(STATUS "This is APPLE, set INCLUDE_DIRS")
set(INCLUDE_DIRS ${Boost_INCLUDE_DIRS} /usr/local/include /usr/local/iODBC/include /opt/snowflake/snowflakeodbc/include/ ${CMAKE_CURRENT_SOURCE_DIR}/../../include/ {CMAKE_CURRENT_SOURCE_DIR}/../)
elseif(UNIX)
    MESSAGE(STATUS "This is linux, set INCLUDE_DIRS")
    set(INCLUDE_DIRS ${Boost_INCLUDE_DIRS} /usr/local/include ${CMAKE_CURRENT_SOURCE_DIR}/../../include/ ${CMAKE_CURRENT_SOURCE_DIR}/../)
endif(APPLE)


if(APPLE)
    MESSAGE(STATUS "This is APPLE, set LINK_DIRS")
    set(LINK_DIRS /usr/local/lib /usr/local/iODBC/lib /opt/snowflake/snowflakeodbc/lib/universal)
elseif(UNIX)
    MESSAGE(STATUS "This is linux, set LINK_DIRS")
    set(LINK_DIRS ${Boost_INCLUDE_DIRS} /usr/local/lib /vcpkg/ports/cppwork/vcpkg_installed/x64-linux/lib)
endif(APPLE)

if(APPLE)
    MESSAGE(STATUS "This is APPLE, set ODBC_LIBS")
    set(ODBC_LIBS iodbc iodbcinst)
elseif(UNIX)
    MESSAGE(STATUS "This is linux, set LINK_DIRS")
    set(ODBC_LIBS odbc odbcinst ltdl)
endif(APPLE)

include_directories(${INCLUDE_DIRS})
LINK_DIRECTORIES(${LINK_DIRS})

file( GLOB main_file_list ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp) 

file( GLOB APP_SOURCES  ${CMAKE_CURRENT_SOURCE_DIR}/*.h ${CMAKE_CURRENT_SOURCE_DIR}/*.hpp ${CMAKE_CURRENT_SOURCE_DIR}/../../include/http/impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../include/yaml/impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../include/df/impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../include/death_handler/impl/*.cpp)

add_library(${PROJECT_NAME}_lib SHARED ${APP_SOURCES})
target_link_libraries(${PROJECT_NAME}_lib ${Boost_LIBRARIES} ZLIB::ZLIB glog::glog DataFrame::DataFrame ${OpenCV_LIBS})
target_link_libraries(${PROJECT_NAME}_lib  ssl crypto libgtest.a pystring libyaml-cpp.a libgmock.a ${ODBC_LIBS} libnanodbc.a pthread dl backtrace)

foreach( main_file ${main_file_list} )
    file(RELATIVE_PATH filename ${CMAKE_CURRENT_SOURCE_DIR} ${main_file})
    string(REPLACE ".cpp" "" file ${filename})
    add_executable(${file}  ${main_file})
    target_link_libraries(${file} ${PROJECT_NAME}_lib)
endforeach( main_file ${main_file_list})

stdobjpolicy.hpp

#ifndef _FREDRIC_STD_OBJ_POLICY_HPP_
#define _FREDRIC_STD_OBJ_POLICY_HPP_

// 标准对象销毁器
class StandardObjectPolicy {
    public:
        template <typename T>
        void dispose(T* object) {
            delete object;
        }
};

#endif

stdarraypolicy.hpp

#ifndef _FREDRIC_STD_ARRAY_POLICY_HPP_
#define _FREDRIC_STD_ARRAY_POLICY_HPP_

// 标准对象销毁器
class StandardArrayPolicy {
    public:
        template <typename T>
        void dispose(T* array) {
            delete [] array;
        }
};

#endif

simplerefcount.hpp

#ifndef _FREDRIC_SIMPLE_REF_COUNT_HPP_
#define _FREDRIC_SIMPLE_REF_COUNT_HPP_

#include "allocator.hpp"

#include <cstddef>

// 简单引用计数策略类
class SimpleReferenceCount {
    private:
        std::size_t* counter;
    
    public:
        SimpleReferenceCount() {
            counter = nullptr;
        }

        template <typename T>
        void init(T*) {
            counter = alloc_counter();
            *counter = 1;
        }

        template <typename T>
        void dispose(T*) {
            dealloc_counter(counter);
        }

        template <typename T>
        void increment(T*) {
            ++*counter;
        }

        template <typename T>
        void decrement(T*) {
            --*counter;
        }

        template <typename T>
        bool is_zero(T*) {
            return *counter == 0;
        }
};

#endif

memberrefcount.hpp

#ifndef _FREDRIC_MEMBER_REF_COUNT_HPP_
#define _FREDRIC_MEMBER_REF_COUNT_HPP_

template <typename ObjectT,
          typename CountT,
          CountT ObjectT::*CountP>
class MemberReferenceCount {
    public: 
        void init(ObjectT* object) {
            object->*CountP = 1;
        }

        // dispose不需要做额外操作
        void dispose (ObjectT*) {

        }

        void increment(ObjectT* object) {
            ++object->*CountP;
        }

        void decrement(ObjectT* object) {
            --object->*CountP;
        }


        bool is_zero(ObjectT* object) {
            return object->*CountP == 0; 
        }
};

#endif

countptr.hpp

#ifndef _FREDRIC_COUNT_PTR_HPP_
#define _FREDRIC_COUNT_PTR_HPP_

#include "simplerefcount.hpp"
#include "stdobjpolicy.hpp"


template <typename T,
          typename CountPolicy = SimpleReferenceCount,
          typename ObjectPolicy = StandardObjectPolicy>
class CountingPtr: private CountPolicy, private ObjectPolicy {
    private:
        using CP = CountPolicy;
        using OP = ObjectPolicy;

        T* object_pointed_to;

    public:
        CountingPtr() {
            this->object_pointed_to = nullptr;
        }

        explicit CountingPtr(T* p) {
            this->init(p);   
        }

        CountingPtr(CountingPtr<T, CP, OP> const& cp):
            CP((CP const&)cp),
            OP((OP const&)cp) {
                this->attach(cp);
        }

        ~CountingPtr() {
            this->detach();
        }

        CountingPtr<T, CP, OP>& operator=(T* p) {
            assert(p != this->object_pointed_to);
            this->detach();

            this->init(p);
            return *this;
        }

        CountingPtr<T, CP, OP>& operator=(CountingPtr<T, CP, OP> const& cp) { 
            if(this->object_pointed_to != cp.object_pointed_to) {
                this->detach();

                CP::operator=((CP const&)cp);
                OP::operator=((OP const&)cp);
                this->attach(cp);
            }
            return *this;
        }

        T* operator->() const {
            return this->object_pointed_to;
        }

        T& operator*() const {
            return *this->object_pointed_to;
        }

        T* get() const {
            return this->object_pointed_to;
        }
    
    private:
        void init(T* p) {
            if(p!=nullptr) {
                CountPolicy::init(p);
            }
            this->object_pointed_to = p;
        }

        // 拷贝指针并增加引用计数
        void attach(CountingPtr<T, CP, OP> const& cp) {
            this->object_pointed_to = cp.object_pointed_to;
            if(cp.object_pointed_to != nullptr) {
                CountPolicy::increment(cp.object_pointed_to);
            }
        }

        // 减少引用计数,并销毁引用计数和对象,如果是最后一个持有引用计数的对象
        void detach() {
            if(this->object_pointed_to != nullptr) {
                CountPolicy::decrement(this->object_pointed_to);
                if(CountPolicy::is_zero(this->object_pointed_to)) {
                    // 销毁counter
                    CountPolicy::dispose(this->object_pointed_to);
                    // 销毁指向的对象
                    ObjectPolicy::dispose(this->object_pointed_to);
                }
            }
        }

};

#endif

allocator.hpp

#ifndef _FREDRIC_ALLOCATOR_HPP_
#define _FREDRIC_ALLOCATOR_HPP_

#include <cstddef>

std::size_t* alloc_counter() {
    return ::new std::size_t;
}

void dealloc_counter(std::size_t* ptr) {
    ::delete ptr;
}

#endif

main.cpp

#include "stdobjpolicy.hpp"
#include "stdarraypolicy.hpp"
#include "simplerefcount.hpp"
#include "memberrefcount.hpp"
#include "countptr.hpp"

#include <iostream>
#include <vector>

class MyObject {
    int count{0};
    
    public:
        using Ptr = CountingPtr<MyObject, MemberReferenceCount<MyObject, int, &MyObject::count>>;

        void do_something() {
            std::cout << "Do something ..." << std::endl;
        }
};

void test1();
void test2();

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

    return EXIT_SUCCESS;
}

void test1() {
    std::cout << "\ntest1()\n";
    CountingPtr<int> p0;

    {
        CountingPtr<int> p1 {new int(42)};
        std::cout << "*p1 = " << *p1 << std::endl;

        *p1 = 17;
        std::cout << "*p1 = " << *p1 << std::endl;

        CountingPtr<int> p2 = p1;
        std::cout << "*p2 = " << *p2 << std::endl;

        *p1 = 33;
        std::cout << "*p2 = " << *p2 << std::endl;

        p0 = p2;
        std::cout << "*p0 = " << *p0 << std::endl;

        ++*p0;
        ++*p1;
        ++*p2;

        std::cout << "*p0 = " << *p0 << std::endl;
        std::cout << "*p1 = " << *p1 << std::endl;
        std::cout << "*p2 = " << *p2 << std::endl;
    }

    std::cout << "After block: *p0 = " << *p0 << std::endl;
}

void test2() {
    std::cout << "\ntest2()\n";
    
    MyObject::Ptr obj_ptr {new MyObject};
    obj_ptr->do_something();

    CountingPtr<int,SimpleReferenceCount, StandardArrayPolicy> int_arr {
        new int[10]
    };

    auto arr = int_arr.get();
    for(int i=0; i<10; ++i) {
        arr[i] = i + 1;
    }

    for(int i=0; i<10; ++i) {
       std::cout << arr[i] << " ";
    }

    std::cout << "\n";
}

程序输出如下,


image.png
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 222,252评论 6 516
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 94,886评论 3 399
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 168,814评论 0 361
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,869评论 1 299
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,888评论 6 398
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,475评论 1 312
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 41,010评论 3 422
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,924评论 0 277
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,469评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,552评论 3 342
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,680评论 1 353
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 36,362评论 5 351
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 42,037评论 3 335
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,519评论 0 25
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,621评论 1 274
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 49,099评论 3 378
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,691评论 2 361

推荐阅读更多精彩内容