90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
|
package strategy
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"sync/atomic"
|
|||
|
|
|
|||
|
|
"github.com/shopspring/decimal"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// =========================
|
|||
|
|
// MACD（單寫；快照多讀）
|
|||
|
|
// fastEMA / slowEMA 吃「收盤價」；signalEMA 吃「MACDLine」
|
|||
|
|
// 與 EMA/SMA 一致風格：Update()（單寫）、Load()（多讀）
|
|||
|
|
// =========================
|
|||
|
|
|
|||
|
|
// MACDSnapshot 對外發佈的不可變快照
|
|||
|
|
type MACDSnapshot struct {
|
|||
|
|
Params [3]uint // (fast, slow, signal)
|
|||
|
|
LastPrice decimal.Decimal // 最近一次喂入的收盤價
|
|||
|
|
MACDLine decimal.Decimal // DIF = fastEMA - slowEMA
|
|||
|
|
SignalLine decimal.Decimal // DEA = EMA(MACDLine, signalPeriod)
|
|||
|
|
Histogram decimal.Decimal // Hist = MACDLine - SignalLine
|
|||
|
|
Ready bool // 三條線皆 ready 才會 true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MACD 快照發佈器（單寫多讀）
|
|||
|
|
type MACD struct {
|
|||
|
|
// 內部用你已完成的 EMA（其本身就含 SMA-seed 與 ready 狀態）
|
|||
|
|
fastEMA *EMA
|
|||
|
|
slowEMA *EMA
|
|||
|
|
signalEMA *EMA
|
|||
|
|
|
|||
|
|
fastPeriod, slowPeriod, signalPeriod uint
|
|||
|
|
|
|||
|
|
last decimal.Decimal // 只作紀錄（debug/觀察用）
|
|||
|
|
snap atomic.Value // holds MACDSnapshot
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewMACD 建立 MACD（預設 12,26,9；也可自訂）
|
|||
|
|
func NewMACD(fast, slow, signal uint) *MACD {
|
|||
|
|
if !(fast > 0 && slow > 0 && signal > 0) {
|
|||
|
|
panic("MACD periods must be > 0")
|
|||
|
|
}
|
|||
|
|
if fast >= slow {
|
|||
|
|
panic("MACD requires fast < slow (e.g., 12 < 26)")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
m := &MACD{
|
|||
|
|
fastEMA: NewEMA(fast),
|
|||
|
|
slowEMA: NewEMA(slow),
|
|||
|
|
signalEMA: NewEMA(signal),
|
|||
|
|
fastPeriod: fast,
|
|||
|
|
slowPeriod: slow,
|
|||
|
|
signalPeriod: signal,
|
|||
|
|
}
|
|||
|
|
// 初始化一個空快照
|
|||
|
|
m.snap.Store(MACDSnapshot{Params: [3]uint{fast, slow, signal}})
|
|||
|
|
return m
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Update 僅供單一 writer 呼叫；回傳並發佈最新快照
|
|||
|
|
func (m *MACD) Update(close decimal.Decimal) MACDSnapshot {
|
|||
|
|
// 先用 EMA（你已改好的版本）各自更新快、慢線
|
|||
|
|
fast := m.fastEMA.Update(close)
|
|||
|
|
slow := m.slowEMA.Update(close)
|
|||
|
|
|
|||
|
|
macdLine := fast.Value.Sub(slow.Value)
|
|||
|
|
|
|||
|
|
// 信號線吃的是 MACDLine（非收盤價）
|
|||
|
|
sig := m.signalEMA.Update(macdLine)
|
|||
|
|
|
|||
|
|
ready := fast.Ready && slow.Ready && sig.Ready
|
|||
|
|
|
|||
|
|
snap := MACDSnapshot{
|
|||
|
|
Params: [3]uint{m.fastPeriod, m.slowPeriod, m.signalPeriod},
|
|||
|
|
LastPrice: close,
|
|||
|
|
MACDLine: macdLine,
|
|||
|
|
SignalLine: sig.Value,
|
|||
|
|
Histogram: macdLine.Sub(sig.Value),
|
|||
|
|
Ready: ready,
|
|||
|
|
}
|
|||
|
|
m.last = close
|
|||
|
|
m.snap.Store(snap)
|
|||
|
|
return snap
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Load 多 goroutine 零鎖讀取最新快照
|
|||
|
|
func (m *MACD) Load() MACDSnapshot {
|
|||
|
|
return m.snap.Load().(MACDSnapshot)
|
|||
|
|
}
|