78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"io/ioutil"
|
|||
|
|
"log"
|
|||
|
|
|
|||
|
|
"git.apinb.com/quant/gostock/internal/config"
|
|||
|
|
"git.apinb.com/quant/gostock/internal/impl"
|
|||
|
|
"git.apinb.com/quant/gostock/internal/models"
|
|||
|
|
"github.com/bitly/go-simplejson"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
ServiceKey = "gostock"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
log.Println("Hello Cli!")
|
|||
|
|
config.New(ServiceKey)
|
|||
|
|
impl.NewImpl()
|
|||
|
|
|
|||
|
|
var stocks []models.StockBasic
|
|||
|
|
impl.DBService.Find(&stocks)
|
|||
|
|
|
|||
|
|
for _, stock := range stocks {
|
|||
|
|
var args models.StockArgs
|
|||
|
|
// 读取JSON：C:\Users\Administrator\Documents\stock-main\stock\reports\r_000001.SZ.json
|
|||
|
|
jsonBytes, err := ioutil.ReadFile("C:\\Users\\Administrator\\Documents\\stock-main\\stock\\reports\\r_" + stock.TsCode + ".json")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println(err)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
js, err := simplejson.NewJson(jsonBytes)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println(err)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
args.TsCode = stock.TsCode
|
|||
|
|
c, err := js.Get("comparison").Encode()
|
|||
|
|
if err == nil {
|
|||
|
|
args.Config = string(c)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
rsi_period, err := js.Get("strategies").Get("rsi").Get("best_params").Get("rsi_period").Int()
|
|||
|
|
if err == nil {
|
|||
|
|
args.RsiPeriod = rsi_period
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
rsi_oversold, err := js.Get("strategies").Get("rsi").Get("best_params").Get("rsi_oversold").Int()
|
|||
|
|
if err == nil {
|
|||
|
|
args.RsiOversold = rsi_oversold
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
rsi_overbought, err := js.Get("strategies").Get("rsi").Get("best_params").Get("rsi_overbought").Int()
|
|||
|
|
if err == nil {
|
|||
|
|
args.RsiOverbought = rsi_overbought
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ema_fast, err := js.Get("strategies").Get("ema").Get("best_params").Get("ema_fast").Int()
|
|||
|
|
if err == nil {
|
|||
|
|
args.EmaFast = ema_fast
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ema_slow, err := js.Get("strategies").Get("ema").Get("best_params").Get("ema_slow").Int()
|
|||
|
|
if err == nil {
|
|||
|
|
args.EmaSlow = ema_slow
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var cnt int64
|
|||
|
|
impl.DBService.Model(&models.StockArgs{}).Where("ts_code = ?", stock.TsCode).Count(&cnt)
|
|||
|
|
if cnt > 0 {
|
|||
|
|
impl.DBService.Model(&models.StockArgs{}).Where("ts_code = ?", stock.TsCode).Updates(args)
|
|||
|
|
} else {
|
|||
|
|
impl.DBService.Create(&args)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|