diff --git a/src/Server/src/main/java/greenify/server/data/model/User.java b/src/Server/src/main/java/greenify/server/data/model/User.java index 39f385b..bb3a5d6 100644 --- a/src/Server/src/main/java/greenify/server/data/model/User.java +++ b/src/Server/src/main/java/greenify/server/data/model/User.java @@ -13,6 +13,8 @@ import javax.persistence.GeneratedValue; import javax.persistence.GenerationType; import javax.persistence.Id; import javax.persistence.Table; +import java.util.*; +import javax.persistence.*; import javax.validation.constraints.NotNull; @Entity @@ -21,7 +23,9 @@ import javax.validation.constraints.NotNull; public class User { @Id + @NotNull @GeneratedValue(strategy = GenerationType.AUTO) +// @Column(name = "id") private Long id; @NotNull @@ -36,6 +40,10 @@ public class User { @ElementCollection private Map footPrintInputs = new HashMap<>(); + @ManyToMany + @JoinColumn + private Collection friends; + public User() {} /** @@ -103,6 +111,10 @@ public class User { this.footPrintInputs = footPrintInputs; } + public ArrayList getFriends(){ + return (ArrayList)this.friends; + } + public void setFootPrint(Float footPrint) { this.footPrint = footPrint; }