Added an impl for stating a file

This commit is contained in:
Diptanu Choudhury 2016-01-26 15:03:26 -08:00
parent edd3194a16
commit a70e5dbdc7
4 changed files with 125 additions and 9 deletions

View File

@ -52,3 +52,38 @@ func (a *AllocFS) List(alloc *Allocation, path string, q *QueryOptions) ([]*allo
}
return files, nil, nil
}
// Stat is used to stat a file at a given path of an allocation directory
func (a *AllocFS) Stat(alloc *Allocation, path string, q *QueryOptions) (*allocdir.AllocFileInfo, *QueryMeta, error) {
node, _, err := a.client.Nodes().Info(alloc.NodeID, &QueryOptions{})
if err != nil {
return nil, nil, err
}
if node.HTTPAddr == "" {
return nil, nil, fmt.Errorf("http addr of the node where alloc %q is running is not advertised", alloc.ID)
}
u := &url.URL{
Scheme: "http",
Host: node.HTTPAddr,
Path: fmt.Sprintf("/v1/client/fs/stat/%s", alloc.ID),
}
v := url.Values{}
v.Set("path", path)
u.RawQuery = v.Encode()
req := &http.Request{
Method: "GET",
URL: u,
}
c := http.Client{}
resp, err := c.Do(req)
if err != nil {
return nil, nil, err
}
decoder := json.NewDecoder(resp.Body)
var file *allocdir.AllocFileInfo
if err := decoder.Decode(&file); err != nil {
return nil, nil, err
}
return file, nil, nil
}

View File

@ -23,9 +23,9 @@ func (f *FSListCommand) Synopsis() string {
return "Displays list of files of an alloc dir"
}
func (c *FSListCommand) Run(args []string) int {
flags := c.Meta.FlagSet("fs-list", FlagSetClient)
flags.Usage = func() { c.Ui.Output(c.Help()) }
func (f *FSListCommand) Run(args []string) int {
flags := f.Meta.FlagSet("fs-list", FlagSetClient)
flags.Usage = func() { f.Ui.Output(f.Help()) }
if err := flags.Parse(args); err != nil {
return 1
@ -33,7 +33,7 @@ func (c *FSListCommand) Run(args []string) int {
args = flags.Args()
if len(args) < 1 {
c.Ui.Error("a valid alloc id is essential")
f.Ui.Error("a valid alloc id is essential")
return 1
}
@ -43,21 +43,21 @@ func (c *FSListCommand) Run(args []string) int {
path = args[1]
}
client, err := c.Meta.Client()
client, err := f.Meta.Client()
if err != nil {
c.Ui.Error(fmt.Sprintf("Error inititalizing client: %v", err))
f.Ui.Error(fmt.Sprintf("Error inititalizing client: %v", err))
return 1
}
alloc, _, err := client.Allocations().Info(allocID, nil)
if err != nil {
c.Ui.Error(fmt.Sprintf("Error getting alloc: %v", err))
f.Ui.Error(fmt.Sprintf("Error getting alloc: %v", err))
return 1
}
files, _, err := client.AllocFS().List(alloc, path, nil)
if err != nil {
c.Ui.Error(fmt.Sprintf("Error listing alloc dir: %v", err))
f.Ui.Error(fmt.Sprintf("Error listing alloc dir: %v", err))
return 1
}
@ -73,6 +73,6 @@ func (c *FSListCommand) Run(args []string) int {
file.Size)
}
c.Ui.Output(formatList(out))
f.Ui.Output(formatList(out))
return 0
}

75
command/fs_stat.go Normal file
View File

@ -0,0 +1,75 @@
package command
import (
"fmt"
"strings"
)
type FSStatCommand struct {
Meta
}
func (f *FSStatCommand) Help() string {
helpText := `
Usage: nomad fs-stat [alloc-id] [path]
Displays information about a file in an allocation directory at the given path.
The path is relative to the allocation directory.
`
return strings.TrimSpace(helpText)
}
func (f *FSStatCommand) Synopsis() string {
return "Stats a file in an allocation directory"
}
func (f *FSStatCommand) Run(args []string) int {
flags := f.Meta.FlagSet("fs-list", FlagSetClient)
flags.Usage = func() { f.Ui.Output(f.Help()) }
if err := flags.Parse(args); err != nil {
return 1
}
args = flags.Args()
if len(args) < 1 {
f.Ui.Error("a valid alloc id is essential")
return 1
}
allocID := args[0]
path := "/"
if len(args) == 2 {
path = args[1]
}
client, err := f.Meta.Client()
if err != nil {
f.Ui.Error(fmt.Sprintf("Error inititalizing client: %v", err))
return 1
}
alloc, _, err := client.Allocations().Info(allocID, nil)
if err != nil {
f.Ui.Error(fmt.Sprintf("Error getting alloc: %v", err))
return 1
}
file, _, err := client.AllocFS().Stat(alloc, path, nil)
if err != nil {
f.Ui.Error(fmt.Sprintf("Error stating file: %v:", err))
return 1
}
out := make([]string, 2)
out[0] = "Name|Size"
if file != nil {
fn := file.Name
if file.IsDir {
fn = fmt.Sprintf("%s/", fn)
}
out[1] = fmt.Sprintf("%s|%d", fn, file.Size)
}
f.Ui.Output(formatList(out))
return 0
}

View File

@ -62,6 +62,12 @@ func Commands(metaPtr *command.Meta) map[string]cli.CommandFactory {
Meta: meta,
}, nil
},
"fs-stat": func() (cli.Command, error) {
return &command.FSStatCommand{
Meta: meta,
}, nil
},
"init": func() (cli.Command, error) {
return &command.InitCommand{
Meta: meta,