Training.java

package com.exadel.model.entity.training;

import com.exadel.dto.TrainingDTO;
import com.exadel.model.entity.ParticipationStatus;
import com.exadel.model.entity.feedback.TrainingFeedback;
import com.exadel.model.entity.user.ExternalTrainer;
import com.exadel.model.entity.user.User;
import org.apache.lucene.analysis.core.LowerCaseFilterFactory;
import org.apache.lucene.analysis.ngram.EdgeNGramFilterFactory;
import org.apache.lucene.analysis.ngram.NGramFilterFactory;
import org.apache.lucene.analysis.snowball.SnowballPorterFilterFactory;
import org.apache.lucene.analysis.standard.StandardTokenizerFactory;
import org.hibernate.search.annotations.*;
import org.apache.lucene.document.Field.Index;
import org.hibernate.search.annotations.Parameter;

import javax.persistence.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.List;

@Entity
@Indexed
@Table(name = "trainings")
@AnalyzerDef(name = "customanalyzer",
        tokenizer = @TokenizerDef(factory = StandardTokenizerFactory.class),
        filters = {
                @TokenFilterDef(factory = LowerCaseFilterFactory.class),
                @TokenFilterDef(factory = SnowballPorterFilterFactory.class, params = {
                        @Parameter(name = "language", value = "English")
                }),
                @TokenFilterDef(factory = SnowballPorterFilterFactory.class, params = {
                        @Parameter(name = "language", value = "Russian")
                }),
                @TokenFilterDef(factory =EdgeNGramFilterFactory.class, params  ={
                        @Parameter(name="minGramSize",value="1"),@Parameter(name="maxGramSize",value="10")})

        })
public class Training{

    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private long id;

    @Field(analyze= Analyze.YES, store= Store.NO)
    @Analyzer(definition = "customanalyzer")
    @Column(name = "name")
    private String name;

    @IndexedEmbedded
    @ManyToOne
    @JoinColumn(name = "trainer_id")
    private ExternalTrainer trainer;



    @Field( analyze= Analyze.YES, store= Store.NO)
    @Column(name = "target_audience")
    @Analyzer(definition = "customanalyzer")
    private String targetAudience;

    @Column(name = "is_external")
    private boolean isExternal;

    private String language;
    private String description;
    private TrainingStatus status;

    @Column(name = "members_count_max")
    private int membersCountMax;

    @Transient
    private double rating;

    @Column(name = "rating_sum")
    private int ratingSum;

    @Column(name = "valuer_count")
    private int valuerCount;

    private boolean repeated;

    @ManyToMany(fetch = FetchType.EAGER)
    @JoinTable(name = "training_users", joinColumns = @JoinColumn(name = "training_id"),
            inverseJoinColumns = @JoinColumn(name = "user_id"))
    private List<User> participants;


    @OneToMany(mappedBy = "training", cascade = CascadeType.ALL)
    private List<TrainingFeedback> feedbacks;

    @IndexedEmbedded
    @OneToMany(mappedBy = "training", cascade = CascadeType.ALL)
    private List<Entry> entries;

    @OneToMany(mappedBy = "training", cascade = CascadeType.ALL)
    private List<Attachment> attachments;

    @OneToMany(mappedBy = "training", cascade = CascadeType.ALL)
    private List<Reserve> reserves;

    public void addAttachment(Attachment attachment) {
        attachment.setTraining(this);
        attachments.add(attachment);
    }

    public Training() {
    }
    public Training(long id) {
        this.id = id;
    }

    public Training(TrainingDTO trainingDTO) {
        this.id = trainingDTO.getId();
        this.name = trainingDTO.getName();
        this.targetAudience = trainingDTO.getTargetAudience();
        this.language = trainingDTO.getLanguage();
        this.isExternal = trainingDTO.isExternal();
        this.description = trainingDTO.getDescription();
        this.status = trainingDTO.getStatus();
        this.membersCountMax = trainingDTO.getMembersCountMax();
        this.repeated = trainingDTO.isRepeated();

        this.trainer = new ExternalTrainer();
        this.trainer.setId(trainingDTO.getTrainer().getId());

        this.participants = new ArrayList<>();
        this.feedbacks = new ArrayList<>();
        this.entries = new ArrayList<>();
        this.reserves = new ArrayList<>();
    }

    public boolean isParticipant(long userId) {
        for (User participant : participants) {
            if (participant.getId() == userId) {
                return true;
            }
        }
        return false;
    }

    public boolean isReservist(long userId) {
        for (Reserve reserve : reserves) {
            if (reserve.getReservist().getId() == userId)
                return true;
        }
        return false;
    }

    public void updateTraining(TrainingDTO trainingDTO) {
        this.name = trainingDTO.getName();
        this.targetAudience = trainingDTO.getTargetAudience();
        this.language = trainingDTO.getLanguage();
        this.isExternal = trainingDTO.isExternal();
        this.description = trainingDTO.getDescription();
        this.status = trainingDTO.getStatus();
        this.membersCountMax = trainingDTO.getMembersCountMax();
    }

    public void updateTraining(Training training) {
        this.name = training.getName();
        this.trainer = training.getTrainer();
        this.targetAudience = training.getTargetAudience();
        this.isExternal = training.isExternal();
        this.language = training.getLanguage();
        this.description = training.getDescription();
        this.status = training.getStatus();
        this.membersCountMax = training.getMembersCountMax();
        this.rating = training.getRating();
        this.ratingSum = training.getRatingSum();
        this.valuerCount = training.getValuerCount();
        this.repeated = training.isRepeated();
    }

    public void addFeedback(TrainingFeedback feedback) {
        feedback.setTraining(this);
        feedbacks.add(feedback);
    }

    public void addEntry(Entry entry) {
        entry.setTraining(this);
        entries.add(entry);
    }

    public void addParticipant(User user) {
        this.participants.add(user);
    }

    public long getId() {
        return id;
    }

    public void setId(long id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public ExternalTrainer getTrainer() {
        return trainer;
    }

    public void setTrainer(ExternalTrainer trainer) {
        this.trainer = trainer;
    }

    public String getTargetAudience() {
        return targetAudience;
    }

    public void setTargetAudience(String targetAudience) {
        this.targetAudience = targetAudience;
    }

    public int getMembersCountMax() {
        return membersCountMax;
    }

    public void setMembersCountMax(int membersCountMax) {
        this.membersCountMax = membersCountMax;
    }

    public TrainingStatus getStatus() {
        return status;
    }

    public void setStatus(TrainingStatus status) {
        this.status = status;
    }

    public List<User> getParticipants() {
        return participants;
    }

    public void setParticipants(List<User> participants) {
        this.participants = participants;
    }

    public List<TrainingFeedback> getFeedbacks() {
        return feedbacks;
    }

    public void setFeedbacks(List<TrainingFeedback> feedbacks) {
        this.feedbacks = feedbacks;
    }

    public String getLanguage() {
        return language;
    }

    public void setLanguage(String language) {
        this.language = language;
    }

    public List<Entry> getEntries() {
        Collections.sort(entries);
        return entries;
    }

    public void setEntries(List<Entry> entries) {
        this.entries = entries;
    }

    public boolean IsExternal() {
        return isExternal;
    }

    public void setIsExternal(boolean isExternal) {
        this.isExternal = isExternal;
    }

    public String getDescription() {
        return description;
    }

    public void setDescription(String description) {
        this.description = description;
    }

    public boolean isExternal() {
        return isExternal;
    }

    public int getRatingSum() {
        return ratingSum;
    }

    public void setRatingSum(int ratingSum) {
        this.ratingSum = ratingSum;
    }

    public int getValuerCount() {
        return valuerCount;
    }

    public void setValuerCount(int valuerCount) {
        this.valuerCount = valuerCount;
    }

    public double getRating() {
        if (valuerCount != 0)
            return (double) ratingSum / valuerCount;
        return 0;
    }

    public void setRating(double rating) {
        this.rating = rating;
    }

    public Training addRating(int grade) {
        setValuerCount(getValuerCount() + 1);
        setRatingSum(getRatingSum() + grade);
        rating = getRatingSum() / getValuerCount();
        return this;
    }

    public List<Attachment> getAttachments() {
        return attachments;
    }

    public void setAttachments(List<Attachment> attachments) {
        this.attachments = attachments;
    }

    public boolean isRepeated() {
        return repeated;
    }

    public void setRepeated(boolean repeated) {
        this.repeated = repeated;
    }

    public List<Reserve> getReserves() {
        return reserves;
    }

    public void setReserves(List<Reserve> reserves) {
        this.reserves = reserves;
    }
}