Merge pull request #1373 from joto/print-memory-usage

Print memory usage at the end of running osm2pgsql

Author: lonvia

src/osm2pgsql.cpp

@@ -35,6 +35,8 @@
 #include "util.hpp"
 #include "version.hpp"
 
+#include <osmium/util/memory.hpp>
+
 #include <exception>
 #include <memory>
 
@@ -72,17 +74,26 @@ int main(int argc, char *argv[])
 
         // Processing: In this phase the input file(s) are read and parsed,
         // populating some of the tables.
-        process_files(files, osmdata, options.append, get_logger().show_progress());
+        process_files(files, osmdata, options.append,
+                      get_logger().show_progress());
 
         // Process pending ways and relations. Cluster database tables and
         // create indexes.
         osmdata.stop();
 
+        // Output overall memory usage. This only works on Linux.
+        osmium::MemoryUsage mem;
+        if (mem.peak() != 0) {
+            log_debug("Overall memory usage: peak={}MByte current={}MByte",
+                      mem.peak(), mem.current());
+        }
+
         log_info("Osm2pgsql took {} overall.",
                  util::human_readable_duration(timer_overall.stop()));
     } catch (std::exception const &e) {
         log_error("{}", e.what());
         return 1;
     }
+
     return 0;
 }