From e16c958e5ba47834dc1624e09ed8e5074a60d1c6 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Wed, 28 Apr 2021 19:20:40 +0800
Subject: refator: ...

---
 .../Services/Mapper/MapperServiceCollectionExtensions.cs         | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs')

diff --git a/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs b/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs
index 262b2f20..0c2d0cfd 100644
--- a/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs
+++ b/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs
@@ -1,4 +1,6 @@
 using Microsoft.Extensions.DependencyInjection;
+using Timeline.Entities;
+using Timeline.Models.Http;
 
 namespace Timeline.Services.Mapper
 {
@@ -6,8 +8,11 @@ namespace Timeline.Services.Mapper
     {
         public static void AddMappers(this IServiceCollection services)
         {
-            services.AddScoped<UserMapper, UserMapper>();
-            services.AddScoped<TimelineMapper, TimelineMapper>();
+            services.AddAutoMapper(typeof(Startup).Assembly);
+            services.AddScoped<IMapper<UserEntity, HttpUser>, UserMapper>();
+            services.AddScoped<IMapper<TimelineEntity, HttpTimeline>, TimelineMapper>();
+            services.AddScoped<IMapper<TimelinePostEntity, HttpTimelinePost>, TimelineMapper>();
+            services.AddScoped<IGenericMapper, GenericMapper>();
         }
     }
 }
-- 
cgit v1.2.3