记录下遇到的共享指针shared_ptr不恰当的使用,导致指针一直不被释放,进而内存泄漏问题
原因:bind时其中一个函数入参绑定了一个共享指针,而这个绑定后的函数对象又保存在共享指针所指向的对象里(在本例中是间接存在,即共享指针所指向的对象有个成员变量,然后这里变量也是个类对象且其中保存了bind后的函数对象)
场景是:
- 用boost网络库写服务端代码,用bind绑定了session会话对象然后生成了一个回调函数对象,用于后续请求处理时调用这个回调函数进行请求应答,将应答数据返回给客户端。
- 问题出现:当客户端关闭连接后,正常session也会释放,因为boost的底层监听列表已经移除了这个session指针,但是实际是没有释放,因为session->m_reqclient.m_funcResponse这个函数对象bind了session而且还是以共享指针的形式绑定的,所以产生了循环引用导致一直计数一直不为0
解决方法:
使用弱指针weak_ptr,即response这个函数入参由
(std::shared_ptr session, const std::string& data)
改为
(std::weak_ptr session, const std::string& data)
示例代码如下:
精简了很多代码,方便看。
#pragma once
#include <iostream>
#include <functional>
#include <memory>
class Session;
class ReqInterface;
static int dorequst(std::shared_ptr<Session> session);
class ReqInterface
{
public:
void Dorequest(std::string data) {//请求业务处理
data = data + "toresponse";
if (m_funcResponse)
{
m_funcResponse(data);//请求应答
}
}
public:
std::function<int(std::string)> m_funcResponse;
};
class Session : public std::enable_shared_from_this<Session>
{
public:
ReqInterface* m_reqclient;
public:
Session(){}
~Session(){}
void run() {//开始运行 不断监听客户端请求 有请求收到时调用on_recv
}
private:
void on_recv() {
dorequst(shared_from_this());
}
public:
std::string get_reqdata() {
return std::string();//从sokect读取请求数据
}
int do_write(const std::string& data) {
//将数据写入sokcet
return 0;
}
};
static int response(std::shared_ptr<Session> session, const std::string& data) {
session->do_write(data);
return 0;
}
static int dorequst(std::shared_ptr<Session> session)
{//请求分发
if (session->m_reqclient == nullptr) {
auto client = new ReqInterface;
client->m_funcResponse = std::bind(&response, session, std::placeholders::_1);
session->m_reqclient = client;
}
session->m_reqclient->Dorequest(session->get_reqdata());
return 0;
}
修复后的代码,改为了weak_ptr
#pragma once
#include <iostream>
#include <functional>
#include <memory>
class Session;
class ReqInterface;
static int dorequst(std::shared_ptr<Session> session);
class ReqInterface
{
public:
void Dorequest(std::string data) {//请求业务处理
data = data + "toresponse";
if (m_funcResponse)
{
m_funcResponse(data);//请求应答
}
}
public:
std::function<int(std::string)> m_funcResponse;
};
class Session : public std::enable_shared_from_this<Session>
{
public:
ReqInterface* m_reqclient;
public:
Session(){}
~Session(){}
void run() {//开始运行 不断监听客户端请求 有请求收到时调用on_recv
}
private:
void on_recv() {
dorequst(shared_from_this());
}
public:
std::string get_reqdata() {
return std::string();//从sokect读取请求数据
}
int do_write(const std::string& data) {
//将数据写入sokcet
return 0;
}
};
//改用弱指针 防止循环引用不能释放问题
static int response(std::weak_ptr<Session> session, const std::string& data) {
if (!session.expired())
{
session.lock()->do_write(data);
}
return 0;
}
static int dorequst(std::shared_ptr<Session> session)
{
if (session->m_reqclient == nullptr) {
auto client = new ReqInterface;
client->m_funcResponse = std::bind(&response, std::weak_ptr<Session>(session), std::placeholders::_1);
session->m_reqclient = client;
}
session->m_reqclient->Dorequest(session->get_reqdata());
return 0;
}