[log_vtab_manager] fix lookup_impl()

tstack · Apr 9, 2017 · aae3cf4 · aae3cf4
1 parent fea3cb8
commit aae3cf4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/lnav.cc b/src/lnav.cc
@@ -721,8 +721,7 @@ static void open_schema_view(void)
     schema += "\n\n-- Virtual Table Definitions --\n\n";
     schema += ENVIRON_CREATE_STMT;
     schema += vtab_module_schemas;
-    for (log_vtab_manager::iterator vtab_iter =
-            lnav_data.ld_vtab_manager->begin();
+    for (auto vtab_iter = lnav_data.ld_vtab_manager->begin();
          vtab_iter != lnav_data.ld_vtab_manager->end();
          ++vtab_iter) {
         schema += "\n" + vtab_iter->second->get_table_statement();

diff --git a/src/log_vtab_impl.hh b/src/log_vtab_impl.hh
@@ -202,9 +202,15 @@ public:
     std::string register_vtab(log_vtab_impl *vi);
     std::string unregister_vtab(intern_string_t name);
 
-    log_vtab_impl *lookup_impl(intern_string_t name)
+    log_vtab_impl *lookup_impl(intern_string_t name) const
     {
-        return this->vm_impls[name];
+        log_vtab_impl *retval = nullptr;
+        auto iter = this->vm_impls.find(name);
+
+        if (iter != this->vm_impls.end()) {
+            retval = iter->second;
+        }
+        return retval;
     };
 
     iterator begin() const