Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 49 additions & 0 deletions src/main/java/com/wellsfargo/counselor/entity/Client.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
package com.wellsfargo.counselor.entity;

import jakarta.persistence.*;
import java.util.List;

@Entity
public class Client {

@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private long clientId;

@Column(nullable = false)
private String firstName;

@Column(nullable = false)
private String lastName;

@Column(nullable = false)
private String email;

@ManyToOne
@JoinColumn(name = "advisor_id", nullable = false)
private Advisor advisor;

@OneToOne(mappedBy = "client", cascade = CascadeType.ALL)
private Portfolio portfolio;

protected Client() {}

public Client(String firstName, String lastName, String email, Advisor advisor) {
this.firstName = firstName;
this.lastName = lastName;
this.email = email;
this.advisor = advisor;
}

public long getClientId() { return clientId; }
public String getFirstName() { return firstName; }
public void setFirstName(String firstName) { this.firstName = firstName; }
public String getLastName() { return lastName; }
public void setLastName(String lastName) { this.lastName = lastName; }
public String getEmail() { return email; }
public void setEmail(String email) { this.email = email; }
public Advisor getAdvisor() { return advisor; }
public void setAdvisor(Advisor advisor) { this.advisor = advisor; }
public Portfolio getPortfolio() { return portfolio; }
public void setPortfolio(Portfolio portfolio) { this.portfolio = portfolio; }
}
37 changes: 37 additions & 0 deletions src/main/java/com/wellsfargo/counselor/entity/Portfolio.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
package com.wellsfargo.counselor.entity;

import jakarta.persistence.*;
import java.util.List;

@Entity
public class Portfolio {

@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private long portfolioId;

@Column(nullable = false)
private String name;

@OneToOne
@JoinColumn(name = "client_id", nullable = false)
private Client client;

@OneToMany(mappedBy = "portfolio", cascade = CascadeType.ALL)
private List<Security> securities;

protected Portfolio() {}

public Portfolio(String name, Client client) {
this.name = name;
this.client = client;
}

public long getPortfolioId() { return portfolioId; }
public String getName() { return name; }
public void setName(String name) { this.name = name; }
public Client getClient() { return client; }
public void setClient(Client client) { this.client = client; }
public List<Security> getSecurities() { return securities; }
public void setSecurities(List<Security> securities) { this.securities = securities; }
}
56 changes: 56 additions & 0 deletions src/main/java/com/wellsfargo/counselor/entity/Security.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
package com.wellsfargo.counselor.entity;

import jakarta.persistence.*;
import java.time.LocalDate;

@Entity
public class Security {

@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private long securityId;

@Column(nullable = false)
private String name;

@Column(nullable = false)
private String category;

@Column(nullable = false)
private LocalDate purchaseDate;

@Column(nullable = false)
private double purchasePrice;

@Column(nullable = false)
private int quantity;

@ManyToOne
@JoinColumn(name = "portfolio_id", nullable = false)
private Portfolio portfolio;

protected Security() {}

public Security(String name, String category, LocalDate purchaseDate, double purchasePrice, int quantity, Portfolio portfolio) {
this.name = name;
this.category = category;
this.purchaseDate = purchaseDate;
this.purchasePrice = purchasePrice;
this.quantity = quantity;
this.portfolio = portfolio;
}

public long getSecurityId() { return securityId; }
public String getName() { return name; }
public void setName(String name) { this.name = name; }
public String getCategory() { return category; }
public void setCategory(String category) { this.category = category; }
public LocalDate getPurchaseDate() { return purchaseDate; }
public void setPurchaseDate(LocalDate purchaseDate) { this.purchaseDate = purchaseDate; }
public double getPurchasePrice() { return purchasePrice; }
public void setPurchasePrice(double purchasePrice) { this.purchasePrice = purchasePrice; }
public int getQuantity() { return quantity; }
public void setQuantity(int quantity) { this.quantity = quantity; }
public Portfolio getPortfolio() { return portfolio; }
public void setPortfolio(Portfolio portfolio) { this.portfolio = portfolio; }
}