51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
|
package strategy
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"container/list"
|
|||
|
|
"github.com/shopspring/decimal"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
/************** 基礎的固定長度隊列，用來計算移動平均 **************/
|
|||
|
|
type ringQD struct {
|
|||
|
|
N int // 窗口大小（需要保留的資料數量）
|
|||
|
|
l *list.List // 用於儲存資料的雙向鏈表
|
|||
|
|
sum decimal.Decimal // 當前窗口的總和，方便快速計算平均值
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 建立一個固定長度的隊列
|
|||
|
|
func newRingQD(n int) *ringQD {
|
|||
|
|
return &ringQD{N: n, l: list.New(), sum: decimal.Zero}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// push：將新的數值放入隊列，並維護總和
|
|||
|
|
func (q *ringQD) push(x decimal.Decimal) {
|
|||
|
|
q.l.PushBack(x)
|
|||
|
|
q.sum = q.sum.Add(x)
|
|||
|
|
// 如果超出最大長度，移除最舊的數值
|
|||
|
|
if q.l.Len() > q.N {
|
|||
|
|
f := q.l.Front()
|
|||
|
|
q.sum = q.sum.Sub(f.Value.(decimal.Decimal))
|
|||
|
|
q.l.Remove(f)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ready：判斷隊列是否已經填滿
|
|||
|
|
func (q *ringQD) ready() bool { return q.l.Len() == q.N }
|
|||
|
|
|
|||
|
|
// mean：計算平均值
|
|||
|
|
func (q *ringQD) mean() decimal.Decimal {
|
|||
|
|
if q.l.Len() == 0 {
|
|||
|
|
return decimal.Zero
|
|||
|
|
}
|
|||
|
|
return q.sum.Div(decimal.NewFromInt(int64(q.l.Len())))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// values：返回隊列中所有的值（複製一份，不影響原資料）
|
|||
|
|
func (q *ringQD) values() []decimal.Decimal {
|
|||
|
|
out := make([]decimal.Decimal, 0, q.l.Len())
|
|||
|
|
for e := q.l.Front(); e != nil; e = e.Next() {
|
|||
|
|
out = append(out, e.Value.(decimal.Decimal))
|
|||
|
|
}
|
|||
|
|
return out
|
|||
|
|
}
|