From 91d4ed5fdfc9741e10c62c35894ba5f3f20b52b2 Mon Sep 17 00:00:00 2001
From: Kishore Nallan <kishore@kishorelive.com>
Date: Sun, 17 Dec 2017 15:40:57 +0530
Subject: [PATCH] Free custom generator after collection export is done.

---
 src/http_server.cpp | 2 ++
 src/main/main.cpp   | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/http_server.cpp b/src/http_server.cpp
index 83ae5099..efb4a820 100644
--- a/src/http_server.cpp
+++ b/src/http_server.cpp
@@ -372,6 +372,7 @@ void HttpServer::response_proceed(h2o_generator_t *generator, h2o_req_t *req) {
         h2o_dispose_request(req);
         delete custom_generator->req_res.req;
         delete custom_generator->req_res.response;
+        delete custom_generator;
     }
 }
 
@@ -381,6 +382,7 @@ void HttpServer::response_stop(h2o_generator_t *generator, h2o_req_t *req) {
     h2o_dispose_request(req);
     delete custom_generator->req_res.req;
     delete custom_generator->req_res.response;
+    delete custom_generator;
 }
 
 void HttpServer::stream_response(void (*handler)(http_req* req, http_res* res, void* data),
diff --git a/src/main/main.cpp b/src/main/main.cpp
index 0b3a027c..6b8ce50d 100644
--- a/src/main/main.cpp
+++ b/src/main/main.cpp
@@ -21,7 +21,7 @@ int main(int argc, char* argv[]) {
     Store *store = new Store(state_dir_path);
 
     CollectionManager & collectionManager = CollectionManager::get_instance();
-    collectionManager.init(store, "abcd");
+    collectionManager.init(store, "abcd", "123");
 
     std::vector<field> fields_to_index = {
             field("lang", field_types::STRING, true),