From 1b93136e14bb20d908a9b41d9307004e1fc7552a Mon Sep 17 00:00:00 2001 From: crupest Date: Fri, 8 Apr 2022 20:44:39 +0800 Subject: ... --- .../Timeline/Controllers/TimelineV2Controller.cs | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) (limited to 'BackEnd/Timeline') diff --git a/BackEnd/Timeline/Controllers/TimelineV2Controller.cs b/BackEnd/Timeline/Controllers/TimelineV2Controller.cs index 63beb357..9811cbed 100644 --- a/BackEnd/Timeline/Controllers/TimelineV2Controller.cs +++ b/BackEnd/Timeline/Controllers/TimelineV2Controller.cs @@ -5,6 +5,7 @@ using Microsoft.AspNetCore.Mvc; using Timeline.Entities; using Timeline.Models.Http; using Timeline.Models.Validation; +using Timeline.Services; using Timeline.Services.Mapper; using Timeline.Services.Timeline; using Timeline.Services.User; @@ -91,7 +92,15 @@ namespace Timeline.Controllers return Forbid(); } - var userId = await _userService.GetUserIdByUsernameAsync(member); + long userId; + try + { + userId = await _userService.GetUserIdByUsernameAsync(member); + } + catch (EntityNotExistException e) when (e.EntityType.Equals(EntityTypes.User)) + { + return UnprocessableEntity(new CommonResponse(ErrorCodes.Common.InvalidModel, "Member username does not exist.")); + } await _timelineService.AddMemberAsync(timelineId, userId); return NoContent(); } @@ -111,7 +120,15 @@ namespace Timeline.Controllers return Forbid(); } - var userId = await _userService.GetUserIdByUsernameAsync(member); + long userId; + try + { + userId = await _userService.GetUserIdByUsernameAsync(member); + } + catch (EntityNotExistException e) when (e.EntityType.Equals(EntityTypes.User)) + { + return UnprocessableEntity(new CommonResponse(ErrorCodes.Common.InvalidModel, "Member username does not exist.")); + } await _timelineService.RemoveMemberAsync(timelineId, userId); return NoContent(); } -- cgit v1.2.3