Add ProjectToAttribute

Yoshka · Feb 24, 2021 · d0c9c70 · d0c9c70
1 parent cd0095a
commit d0c9c70
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/src/Mapster.Async.Tests/AsyncTest.cs b/src/Mapster.Async.Tests/AsyncTest.cs
@@ -1,8 +1,5 @@
 using System;
-using System.Linq.Expressions;
-using System.Threading;
 using System.Threading.Tasks;
-using ExpressionDebugger;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using Shouldly;
 

diff --git a/src/Mapster.Core/Attributes/BaseAdaptAttribute.cs b/src/Mapster.Core/Attributes/BaseAdaptAttribute.cs
@@ -46,4 +46,17 @@ public class AdaptTwoWaysAttribute : AdaptToAttribute
  public AdaptTwoWaysAttribute(Type type) : base(type) { }
  public AdaptTwoWaysAttribute(string name) : base(name) { }
  }
+
+ public class ProjectToAttribute : AdaptToAttribute
+ {
+ public ProjectToAttribute(Type type) : base(type)
+ {
+ MapType = MapType.Projection;
+ }
+
+ public ProjectToAttribute(string name) : base(name)
+ {
+ MapType = MapType.Projection;
+ }
+ }
 }
diff --git a/src/Mapster.Tool/Program.cs b/src/Mapster.Tool/Program.cs
@@ -389,7 +389,7 @@ private static void GenerateExtensions(ExtensionOptions opt)
  {
  var tuple = new TypeTuple(type, toType);
  var mapType = attr.MapType == 0
- ? MapType.Map | MapType.MapToTarget | MapType.Projection
+ ? MapType.Map | MapType.MapToTarget
  : attr.MapType;
  GenerateExtensionMethods(mapType, cloned, tuple, translator, type, mapperAttr.IsHelperClass);
  }