-
Notifications
You must be signed in to change notification settings - Fork 23
/
filestats.go
70 lines (64 loc) · 2.19 KB
/
filestats.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
68
69
70
package cmd
import (
"os"
"strconv"
"github.com/0chain/gosdk/zboxcore/sdk"
"github.com/0chain/zboxcli/util"
"github.com/spf13/cobra"
)
// statsCmd represents list command
var statsCmd = &cobra.Command{
Use: "stats",
Short: "stats for file from blobbers",
Long: `stats for file from blobbers`,
Args: cobra.MinimumNArgs(0),
Run: func(cmd *cobra.Command, args []string) {
fflags := cmd.Flags() // fflags is a *flag.FlagSet
if fflags.Changed("allocation") == false { // check if the flag "path" is set
PrintError("Error: allocation flag is missing") // If not, we'll let the user know
os.Exit(1) // and return
}
if fflags.Changed("remotepath") == false {
PrintError("Error: remotepath flag is missing")
os.Exit(1)
}
allocationID := cmd.Flag("allocation").Value.String()
doJSON, _ := cmd.Flags().GetBool("json")
allocationObj, err := sdk.GetAllocation(allocationID)
if err != nil {
PrintError("Error fetching the allocation", err)
os.Exit(1)
}
remotepath := cmd.Flag("remotepath").Value.String()
ref, err := allocationObj.GetFileStats(remotepath)
if err != nil {
PrintError(err.Error())
os.Exit(1)
}
if doJSON {
util.PrintJSON(ref)
return
}
header := []string{"Blobber", "Name", "Path", "Size", "Uploads", "Block Downloads", "Challenges", "Blockchain Aware"}
data := make([][]string, 0)
idx := 0
for k, v := range ref {
if v != nil {
size := strconv.FormatInt(v.Size, 10)
rowData := []string{k, v.Name, v.Path, size, strconv.FormatInt(v.NumUpdates, 10), strconv.FormatInt(v.NumBlockDownloads, 10), strconv.FormatInt(v.SuccessChallenges, 10), strconv.FormatBool(len(v.WriteMarkerRedeemTxn) > 0)}
data = append(data, rowData)
idx++
}
}
util.WriteTable(os.Stdout, header, []string{}, data)
return
},
}
func init() {
rootCmd.AddCommand(statsCmd)
statsCmd.PersistentFlags().String("allocation", "", "Allocation ID")
statsCmd.PersistentFlags().String("remotepath", "", "Remote path to list from")
statsCmd.MarkFlagRequired("allocation")
statsCmd.MarkFlagRequired("remotepath")
statsCmd.Flags().Bool("json", false, "pass this option to print response as json data")
}