FreelanceBot

1
package telegramBot.entity;
2

3

4
import lombok.Setter;
5
import telegramBot.enums.Language;
6
import lombok.Getter;
7
import lombok.NoArgsConstructor;
8

9
import jakarta.persistence.*;
10
import telegramBot.enums.SubscriptionStatus;
11

12
import java.util.ArrayList;
13
import java.util.List;
14

15
@Table(name = "Subscription")
16
@Entity
17
@NoArgsConstructor
18
public class Subscription extends BaseEntity {
19

20
    @Getter
21
    @Column(name = "lang")
22
    private String language;
23

24
    @Getter
25
    @Setter
26
    @Column(name = "status")
27
    @Enumerated(EnumType.STRING)
28
    private SubscriptionStatus status;
29

30
    public Subscription(Language language, SubscriptionStatus status){
31
        this.language = language.getName();
32
        this.status = status;
33
    }
34

35
    @Getter
36
    @OneToMany(mappedBy = "subscription", cascade = CascadeType.REFRESH)
37
    private List<Order> orders;
38

39

40
    public void addOrder(Order order){
41
        if(this.orders == null) this.orders = new ArrayList<>();
42
        orders.add(order);
43
    }
44

45

46
}
47