Rewrite alter table in ANTLR

amazon-aurora · Mar 5, 2025 · 38436fd · 38436fd
1 parent a9d41c7
commit 38436fd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/contrib/babelfishpg_tsql/src/tsqlIface.cpp b/contrib/babelfishpg_tsql/src/tsqlIface.cpp
@@ -1811,6 +1811,10 @@ class tsqlBuilder : public tsqlCommonMutator
 
 			rewritten_query_fragment.emplace(std::make_pair(ctx->start->getStartIndex(), std::make_pair(::getFullText(ctx), str)));
 		}
+		else if(ctx->TRIGGER() && ctx->ALL())
+		{
+			rewritten_query_fragment.emplace(std::make_pair(ctx->ALL()->getSymbol()->getStartIndex(),std::make_pair(::getFullText(ctx->ALL()), "USER")));
+		}
 	}
 
 	void exitEnable_trigger(TSqlParser::Enable_triggerContext *ctx) override