/
rewriteProcedureCalls.scala
58 lines (52 loc) · 2.56 KB
/
rewriteProcedureCalls.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
/*
* Copyright (c) 2002-2016 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.neo4j.cypher.internal.compiler.v3_1
import org.neo4j.cypher.internal.compiler.v3_1.ast.ResolvedCall
import org.neo4j.cypher.internal.compiler.v3_1.spi.{ProcedureSignature, QualifiedProcedureName}
import org.neo4j.cypher.internal.frontend.v3_1.ast._
import org.neo4j.cypher.internal.frontend.v3_1.{Rewriter, bottomUp}
// Given a way to lookup procedure signatures, this factory returns a rewriter that
// turns unresolved calls into resolved calls
object rewriteProcedureCalls {
def apply(signatureLookup: QualifiedProcedureName => ProcedureSignature) = {
// rewriter that amends unresolved procedure calls with procedure signature information
val resolveCalls = bottomUp(Rewriter.lift {
case unresolved: UnresolvedCall =>
val resolved = ResolvedCall(signatureLookup)(unresolved)
// We coerce here to ensure that the semantic check run after this rewriter assigns a type
// to the coercion expression
val coerced = resolved.coerceArguments
coerced
})
resolveCalls andThen fakeStandaloneCallDeclarations
}
// Current procedure calling syntax allows simplified short-hand syntax for queries
// that only consist of a standalone procedure call. In all other cases attempts to
// use the simplified syntax lead to errors during semantic checking.
//
// This rewriter rewrites standalone calls in simplified syntax to calls in standard
// syntax to prevent them from being rejected during semantic checking.
//
private val fakeStandaloneCallDeclarations = Rewriter.lift {
case q@Query(None, part@SingleQuery(Seq(resolved@ResolvedCall(_, _, _, _, _)))) if !resolved.fullyDeclared =>
val result = q.copy(part = part.copy(clauses = Seq(resolved.withFakedFullDeclarations))(part.position))(q.position)
result
}
}