diff --git a/src/main/java/org/ntlab/nemophila/models/shops/Shop.java b/src/main/java/org/ntlab/nemophila/models/shops/Shop.java index 074ad45..59368ca 100644 --- a/src/main/java/org/ntlab/nemophila/models/shops/Shop.java +++ b/src/main/java/org/ntlab/nemophila/models/shops/Shop.java @@ -1,6 +1,8 @@ package org.ntlab.nemophila.models.shops; import com.fasterxml.jackson.annotation.JsonProperty; +import org.ntlab.nemophila.models.accounts.Post; +import java.util.ArrayList; public class Shop { @JsonProperty("id") @@ -11,6 +13,8 @@ private double longitude; @JsonProperty("latitude") private double latitude; + @JsonProperty("posts") + private ArrayList posts = new ArrayList<>(); //Getter public String getId() { @@ -39,4 +43,18 @@ public void setLatitude(double latitude) { latitude = latitude; } + + //postsに関するメソッド + public void addPost(Post newPost) { + posts.add(newPost); + } + public void deletePost(Post post) { + posts.remove(post); + } + public Post getposts(int index) { + return posts.get(index); + } + public int getPostsCount() { + return posts.size(); + } } diff --git a/src/main/java/org/ntlab/nemophila/models/shops/ShopManager.java b/src/main/java/org/ntlab/nemophila/models/shops/ShopManager.java index 5beb3a2..eae8908 100644 --- a/src/main/java/org/ntlab/nemophila/models/shops/ShopManager.java +++ b/src/main/java/org/ntlab/nemophila/models/shops/ShopManager.java @@ -11,6 +11,7 @@ private ShopManager() { } + //インスタンスがなければ作成する public static ShopManager getInstance() { if (theInstance == null) { theInstance = new ShopManager(); @@ -27,4 +28,8 @@ public void setShopsMap(Shop shop) { shopsMap.put(shop.getId(), shop); } + + public void deleteShop(String id) { + shopsMap.remove(id); + } }