start

Author: DocSvartz

src/Mapster.EFCore.Tests/EFCoreTest.cs

@@ -33,7 +33,8 @@ public void TestFindObject()
                     }
                 }
             };
-            var poco = context.Students.Include(it => it.Enrollments)
+            var poco = context.Students
+                .Include(it => it.Enrollments)
                 .First(it => it.ID == dto.ID);
 
             dto.BuildAdapter()
@@ -56,15 +57,21 @@ public void MapperInstance_From_OrderBy()
 
             var mapsterInstance = new Mapper();
 
-            var query = context.Students.Include(it => it.Enrollments);
-            var orderedQuery = mapsterInstance.From(query)
-                .ProjectToType<StudentDto>()
+            TypeAdapterConfig<Student, StudentDto>
+                .NewConfig()
+                .Map(x => x.Enrollments, x => x.Enrollments);
+
+
+            var query = context.Students
+                .Include(it => it.Enrollments.OrderByDescending(x=>x.StudentID).Take(1))
+                .Include(it=> it.Students)
+                .EFCoreProjectToType<StudentDto>()
                 .OrderBy(s => s.LastName);
 
-            var first = orderedQuery.First();
+            var first = query.First();
             first.LastName.ShouldBe("Alexander");
 
-            var last = orderedQuery.Last();
+            var last = query.Last();
             last.LastName.ShouldBe("Olivetto");
         }
     }

src/Mapster.EFCore.Tests/Models/Student.cs

@@ -11,5 +11,7 @@ public class Student
         public DateTime EnrollmentDate { get; set; }
 
         public ICollection<Enrollment> Enrollments { get; set; }
+
+        public ICollection<Student> Students { get; set; }
     }
 }

src/Mapster.EFCore/EFCoreExtensions.cs

@@ -0,0 +1,121 @@
+using System.Collections.Generic;
+using System.Diagnostics.CodeAnalysis;
+using System.Linq;
+using System.Linq.Expressions;
+using System.Security.Cryptography.X509Certificates;
+
+namespace Mapster.EFCore
+{
+    public static class EFCoreExtensions
+    {
+        public static IQueryable<TDestination> EFCoreProjectToType<TDestination>(this IQueryable source, TypeAdapterConfig? config = null)
+        {
+
+            var allInclude = new IncludeVisiter();
+
+            allInclude.Visit(source.Expression);
+
+            if (config == null)
+            {
+                config = TypeAdapterConfig.GlobalSettings
+                .Clone()
+                .ForType(source.ElementType, typeof(TDestination))
+                .Config;
+            }
+            else
+            {
+                config = config.Clone()
+                    .ForType(source.ElementType, typeof(TDestination))
+                    .Config;
+            }
+                
+
+            return source.ProjectToType<TDestination>(config);
+
+           // var mockCall = config.GetProjectionCallExpression(source.ElementType, typeof(TDestination));
+           // var sourceCall = Expression.Call(mockCall.Method, source.Expression, mockCall.Arguments[1]);
+           // return source.Provider.CreateQuery<TDestination>(sourceCall);
+        }
+
+    }
+
+    internal class IncludeVisiter : ExpressionVisitor
+    {
+        [return: NotNullIfNotNull("node")]
+        public override Expression Visit(Expression node)
+        {
+            if (node == null)
+                return null;
+            switch (node.NodeType)
+            {
+                case ExpressionType.Call:
+                    {
+                        if(IsInclude(node))
+                        {
+                            var QuoteVisiter = new QuoteVisiter();
+                            QuoteVisiter.Visit(node);
+
+                            var memberv = new MemeberVisiter();
+                            memberv.Visit(QuoteVisiter.Quotes[0]);
+                        }
+                        return base.Visit(node);
+                    }
+            }
+
+            return base.Visit(node);
+        }
+
+        public List<Expression> IncludeExpression { get; protected set; } = new();
+        private bool IsInclude (Expression node) => node.Type.Name.StartsWith("IIncludableQueryable");
+
+    }
+
+    internal class QuoteVisiter: ExpressionVisitor
+    {
+        public List<UnaryExpression> Quotes { get; private set; } = new();
+
+        public override Expression Visit(Expression node)
+        {
+            if (node == null)
+                return null;
+            switch (node.NodeType)
+            {
+                case ExpressionType.Quote:
+                    {
+                        Quotes.Add((UnaryExpression)node);
+                        return base.Visit(node);
+                    }
+            }
+
+            return base.Visit(node);
+
+        }
+    }
+
+    internal class MemeberVisiter : ExpressionVisitor
+    {
+        public string? Key { get; private set; }
+
+        public override Expression Visit(Expression node)
+        {
+            
+
+            if (node == null)
+                return null;
+            switch (node.NodeType)
+            {
+                case ExpressionType.MemberAccess:
+                    {
+                        if (string.IsNullOrEmpty(Key))
+                            Key = ((MemberExpression)node).Member.Name;
+
+                        return base.Visit(node);
+                    }
+            }
+
+            return base.Visit(node);
+
+             
+        }
+    }
+}