Fix caching of aggregate values with @parent

And add a test for referencedVariables
qgis · Oct 28, 2016 · 7a05a7a8c4b9184f6cad7d155b6714076b1dccd9 · 7a05a7a · nirvn · Oct 31, 2016
1 parent 8e25b89
commit 7a05a7a8c4b9184f6cad7d155b6714076b1dccd9
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 3 deletions.
diff --git a/src/core/qgsexpression.cpp b/src/core/qgsexpression.cpp
@@ -682,8 +682,8 @@ static QVariant fcnAggregate( const QVariantList& values, const QgsExpressionCon
   {
     QString cacheKey = QStringLiteral( "aggfcn:%1:%2:%3:%4" ).arg( vl->id(), QString::number( aggregate ), subExpression, parameters.filter );
 
-    QgsExpression subExp( subExpression );
-    if ( subExp.referencedVariables().contains( "parent" ) || subExp.referencedVariables().contains( QString() ) )
+    QgsExpression filterExp( parameters.filter );
+    if ( filterExp.referencedVariables().contains( "parent" ) || filterExp.referencedVariables().contains( QString() ) )
     {
       cacheKey += ':' + qHash( context->feature() );
     }
@@ -5106,7 +5106,19 @@ QSet<QString> QgsExpression::NodeFunction::referencedVariables() const
     return QSet<QString>() << QString();
   }
   else
-    return QSet<QString>();
+  {
+    QSet<QString> functionVariables = QSet<QString>();
+
+    if ( !mArgs )
+      return functionVariables;
+
+    Q_FOREACH ( Node* n, mArgs->list() )
+    {
+      functionVariables.unite( n->referencedVariables() );
+    }
+
+    return functionVariables;
+  }
 }
 
 bool QgsExpression::NodeFunction::needsGeometry() const

diff --git a/tests/src/core/testqgsexpression.cpp b/tests/src/core/testqgsexpression.cpp
@@ -1571,6 +1571,17 @@ class TestQgsExpression: public QObject
       QCOMPARE( refColsSet, expectedCols );
     }
 
+    void referenced_variables()
+    {
+      QSet<QString> expectedVars;
+      expectedVars << QStringLiteral( "foo" ) << QStringLiteral( "bar" ) << QStringLiteral( "ppp" ) << QStringLiteral( "qqq" ) << QStringLiteral( "rrr" );
+      QgsExpression exp( QStringLiteral( "CASE WHEN intersects(@bar, $geometry) THEN @ppp ELSE @qqq * @rrr END + @foo IN (1, 2, 3)" ) );
+      QCOMPARE( exp.hasParserError(), false );
+      QSet<QString> refVar = exp.referencedVariables();
+
+      QCOMPARE( refVar, expectedVars );
+    }
+
     void referenced_columns_all_attributes()
     {
       QgsExpression exp( QStringLiteral( "attribute($currentfeature,'test')" ) );