fixes memory leak in facade phrase table2

1 files changed, 24 insertions, 0 deletions

diff --git a/src/storage/facade_phrase_table2.h b/src/storage/facade_phrase_table2.h
index 3f40f57..b9bc9ca 100644
--- a/src/storage/facade_phrase_table2.h
+++ b/src/storage/facade_phrase_table2.h
@@ -38,6 +38,18 @@ private:
     PhraseLargeTable2 * m_system_phrase_table;
     PhraseLargeTable2 * m_user_phrase_table;
 
+    void reset(){
+        if (m_system_phrase_table) {
+            delete m_system_phrase_table;
+            m_system_phrase_table = NULL;
+        }
+
+        if (m_user_phrase_table) {
+            delete m_user_phrase_table;
+            m_user_phrase_table = NULL;
+        }
+    }
+
 public:
     /**
      * FacadePhraseTable2::FacadePhraseTable2:
@@ -51,6 +63,16 @@ public:
     }
 
     /**
+     * FacadePhraseTable2::~FacadePhraseTable2:
+     *
+     * The destructor of the FacadePhraseTable2.
+     *
+     */
+    ~FacadePhraseTable2() {
+        reset();
+    }
+
+    /**
      * FacadePhraseTable2::load:
      * @system: the memory chunk of the system phrase table.
      * @user: the memory chunk of the user phrase table.
@@ -60,6 +82,8 @@ public:
      *
      */
     bool load(MemoryChunk * system, MemoryChunk * user) {
+        reset();
+
         bool result = false;
         if (system) {
             m_system_phrase_table = new PhraseLargeTable2;