diff --git a/src/main/java/org/ntlab/acanthus_server/models/Gallery.java b/src/main/java/org/ntlab/acanthus_server/models/Gallery.java index 0f73498..a5c3e1c 100644 --- a/src/main/java/org/ntlab/acanthus_server/models/Gallery.java +++ b/src/main/java/org/ntlab/acanthus_server/models/Gallery.java @@ -2,6 +2,7 @@ import org.ntlab.acanthus_server.entities.Account; import org.ntlab.acanthus_server.entities.Animation; +import org.ntlab.acanthus_server.entities.Work; import java.util.*; import java.util.ArrayList; @@ -65,6 +66,7 @@ var aid = new Random().nextInt(); var newAnimation = new Animation(aid, animationName, owner); animationHashMap.put(aid,newAnimation); + owner.getWorkHashMap().put(aid, new Work(newAnimation)); newAnimation.setName(animationName); return aid; } diff --git a/src/main/java/org/ntlab/acanthus_server/resources/gallery/InvitesRest.java b/src/main/java/org/ntlab/acanthus_server/resources/gallery/InvitesRest.java index eb3a411..ce86f18 100644 --- a/src/main/java/org/ntlab/acanthus_server/resources/gallery/InvitesRest.java +++ b/src/main/java/org/ntlab/acanthus_server/resources/gallery/InvitesRest.java @@ -90,8 +90,8 @@ } //既に招待されているかのチェック - for (var invite : invitedAccount.getInvitedMap().values()) { - if(invite.isInviting()){ + for (Account account : animation.getInvites()) { + if (account.getUid() == Integer.parseInt(invitedUid)) { var response = Response.status(401).entity("既にユーザーを招待しています."); throw new WebApplicationException(response.build()); }