ExternalTrainer.java

package com.exadel.model.entity.user;

import com.exadel.dto.NewTrainerDTO;
import com.exadel.dto.UserDTO;
import com.exadel.model.entity.feedback.UserFeedback;
import com.exadel.model.entity.training.Training;

import javax.persistence.CascadeType;
import javax.persistence.DiscriminatorValue;
import javax.persistence.Entity;
import javax.persistence.OneToMany;
import java.util.List;
import java.util.Set;

/**
 * Created by Виктория on 13.07.2015.
 */
@Entity
@DiscriminatorValue(value = "2")
public class ExternalTrainer extends User {

    @OneToMany(mappedBy = "trainer", cascade = CascadeType.ALL)
    private List<Training> mentoringTrainings;

    @OneToMany(mappedBy = "trainer", cascade = CascadeType.ALL)
    private Set<UserFeedback> receivedFeedbacks;

    public ExternalTrainer() {
        super.setRole(UserRole.EXTERNAL_TRAINER);
    }

    public ExternalTrainer(NewTrainerDTO trainerDTO) {
        super(trainerDTO);
        super.setRole(UserRole.EXTERNAL_TRAINER);
    }

    public List<Training> getMentoringTrainings() {
        return mentoringTrainings;
    }

    public void setMentoringTrainings(List<Training> mentoringTrainings) {
        this.mentoringTrainings = mentoringTrainings;
    }

    public Set<UserFeedback> getReceivedFeedbacks() {
        return receivedFeedbacks;
    }

    public void setReceivedFeedbacks(Set<UserFeedback> receivedFeedbacks) {
        this.receivedFeedbacks = receivedFeedbacks;
    }
}