From 10c95b9a48fc002889fcf21405355a4fdbade903 Mon Sep 17 00:00:00 2001
From: xulongteng <xulongteng@baidu.com>
Date: Mon, 17 Jun 2019 16:17:28 +0800
Subject: [PATCH] Fix memory leak

Change-Id: I5297891f3002f851c3eb0988205f2abf108572f5
---
 predictor/framework/memory.cpp | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/predictor/framework/memory.cpp b/predictor/framework/memory.cpp
index 144094e5..78a46434 100644
--- a/predictor/framework/memory.cpp
+++ b/predictor/framework/memory.cpp
@@ -28,6 +28,16 @@ struct MempoolRegion {
 
   im::fugue::memory::Region* _region;
   im::Mempool* _mempool;
+  ~MempoolRegion() {
+    if (_region) {
+      delete _region;
+      _region = NULL;
+    }
+    if (_mempool) {
+      delete _mempool;
+      _mempool = NULL;
+    }
+  }
 };
 
 int MempoolWrapper::initialize() {
@@ -57,6 +67,7 @@ int MempoolWrapper::thread_initialize() {
     LOG(ERROR) << "unable to set the thrd_data";
     delete region;
     delete mempool;
+    delete mempool_region;
     return -1;
   }
 
-- 
GitLab