lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-25 16:25:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

make the first arg of WithCache as a key var

Browse Source
This commit is contained in:
c9s 2020-10-20 12:18:29 +08:00
parent 40c697275d
commit 2bbee6671a
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/bbgo/cache.go
View File

@ -10,9 +10,9 @@ import (
"github.com/pkg/errors" "github.com/pkg/errors"
) )
func WithCache(file string, obj interface{}, do func() (interface{}, error)) error { func WithCache(key string, obj interface{}, do func() (interface{}, error)) error {
cacheDir := CacheDir() cacheDir := CacheDir()
cacheFile := path.Join(cacheDir, file) cacheFile := path.Join(cacheDir, key+ ".json")
if _, err := os.Stat(cacheFile); os.IsNotExist(err) { if _, err := os.Stat(cacheFile); os.IsNotExist(err) {
data, err := do() data, err := do()

2
pkg/bbgo/environment.go
View File

@ -63,7 +63,7 @@ func (environ *Environment) Init(ctx context.Context) (err error) {
for _, session := range environ.sessions { for _, session := range environ.sessions {
var markets types.MarketMap var markets types.MarketMap
err = WithCache(fmt.Sprintf("%s-markets.json", session.Exchange.Name()), &markets, func() (interface{}, error) { err = WithCache(fmt.Sprintf("%s-markets", session.Exchange.Name()), &markets, func() (interface{}, error) {
return session.Exchange.QueryMarkets(ctx) return session.Exchange.QueryMarkets(ctx)
}) })
if err != nil { if err != nil {