cloud-fan commented on a change in pull request #28875:
URL: https://github.com/apache/spark/pull/28875#discussion_r443952141
##########
File path:
sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/parser/DDLParserSuite.scala
##########
@@ -1134,58 +1134,70 @@ class DDLParserSuite extends AnalysisTest {
}
}
- test("merge into table: at most two matched clauses") {
- val exc = intercept[ParseException] {
- parsePlan(
- """
- |MERGE INTO testcat1.ns1.ns2.tbl AS target
- |USING testcat2.ns1.ns2.tbl AS source
- |ON target.col1 = source.col1
- |WHEN MATCHED AND (target.col2='delete') THEN DELETE
- |WHEN MATCHED AND (target.col2='update1') THEN UPDATE SET
target.col2 = source.col2
- |WHEN MATCHED AND (target.col2='update2') THEN UPDATE SET
target.col2 = source.col2
- |WHEN NOT MATCHED AND (target.col2='insert')
- |THEN INSERT (target.col1, target.col2) values (source.col1,
source.col2)
- """.stripMargin)
- }
-
- assert(exc.getMessage.contains("There should be at most 2 'WHEN MATCHED'
clauses."))
+ test("merge into table: multi matched and not matched clauses") {
+ parseCompare(
+ """
+ |MERGE INTO testcat1.ns1.ns2.tbl AS target
+ |USING testcat2.ns1.ns2.tbl AS source
+ |ON target.col1 = source.col1
+ |WHEN MATCHED AND (target.col2='delete') THEN DELETE
+ |WHEN MATCHED AND (target.col2='update to 1') THEN UPDATE SET
target.col2 = 1
+ |WHEN MATCHED AND (target.col2='update to 2') THEN UPDATE SET
target.col2 = 2
+ |WHEN NOT MATCHED AND (target.col2='insert 1')
+ |THEN INSERT (target.col1, target.col2) values (source.col1, 1)
+ |WHEN NOT MATCHED AND (target.col2='insert 2')
+ |THEN INSERT (target.col1, target.col2) values (source.col1, 2)
+ """.stripMargin,
+ MergeIntoTable(
+ SubqueryAlias("target", UnresolvedRelation(Seq("testcat1", "ns1",
"ns2", "tbl"))),
+ SubqueryAlias("source", UnresolvedRelation(Seq("testcat2", "ns1",
"ns2", "tbl"))),
+ EqualTo(UnresolvedAttribute("target.col1"),
UnresolvedAttribute("source.col1")),
+ Seq(DeleteAction(Some(EqualTo(UnresolvedAttribute("target.col2"),
Literal("delete")))),
+ UpdateAction(Some(EqualTo(UnresolvedAttribute("target.col2"),
Literal("update to 1"))),
+ Seq(Assignment(UnresolvedAttribute("target.col2"), Literal(1)))),
+ UpdateAction(Some(EqualTo(UnresolvedAttribute("target.col2"),
Literal("update to 2"))),
+ Seq(Assignment(UnresolvedAttribute("target.col2"), Literal(2))))),
+ Seq(InsertAction(Some(EqualTo(UnresolvedAttribute("target.col2"),
Literal("insert 1"))),
+ Seq(Assignment(UnresolvedAttribute("target.col1"),
UnresolvedAttribute("source.col1")),
+ Assignment(UnresolvedAttribute("target.col2"), Literal(1)))),
+ InsertAction(Some(EqualTo(UnresolvedAttribute("target.col2"),
Literal("insert 2"))),
+ Seq(Assignment(UnresolvedAttribute("target.col1"),
UnresolvedAttribute("source.col1")),
+ Assignment(UnresolvedAttribute("target.col2"), Literal(2)))))))
}
- test("merge into table: at most one not matched clause") {
+ test("merge into table: the first matched clause must have a condition if
there's a second") {
Review comment:
nit: `merge into table: only the last matched clause can omit the
condition`
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
