phyer/siaga
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stock rsi push to elasticsearch

Browse Source
This commit is contained in:
zhangkun9038@dingtalk.com 2024-12-21 22:19:35 +08:00
parent 339445b1dd
commit 57e37b5ae4
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/extent.go
View File

@ -275,16 +275,18 @@ func MakeRsi(cr *core.Core, cl *core.Candle, count int) (error, int) {
// keyName := "candle" + cl.Period + "|" + cl.InstID + "|ts:" + tss // keyName := "candle" + cl.Period + "|" + cl.InstID + "|ts:" + tss
lastTime := time.UnixMilli(tsi) lastTime := time.UnixMilli(tsi)
setName := "candle" + cl.Period + "|" + cl.InstID + "|sortedSet" setName := "candle" + cl.Period + "|" + cl.InstID + "|sortedSet"
dcount := count * 2 // dcount := count * 2
cdl, err := GetRangeCandleSortedSet(cr, setName, dcount, lastTime) cdl, err := GetRangeCandleSortedSet(cr, setName, count+3, lastTime)
if err != nil { if err != nil {
return err, 0 return err, 0
} }
// amountLast := float64(0) // amountLast := float64(0)
// ct := float64(0) // ct := float64(0)
if len(cdl.List) == 0 { if len(cdl.List)+3 < count {
err = errors.New("sortedSet长度不足,无法进行rsi计算")
return err, 0 return err, 0
} }
cdl.RecursiveBubbleS(len(cdl.List), "asc")
closeList := []float64{} closeList := []float64{}
for k, v := range cdl.List { for k, v := range cdl.List {
fmt.Println("candle in list", k, v) fmt.Println("candle in list", k, v)
@ -295,15 +297,11 @@ func MakeRsi(cr *core.Core, cl *core.Candle, count int) (error, int) {
fmt.Println("Error calculating RSI:", err) fmt.Println("Error calculating RSI:", err)
return err, 0 return err, 0
} }
// rv, err := CalculateRSI(closeList, dcount)
percentK, percentD, err := CalculateStochRSI(rsiList, count, 3, 3) percentK, percentD, err := CalculateStochRSI(rsiList, count, 3, 3)
if err != nil { if err != nil {
fmt.Println("Error calculating StochRSI:", err) fmt.Println("Error calculating StochRSI:", err)
return err, 0 return err, 0
} }
if err != nil {
logrus.Error("CalculateRSI err: ", err)
}
rsi := core.StockRsi{ rsi := core.StockRsi{
InstID: cl.InstID, InstID: cl.InstID,
Period: cl.Period, Period: cl.Period,