-
Notifications
You must be signed in to change notification settings - Fork 1
/
stats.go
67 lines (57 loc) · 2.02 KB
/
stats.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package commands // import "github.com/SUSE/groot-btrfs/commands"
import (
"encoding/json"
"fmt"
"os"
"code.cloudfoundry.org/lager"
"github.com/SUSE/groot-btrfs/commands/config"
"github.com/SUSE/groot-btrfs/commands/idfinder"
"github.com/SUSE/groot-btrfs/groot"
"github.com/SUSE/groot-btrfs/metrics"
imageClonerpkg "github.com/SUSE/groot-btrfs/store/image_cloner"
errorspkg "github.com/pkg/errors"
"github.com/urfave/cli"
)
var StatsCommand = cli.Command{
Name: "stats",
Usage: "stats [options] <id|image path>",
Description: "Return filesystem stats",
Action: func(ctx *cli.Context) error {
logger := ctx.App.Metadata["logger"].(lager.Logger)
logger = logger.Session("stats")
newExitError := newErrorHandler(logger, "stats")
if ctx.NArg() != 1 {
logger.Error("parsing-command", errorspkg.New("invalid arguments"), lager.Data{"args": ctx.Args()})
return newExitError(fmt.Sprintf("invalid arguments - usage: %s", ctx.Command.Usage), 1)
}
configBuilder := ctx.App.Metadata["configBuilder"].(*config.Builder)
cfg, err := configBuilder.Build()
logger.Debug("stats-config", lager.Data{"currentConfig": cfg})
if err != nil {
logger.Error("config-builder-failed", err)
return newExitError(err.Error(), 1)
}
storePath := cfg.StorePath
idOrPath := ctx.Args().First()
id, err := idfinder.FindID(storePath, idOrPath)
if err != nil {
logger.Error("find-id-failed", err, lager.Data{"id": idOrPath, "storePath": storePath})
return newExitError(err.Error(), 1)
}
fsDriver, err := createFileSystemDriver(cfg)
if err != nil {
return newExitError(err.Error(), 1)
}
imageCloner := imageClonerpkg.NewImageCloner(fsDriver, storePath)
metricsEmitter := metrics.NewEmitter()
statser := groot.IamStatser(imageCloner, metricsEmitter)
stats, err := statser.Stats(logger, id)
if err != nil {
logger.Error("fetching-stats", err)
return newExitError(err.Error(), 1)
}
_ = json.NewEncoder(os.Stdout).Encode(stats)
metricsEmitter.TryIncrementRunCount("stats", nil)
return nil
},
}