storj/cmd/overlay/main.go
Bryan White f932c3ced8 Overlay cache commands (#236)
* wip overlay cache cmd

* overlay cache add and list commands

* use zap

* fixes

* simplify

* add restrictions

* cleanup

* remove defaults for overlay management commands
2018-08-20 14:21:41 -04:00

113 lines
2.0 KiB
Go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.
package main
import (
"encoding/json"
"io/ioutil"
"github.com/spf13/cobra"
"github.com/zeebo/errs"
"go.uber.org/zap"
"storj.io/storj/pkg/cfgstruct"
"storj.io/storj/pkg/process"
proto "storj.io/storj/protos/overlay"
)
var (
Error = errs.Class("overlay error")
rootCmd = &cobra.Command{
Use: "overlay",
Short: "Overlay cache management",
}
addCmd = &cobra.Command{
Use: "add",
Short: "Add nodes to the overlay cache",
RunE: cmdAdd,
}
listCmd = &cobra.Command{
Use: "list",
Short: "List nodes in the overlay cache",
RunE: cmdList,
}
cacheCfg struct {
cacheConfig
}
)
func init() {
rootCmd.AddCommand(addCmd)
rootCmd.AddCommand(listCmd)
cfgstruct.Bind(addCmd.Flags(), &cacheCfg)
cfgstruct.Bind(listCmd.Flags(), &cacheCfg)
}
func cmdList(cmd *cobra.Command, args []string) (err error) {
c, err := cacheCfg.open()
if err != nil {
return err
}
keys, err := c.DB.List(nil, 0)
if err != nil {
return err
}
for _, k := range keys {
n, err := c.Get(process.Ctx(cmd), string(k))
if err != nil {
zap.S().Infof("ID: %s; error getting value\n", k)
}
if n != nil {
zap.S().Infof("ID: %s; Address: %s\n", k, n.Address.Address)
continue
}
zap.S().Infof("ID: %s: nil\n", k)
}
return nil
}
func cmdAdd(cmd *cobra.Command, args []string) (err error) {
j, err := ioutil.ReadFile(cacheCfg.NodesPath)
if err != nil {
return errs.Wrap(err)
}
var nodes map[string]string
if err := json.Unmarshal(j, &nodes); err != nil {
return errs.Wrap(err)
}
c, err := cacheCfg.open()
if err != nil {
return err
}
for i, a := range nodes {
zap.S().Infof("adding node ID: %s; Address: %s", i, a)
err := c.Put(i, proto.Node{
Id: i,
Address: &proto.NodeAddress{
Transport: 0,
Address: a,
},
Restrictions: &proto.NodeRestrictions{
FreeBandwidth: 2000000000,
FreeDisk: 2000000000,
},
Type: 1,
})
if err != nil {
return err
}
}
return nil
}
func main() {
process.Exec(rootCmd)
}