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)
|
||
}
|
||
}
|
||
}
|