5014a785a0
* remove api key from statdb server reqs; add statdb UpdateUptime and UpdateAuditSuccess to server * update api key authentication in statdb server * add todos for future statdb updates * add UpdateUptime and UpdateAuditSuccess to statdb server * fix apikey stuff in config.go and statdb_test.go * fix tests * update sdbclient.NewClient call in audit package * fix UpdateUptime and UpdateAuditSuccess in sdbclient * set api key from statdb/config.go * change package for statdb tests * linter fixes * remove todo comments * fix sdbclient err checking * move validate auth functionality to auth package * update description for statdb api key * remove import
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package statdb
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/storj/pkg/provider"
|
|
pb "storj.io/storj/pkg/statdb/proto"
|
|
)
|
|
|
|
//CtxKey Used as statdb key
|
|
type CtxKey int
|
|
|
|
const (
|
|
ctxKeyStats CtxKey = iota
|
|
)
|
|
|
|
var (
|
|
apiKey = flag.String("stat-db.auth.api-key", "", "statdb api key")
|
|
)
|
|
|
|
// Config is a configuration struct that is everything you need to start a
|
|
// StatDB responsibility
|
|
type Config struct {
|
|
DatabaseURL string `help:"the database connection string to use" default:"$CONFDIR/stats.db"`
|
|
DatabaseDriver string `help:"the database driver to use" default:"sqlite3"`
|
|
}
|
|
|
|
// Run implements the provider.Responsibility interface
|
|
func (c Config) Run(ctx context.Context, server *provider.Provider) error {
|
|
ns, err := NewServer(c.DatabaseDriver, c.DatabaseURL, *apiKey, zap.L())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pb.RegisterStatDBServer(server.GRPC(), ns)
|
|
|
|
return server.Run(context.WithValue(ctx, ctxKeyStats, ns))
|
|
}
|
|
|
|
// LoadFromContext loads an existing StatDB from the Provider context
|
|
// stack if one exists.
|
|
func LoadFromContext(ctx context.Context) *Server {
|
|
if v, ok := ctx.Value(ctxKeyStats).(*Server); ok {
|
|
return v
|
|
}
|
|
return nil
|
|
}
|