fix ut and agg

csharp-opensource · Dec 14, 2024 · ff3ddef · ff3ddef
1 parent 86a757f
commit ff3ddef
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/LinqToKql.Test/CSharp.OpenSource.LinqToKql.Test.csproj b/LinqToKql.Test/CSharp.OpenSource.LinqToKql.Test.csproj
@@ -19,7 +19,7 @@
   </ItemGroup>
 
   <ItemGroup>
-    <ProjectReference Include="..\CSharp.OpenSource.LinqToKql\CSharp.OpenSource.LinqToKql.csproj" />
+    <ProjectReference Include="..\LinqToKql\CSharp.OpenSource.LinqToKql.csproj" />
   </ItemGroup>
 
   <ItemGroup>

diff --git a/LinqToKql/Translator/Builders/LinqToKQLTranslatorBase.cs b/LinqToKql/Translator/Builders/LinqToKQLTranslatorBase.cs
@@ -120,17 +120,25 @@ protected string GetArgMethod(Expression arg)
         if (arg is MethodCallExpression methodCall)
         {
             var methodName = methodCall.Method.Name;
-            if (methodName == "Count")
+            return methodName switch
             {
-                return "count()";
-            }
-            throw new InvalidOperationException($"{GetType().Name} - fail to translate");
+                "Count" => "count()",
+                "Sum" => $"sum({SelectMembers(methodCall.Arguments[1])})",
+                "Average" => $"avg({SelectMembers(methodCall.Arguments[1])})",
+                "Min" => $"min({SelectMembers(methodCall.Arguments[1])})",
+                "Max" => $"max({SelectMembers(methodCall.Arguments[1])})",
+                _ => throw new InvalidOperationException($"{GetType().Name} - Method {methodName} is not supported."),
+            };
         }
         else if (arg is MemberExpression mb2)
         {
             var propName = mb2.Member.Name;
             return propName;
         }
+        else if (arg is ConstantExpression constant)
+        {
+            return constant.Value.GetKQLValue();
+        }
         else
         {
             throw new InvalidOperationException($"{GetType().Name} - Unsupported expression type for aggregation.");