优化首次拉取会报错的问题(未创建目录)

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"
@@ -41,6 +42,18 @@ var (
 		Quarter:  NewKlineTable("QuarterKline", func(c *tdx.Client) KlineHandler { return c.GetKlineQuarterUntil }),
 		Year:     NewKlineTable("YearKline", func(c *tdx.Client) KlineHandler { return c.GetKlineYearUntil }),
 	}
+	AllKlineTables = []string{
+		"MinuteKline",
+		"Minute5Kline",
+		"Minute15Kline",
+		"Minute30Kline",
+		"HourKline",
+		"DayKline",
+		"WeekKline",
+		"MonthKline",
+		"QuarterKline",
+		"YearKline",
+	}
 )
 
 type PullKlineConfig struct {
@@ -91,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 {
@@ -111,7 +126,7 @@ func (this *PullKline) Run(ctx context.Context, m *tdx.Manage) error {
 				default:
 				}
 
-				db.Sync2(table)
+				logs.PrintErr(db.Sync2(table))
 
 				//2. 获取最后一条数据
 				last := new(Kline)

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

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=