fix: add a lookup function to symbol table

sillydan1 · Dec 9, 2022 · 797cdee · 797cdee
1 parent b1bf12a
commit 797cdee
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/include/symbol_table.h b/include/symbol_table.h
@@ -73,6 +73,8 @@ namespace expr {
     struct symbol_table_t : public underlying_symbol_table_t {
         auto operator+=(const symbol_table_t &) -> symbol_table_t&;
         auto operator*=(const symbol_table_t &) -> symbol_table_t&;
+        auto get(const std::string& key) const -> const symbol_value_t&;
+        auto get(const std::string& key) -> symbol_value_t&;
         auto put(const symbol_table_t &) -> symbol_table_t&;
         auto overwrite_elements(const symbol_table_t &) -> symbol_table_t&;
         auto is_overlapping(const symbol_table_t& other) -> bool;

diff --git a/src/symbol_table.cpp b/src/symbol_table.cpp
@@ -28,6 +28,14 @@ namespace expr {
         return put(other);
     }
 
+    auto symbol_table_t::get(const std::string& key) -> symbol_value_t& {
+        return this->operator[](key);
+    }
+
+    auto symbol_table_t::get(const std::string& key) const -> const symbol_value_t& {
+        return this->at(key);
+    }
+
     auto symbol_table_t::put(const symbol_table_t &other) -> symbol_table_t & {
         for (auto &e: other)
             this->insert_or_assign(e.first, e.second);