Hugging Face's logo

Spaces:
uncertainrods
/
quantengine
Sleeping

App Files Community
quantengine
File size: 2,197 Bytes
f5cd2d3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package com.rods.backtestingstrategies.entity;

import jakarta.persistence.*;
import lombok.*;

import java.time.LocalDate;

@Entity
@Table(
        name = "trade_signals",
        indexes = {
                @Index(name = "idx_trade_signal_date", columnList = "signalDate"),
                @Index(name = "idx_trade_signal_type", columnList = "signalType")
        }
)
@Getter
@NoArgsConstructor(access = AccessLevel.PROTECTED) // JPA requirement
@AllArgsConstructor(access = AccessLevel.PRIVATE) // Force factory usage
@EqualsAndHashCode(of = {"signalDate", "signalType", "price"})
@ToString
public class TradeSignal {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    // Date on which signal is generated
    @Column(nullable = false)
    private LocalDate signalDate;

    @Enumerated(EnumType.STRING)
    @Column(nullable = false, length = 10)
    private SignalType signalType;

    // Price at signal generation (usually close price)
    @Column(nullable = false)
    private double price;

    // Optional: identify which strategy generated this signal
    @Column(length = 100)
    private String strategyName;

    /* ==========================
       Factory Methods
       ========================== */

    public static TradeSignal buy(Candle candle) {
        return new TradeSignal(
                null,
                candle.getDate(),
                SignalType.BUY,
                candle.getClosePrice(),
                null
        );
    }

    public static TradeSignal sell(Candle candle) {
        return new TradeSignal(
                null,
                candle.getDate(),
                SignalType.SELL,
                candle.getClosePrice(),
                null
        );
    }

    public static TradeSignal hold() {
        return new TradeSignal(
                null,
                null,
                SignalType.HOLD,
                0.0,
                null
        );
    }

    /* ==========================
       Optional helpers
       ========================== */

    public TradeSignal withStrategyName(String strategyName) {
        this.strategyName = strategyName;
        return this;
    }
}