Cleanup a symbol table access.

substrait-io · Jul 5, 2023 · 359424d · 359424d
1 parent 07fc316
commit 359424d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/substrait/textplan/converter/PlanPrinterVisitor.cpp b/src/substrait/textplan/converter/PlanPrinterVisitor.cpp
@@ -760,12 +760,10 @@ std::any PlanPrinterVisitor::visitReadRelation(
  return "";
  }
 
- auto mainSymbol = symbolTable_->lookupSymbolByLocationAndType(
- PROTO_LOCATION(relation), SymbolType::kRelation);
-
- auto symbols = symbolTable_->lookupSymbolsByLocation(PROTO_LOCATION(*msg));
- if (!symbols.empty()) {
- text << " source " << symbols[0]->name << ";\n";
+ auto source = symbolTable_->lookupSymbolByLocationAndType(
+ PROTO_LOCATION(*msg), SymbolType::kSource);
+ if (source != nullptr) {
+ text << " source " << source->name << ";\n";
  }
 
  if (relation.has_base_schema()) {