storj/cmd/overlay/main.go
2018-11-30 15:40:13 +02:00

125 lines
2.3 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/pb"
"storj.io/storj/pkg/process"
"storj.io/storj/pkg/storj"
)
var (
// Error is the error class for overlays
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
}
nodeIDs, err := storj.NodeIDsFromBytes(keys.ByteSlices())
if err != nil {
return err
}
for _, id := range nodeIDs {
n, err := c.Get(process.Ctx(cmd), id)
if err != nil {
zap.S().Infof("ID: %s; error getting value\n", id.String())
}
if n != nil {
zap.S().Infof("ID: %s; Address: %s\n", id.String(), n.Address.Address)
continue
}
zap.S().Infof("ID: %s: nil\n", id.String())
}
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 {
id, err := storj.NodeIDFromString(i)
if err != nil {
zap.S().Error(err)
}
zap.S().Infof("adding node ID: %s; Address: %s", i, a)
err = c.Put(id, pb.Node{
Id: id,
// TODO: NodeType is missing
Address: &pb.NodeAddress{
Transport: 0,
Address: a,
},
Restrictions: &pb.NodeRestrictions{
FreeBandwidth: 2000000000,
FreeDisk: 2000000000,
},
Type: 1,
})
if err != nil {
return err
}
}
return nil
}
func main() {
process.Exec(rootCmd)
}