cloud-fan commented on code in PR #53530:
URL: https://github.com/apache/spark/pull/53530#discussion_r2690841208
##########
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/parser/AstBuilder.scala:
##########
@@ -6581,6 +6583,105 @@ class AstBuilder extends DataTypeAstBuilder
)
}
+ /**
+ * Create a [[DeclareCursor]] command wrapped in SingleStatement.
+ *
+ * For example:
+ * {{{
+ * DECLARE cursor_name CURSOR FOR SELECT * FROM table;
+ * }}}
+ */
+ override def visitDeclareCursorStatement(
+ ctx: DeclareCursorStatementContext): LogicalPlan = withOrigin(ctx) {
+ if (!conf.getConf(SQLConf.SQL_SCRIPTING_CURSOR_ENABLED)) {
+ throw SqlScriptingErrors.cursorNotSupported(CurrentOrigin.get)
+ }
+
+ val cursorName = getIdentifierText(ctx.name)
+ // Extract original SQL text to preserve parameter markers
+ val queryText = getOriginalText(ctx.query())
+
+ val asensitive = if (ctx.INSENSITIVE() != null) false else true
+ SingleStatement(DeclareCursor(cursorName, queryText, asensitive))
+ }
+
+ /**
+ * Create an [[OpenCursor]] command wrapped in SingleStatement.
+ *
+ * For example:
+ * {{{
+ * OPEN cursor_name;
+ * OPEN cursor_name USING expr1, expr2;
+ * OPEN cursor_name USING (expr1 AS param1, expr2 AS param2);
+ * }}}
+ */
+ override def visitOpenCursorStatement(
+ ctx: OpenCursorStatementContext): LogicalPlan = withOrigin(ctx) {
+ if (!conf.getConf(SQLConf.SQL_SCRIPTING_CURSOR_ENABLED)) {
+ throw SqlScriptingErrors.cursorNotSupported(CurrentOrigin.get)
+ }
+
+ // Create UnresolvedCursor from the cursor name
+ val nameParts = visitMultipartIdentifier(ctx.multipartIdentifier())
Review Comment:
shall we fail earlier if there are more than two name parts?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
