diff --git a/src/main/java/org/ntlab/tampoposerver/repositories/FriendRepository.java b/src/main/java/org/ntlab/tampoposerver/repositories/FriendRepository.java index 91be6d7..4ab2a8f 100644 --- a/src/main/java/org/ntlab/tampoposerver/repositories/FriendRepository.java +++ b/src/main/java/org/ntlab/tampoposerver/repositories/FriendRepository.java @@ -20,22 +20,21 @@ * @param pairId 取得する対象のフレンドペアのID * @return IDに紐づけられているフレンドペアが存在すればそのペア、存在しなければnull */ - public FriendPair getFriend(int pairId) { + public FriendPair find(int pairId) { return friends.get(pairId); } /** - * 渡された二つのユーザーIDから新しいフレンドのペアを作成し追加する + * 渡されたフレンドのペアを追加する * - * @param user0Id フレンドとして追加するユーザーのユーザーID - * @param user1Id フレンドとして追加するユーザーのユーザーID - * @return 新しく作成したフレンドペア + * @param pair 追加するフレンドペア + * @return 追加したフレンドペア */ - public FriendPair addFriend(String user0Id, String user1Id) { + public FriendPair create(FriendPair pair) { lastPairId++; - FriendPair newPair = new FriendPair(lastPairId, user0Id, user1Id); - friends.put(newPair.getId(), newPair); - return newPair; + pair.setId(lastPairId); + friends.put(pair.getId(), pair); + return pair; } /** @@ -44,7 +43,7 @@ * @param pairId 削除する対象のフレンドペアのID * @return 削除できたらtrue、削除できなければfalse */ - public boolean removeFriend(int pairId) { + public boolean delete(int pairId) { FriendPair removedPair = friends.remove(pairId); return removedPair != null; }