storj/cmd/overlay/cache.go

35 lines
912 B
Go
Raw Normal View History

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.
package main
import (
"context"
"fmt"
"github.com/zeebo/errs"
"storj.io/storj/pkg/overlay"
"storj.io/storj/satellite/satellitedb"
)
type cacheConfig struct {
NodesPath string `help:"the path to a JSON file containing an object with IP keys and nodeID values"`
Database string `help:"overlay database connection string" default:"sqlite3://$CONFDIR/master.db"`
}
func (c cacheConfig) open(ctx context.Context) (cache *overlay.Cache, dbClose func(), err error) {
database, err := satellitedb.New(c.Database)
if err != nil {
return nil, nil, errs.New("error connecting to database: %+v", err)
}
dbClose = func() {
err := database.Close()
if err != nil {
fmt.Printf("error closing connection to database: %+v\n", err)
}
}
2018-11-16 16:31:14 +00:00
return overlay.NewCache(database.OverlayCache(), database.StatDB()), dbClose, nil
}