diff --git a/extend/pull-kline.go b/extend/pull-kline.go index 5652c09..81bec3c 100644 --- a/extend/pull-kline.go +++ b/extend/pull-kline.go @@ -7,6 +7,7 @@ import ( "github.com/injoyai/logs" "github.com/injoyai/tdx" "github.com/injoyai/tdx/protocol" + "os" "path/filepath" "sort" "time" @@ -103,6 +104,8 @@ func (this *PullKline) Run(ctx context.Context, m *tdx.Manage) error { go func(code string) { defer limit.Done() + _ = os.MkdirAll(this.Config.Dir, 0777) + //连接数据库 db, err := xorm.NewEngine("sqlite", filepath.Join(this.Config.Dir, code+".db")) if err != nil { @@ -123,13 +126,13 @@ func (this *PullKline) Run(ctx context.Context, m *tdx.Manage) error { default: } - db.Sync2(table) + logs.PrintErr(db.Sync2(table)) //2. 获取最后一条数据 last := new(Kline) if _, err = db.Table(table).Desc("Date").Get(last); err != nil { logs.Err(err) - //return + return } //3. 从服务器获取数据 diff --git a/go.mod b/go.mod index 8221402..c0d7768 100644 --- a/go.mod +++ b/go.mod @@ -4,6 +4,7 @@ go 1.20 require ( github.com/glebarez/go-sqlite v1.22.0 + github.com/go-sql-driver/mysql v1.7.0 github.com/injoyai/base v1.0.18 github.com/injoyai/conv v1.1.10 github.com/injoyai/ios v0.0.4 diff --git a/go.sum b/go.sum index 8eacaf6..6d0123d 100644 --- a/go.sum +++ b/go.sum @@ -14,6 +14,7 @@ github.com/glebarez/go-sqlite v1.22.0 h1:uAcMJhaA6r3LHMTFgP0SifzgXg46yJkgxqyuyec github.com/glebarez/go-sqlite v1.22.0/go.mod h1:PlBIdHe0+aUEFn+r2/uthrWq4FxbzugL0L8Li6yQJbc= github.com/go-sql-driver/mysql v1.4.1/go.mod h1:zAC/RDZ24gD3HViQzih4MyKcchzm+sOG5ZlKdlhCg5w= github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc= +github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI= github.com/goccy/go-json v0.8.1 h1:4/Wjm0JIJaTDm8K1KcGrLHJoa8EsJ13YWeX+6Kfq6uI= github.com/goccy/go-json v0.8.1/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=