AMT-Cheif
/
drift
mirror of https://github.com/AMT-Cheif/drift.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Resolve and infer types related to like expressions

This commit is contained in:
Simon Binder 2019-07-01 13:59:52 +02:00
parent 791afdb6bf
commit 3e910123d2
No known key found for this signature in database
GPG Key ID: 7891917E4147B8C0
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
sqlparser/lib/src/analysis/types/resolver.dart
View File

@ -89,6 +89,8 @@ class TypeResolver {
[BasicType.int, BasicType.real, BasicType.text, BasicType.blob]);
return ResolveResult(type);
}
} else if (expr is StringComparisonExpression) {
return const ResolveResult(ResolvedType.bool());
} else if (expr is BetweenExpression) {
return const ResolveResult(ResolvedType.bool());
} else if (expr is CaseExpression) {
@ -262,6 +264,14 @@ class TypeResolver {
final relevant = parent.childNodes
.lastWhere((node) => node is Expression && node != argument);
return resolveExpression(relevant as Expression);
} else if (parent is StringComparisonExpression) {
if (argument == parent.escape) {
return const ResolveResult(ResolvedType(type: BasicType.text));
} else {
final otherNode = parent.childNodes
.firstWhere((node) => node is Expression && node != argument);
return resolveExpression(otherNode as Expression);
}
} else if (parent is Parentheses || parent is UnaryExpression) {
return const ResolveResult.needsContext();
} else if (parent is FunctionExpression) {

4
sqlparser/test/analysis/type_resolver_test.dart
View File

@ -16,6 +16,10 @@ Map<String, ResolveResult> _types = {
const ResolveResult(ResolvedType(type: BasicType.int)),
'UPDATE demo SET content = ? WHERE id = 3':
const ResolveResult(ResolvedType(type: BasicType.text)),
'SELECT * FROM demo WHERE content LIKE ?':
const ResolveResult(ResolvedType(type: BasicType.text)),
"SELECT * FROM demo WHERE content LIKE '%e' ESCAPE ?":
const ResolveResult(ResolvedType(type: BasicType.text)),
};
void main() {