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: ...

---
 BackEnd/Timeline/Controllers/BookmarkTimelineController.cs | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

(limited to 'BackEnd/Timeline/Controllers/BookmarkTimelineController.cs')

diff --git a/BackEnd/Timeline/Controllers/BookmarkTimelineController.cs b/BackEnd/Timeline/Controllers/BookmarkTimelineController.cs
index d941013f..e7ffa5c5 100644
--- a/BackEnd/Timeline/Controllers/BookmarkTimelineController.cs
+++ b/BackEnd/Timeline/Controllers/BookmarkTimelineController.cs
@@ -8,7 +8,6 @@ using Timeline.Models.Validation;
 using Timeline.Services.Api;
 using Timeline.Services.Mapper;
 using Timeline.Services.Timeline;
-using Timeline.Services.User;
 
 namespace Timeline.Controllers
 {
@@ -21,18 +20,18 @@ namespace Timeline.Controllers
     {
         private readonly IBookmarkTimelineService _service;
         private readonly ITimelineService _timelineService;
-        private readonly TimelineMapper _timelineMapper;
+        private readonly IGenericMapper _mapper;
 
-        public BookmarkTimelineController(IBookmarkTimelineService service, ITimelineService timelineService, TimelineMapper timelineMapper)
+        public BookmarkTimelineController(IBookmarkTimelineService service, ITimelineService timelineService, IGenericMapper mapper)
         {
             _service = service;
             _timelineService = timelineService;
-            _timelineMapper = timelineMapper;
+            _mapper = mapper;
         }
 
         private Task<List<HttpTimeline>> Map(List<TimelineEntity> timelines)
         {
-            return _timelineMapper.MapToHttp(timelines, Url, this.GetOptionalUserId(), this.UserHasPermission(UserPermission.AllTimelineManagement));
+            return _mapper.MapListAsync<HttpTimeline>(timelines, Url, User);
         }
 
         /// <summary>
-- 
cgit v1.2.3