resolves #262 string as a lookup variable

Litote · Jan 30, 2021 · 60ed58a · 60ed58a
1 parent bca01b9
commit 60ed58a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/kmongo-property/src/main/kotlin/org/litote/kmongo/Aggregates.kt b/kmongo-property/src/main/kotlin/org/litote/kmongo/Aggregates.kt
@@ -484,5 +484,10 @@ fun KProperty<*>.variableDefinition(name: String = path()): Variable<String> = V
 /**
  * Defines a [Variable] projection (ie $$name)
  */
-val KProperty<*>.variable: String get() = "\$\$${path()}"
+val KProperty<*>.variable: String get() = path().variable
+
+/**
+ * Defines a [Variable] projection (ie $$"myString"")
+ */
+val String.variable: String get() = "\$\$$this"
 
diff --git a/kmongo-property/src/test/kotlin/org/litote/kmongo/AggregatesTest.kt b/kmongo-property/src/test/kotlin/org/litote/kmongo/AggregatesTest.kt
@@ -49,4 +49,10 @@ class AggregatesTest {
             )
         )
     }
+
+    @Test
+    fun testVariable() {
+        assertEquals("\$\$s", AggregateData::s.variable)
+        assertEquals("\$\$s", "s".variable)
+    }
 }