diff --git a/src/main/java/org/ntlab/tampoposerver/models/ChatRooms.java b/src/main/java/org/ntlab/tampoposerver/models/ChatRooms.java index ea63b32..542c07d 100644 --- a/src/main/java/org/ntlab/tampoposerver/models/ChatRooms.java +++ b/src/main/java/org/ntlab/tampoposerver/models/ChatRooms.java @@ -13,9 +13,21 @@ private String newContent; private String chatRoomNumber; private List messages = new ArrayList<>(); + private List users = new ArrayList<>(); - public ChatRooms(Integer chatRoomId) { + public ChatRooms(Integer chatRoomId, String user0Id, String user1Id) { + this.chatRoomId = chatRoomId; + this.user0Id = user0Id; + this.user1Id = user1Id; + } + + public String getUser0Id(){ + return user0Id; + } + + public String getUser1Id(){ + return user1Id; } public Integer getChatRoomId() { @@ -27,6 +39,13 @@ } public void addUser(String userId) { + if (!users.contains(userId)) { + users.add(userId); + } + } + + public List getUsers() { + return new ArrayList<>(users); } public void put(Integer chatRoomId, String chatRoomNumber) {