This repository has been archived by the owner on Jan 21, 2020. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 262
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: David Chung <[email protected]>
- Loading branch information
David Chung
authored
Feb 20, 2017
1 parent
050a6a9
commit 7841ecd
Showing
27 changed files
with
1,744 additions
and
70 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,219 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "strconv" | ||
|
|
||
| log "github.com/Sirupsen/logrus" | ||
| "github.com/docker/infrakit/pkg/discovery" | ||
| metadata_plugin "github.com/docker/infrakit/pkg/plugin/metadata" | ||
| "github.com/docker/infrakit/pkg/rpc/client" | ||
| metadata_rpc "github.com/docker/infrakit/pkg/rpc/metadata" | ||
| "github.com/docker/infrakit/pkg/spi/metadata" | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| func getPlugin(plugins func() discovery.Plugins, name string) (found metadata.Plugin, err error) { | ||
| err = forPlugin(plugins, func(n string, p metadata.Plugin) error { | ||
| if n == name { | ||
| found = p | ||
| } | ||
| return nil | ||
| }) | ||
| return | ||
| } | ||
|
|
||
| func forPlugin(plugins func() discovery.Plugins, do func(string, metadata.Plugin) error) error { | ||
| all, err := plugins().List() | ||
| if err != nil { | ||
| return err | ||
| } | ||
| for name, endpoint := range all { | ||
| rpcClient, err := client.New(endpoint.Address, metadata.InterfaceSpec) | ||
| if err != nil { | ||
| continue | ||
| } | ||
| if err := do(name, metadata_rpc.Adapt(rpcClient)); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func listAll(m metadata.Plugin, path metadata.Path) ([]metadata.Path, error) { | ||
| if m == nil { | ||
| return nil, fmt.Errorf("no plugin") | ||
| } | ||
| result := []metadata.Path{} | ||
| nodes, err := m.List(path) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| for _, n := range nodes { | ||
| c := path.Join(n) | ||
| more, err := listAll(m, c) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| if len(more) == 0 { | ||
| result = append(result, c) | ||
| } | ||
| for _, pp := range more { | ||
| result = append(result, pp) | ||
| } | ||
| } | ||
| return result, nil | ||
| } | ||
|
|
||
| func metadataCommand(plugins func() discovery.Plugins) *cobra.Command { | ||
|
|
||
| cmd := &cobra.Command{ | ||
| Use: "metadata", | ||
| Short: "Access metadata exposed by infrakit plugins", | ||
| } | ||
|
|
||
| ls := &cobra.Command{ | ||
| Use: "ls", | ||
| Short: "List all metadata entries", | ||
| } | ||
|
|
||
| long := ls.Flags().BoolP("long", "l", false, "Print full path") | ||
| all := ls.Flags().BoolP("all", "a", false, "Find all under the paths given") | ||
|
|
||
| ls.RunE = func(c *cobra.Command, args []string) error { | ||
| paths := []string{"."} | ||
|
|
||
| // All implies long | ||
| if *all { | ||
| *long = true | ||
| } | ||
|
|
||
| if len(args) > 0 { | ||
| paths = args | ||
| } | ||
|
|
||
| for _, p := range paths { | ||
|
|
||
| if p == "/" { | ||
| // TODO(chungers) -- this is a 'local' infrakit ensemble. | ||
| // Absolute paths will come in a multi-cluster / federated model. | ||
| return fmt.Errorf("No absolute path") | ||
| } | ||
|
|
||
| path := metadata_plugin.Path(p) | ||
| first := path.Index(0) | ||
|
|
||
| targets := []string{} // target plugins to query | ||
|
|
||
| // Check all the plugins -- scanning via discovery | ||
| if err := forPlugin(plugins, | ||
| func(name string, mp metadata.Plugin) error { | ||
| if p == "." || (first != nil && name == *first) { | ||
| targets = append(targets, name) | ||
| } | ||
| return nil | ||
| }); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| for _, target := range targets { | ||
|
|
||
| nodes := []metadata.Path{} // the result set to print | ||
|
|
||
| match, err := getPlugin(plugins, target) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if p == "." { | ||
| if *all { | ||
| allPaths, err := listAll(match, path.Shift(1)) | ||
| if err != nil { | ||
| log.Warningln("Cannot metadata ls on plugin", target, "err=", err) | ||
| } | ||
| for _, c := range allPaths { | ||
| nodes = append(nodes, metadata_plugin.Path(target).Sub(c)) | ||
| } | ||
| } else { | ||
| for _, t := range targets { | ||
| nodes = append(nodes, metadata_plugin.Path(t)) | ||
| } | ||
| } | ||
| } else { | ||
| if *all { | ||
| allPaths, err := listAll(match, path.Shift(1)) | ||
| if err != nil { | ||
| log.Warningln("Cannot metadata ls on plugin", target, "err=", err) | ||
| } | ||
| for _, c := range allPaths { | ||
| nodes = append(nodes, metadata_plugin.Path(target).Sub(c)) | ||
| } | ||
| } else { | ||
| children, err := match.List(path.Shift(1)) | ||
| if err != nil { | ||
| log.Warningln("Cannot metadata ls on plugin", target, "err=", err) | ||
| } | ||
| for _, c := range children { | ||
| nodes = append(nodes, path.Join(c)) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| if len(targets) > 1 { | ||
| fmt.Printf("%s:\n", target) | ||
| } | ||
| if *long { | ||
| fmt.Printf("total %d:\n", len(nodes)) | ||
| } | ||
| for _, l := range nodes { | ||
| if *long { | ||
| fmt.Println(metadata_plugin.String(l)) | ||
| } else { | ||
| fmt.Println(metadata_plugin.String(l.Rel(path))) | ||
| } | ||
| } | ||
| fmt.Println() | ||
| } | ||
|
|
||
| } | ||
| return nil | ||
| } | ||
|
|
||
| cat := &cobra.Command{ | ||
| Use: "cat", | ||
| Short: "Get metadata entry by path", | ||
| RunE: func(c *cobra.Command, args []string) error { | ||
|
|
||
| for _, p := range args { | ||
|
|
||
| path := metadata_plugin.Path(p) | ||
| first := path.Index(0) | ||
| if first != nil { | ||
| match, err := getPlugin(plugins, *first) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| value, err := match.Get(path.Shift(1)) | ||
| if err == nil { | ||
| if value != nil { | ||
| str := value.String() | ||
| if s, err := strconv.Unquote(value.String()); err == nil { | ||
| str = s | ||
| } | ||
| fmt.Println(str) | ||
| } | ||
|
|
||
| } else { | ||
| log.Warningln("Cannot metadata cat on plugin", *first, "err=", err) | ||
| } | ||
| } | ||
| } | ||
| return nil | ||
| }, | ||
| } | ||
|
|
||
| cmd.AddCommand(ls, cat) | ||
|
|
||
| return cmd | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.