1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00

filter flag: split out for separate --filter flags

adding tests and allowing for easy passing of filters.Args from client
to server.

Docker-DCO-1.1-Signed-off-by: Vincent Batts <vbatts@redhat.com> (github: vbatts)
This commit is contained in:
Vincent Batts 2014-05-19 16:31:15 -04:00
parent 3391aac2b2
commit f1cc7ce5d7
5 changed files with 101 additions and 29 deletions

View file

@ -1160,10 +1160,10 @@ func (cli *DockerCli) CmdImages(args ...string) error {
// Consolidate all filter flags, and sanity check them early. // Consolidate all filter flags, and sanity check them early.
// They'll get process in the daemon/server. // They'll get process in the daemon/server.
imageFilters := map[string]string{} imageFilterArgs := filters.Args{}
for _, f := range flFilter.GetAll() { for _, f := range flFilter.GetAll() {
var err error var err error
imageFilters, err = filters.ParseFlag(f, imageFilters) imageFilterArgs, err = filters.ParseFlag(f, imageFilterArgs)
if err != nil { if err != nil {
return err return err
} }
@ -1174,7 +1174,7 @@ func (cli *DockerCli) CmdImages(args ...string) error {
if *flViz || *flTree { if *flViz || *flTree {
v := url.Values{ v := url.Values{
"all": []string{"1"}, "all": []string{"1"},
"filters": []string{filters.ToParam(imageFilters)}, "filters": []string{filters.ToParam(imageFilterArgs)},
} }
body, _, err := readBody(cli.call("GET", "/images/json?"+v.Encode(), nil, false)) body, _, err := readBody(cli.call("GET", "/images/json?"+v.Encode(), nil, false))
if err != nil { if err != nil {
@ -1238,7 +1238,7 @@ func (cli *DockerCli) CmdImages(args ...string) error {
} }
} else { } else {
v := url.Values{ v := url.Values{
"filters": []string{filters.ToParam(imageFilters)}, "filters": []string{filters.ToParam(imageFilterArgs)},
} }
if cmd.NArg() == 1 { if cmd.NArg() == 1 {
// FIXME rename this parameter, to not be confused with the filters flag // FIXME rename this parameter, to not be confused with the filters flag

View file

@ -189,7 +189,7 @@ func getImagesJSON(eng *engine.Engine, version version.Version, w http.ResponseW
) )
job.Setenv("filters", r.Form.Get("filters")) job.Setenv("filters", r.Form.Get("filters"))
// FIXME rename this parameter, to not be confused with the filters flag // FIXME this parameter could just be a match filter
job.Setenv("filter", r.Form.Get("filter")) job.Setenv("filter", r.Form.Get("filter"))
job.Setenv("all", r.Form.Get("all")) job.Setenv("all", r.Form.Get("all"))

View file

@ -700,12 +700,19 @@ func (srv *Server) Images(job *engine.Job) engine.Status {
filt_tagged = true filt_tagged = true
) )
imageFilters, err := filters.ParseFlag(job.Getenv("filters"), nil) utils.Debugf("SUCH JOB: %#v", job)
utils.Debugf("SUCH ENV: %#v", *job.Env())
imageFilters, err := filters.FromParam(job.Getenv("filters"))
if err != nil { if err != nil {
return job.Error(err) return job.Error(err)
} }
if i, ok := imageFilters["untagged"]; ok && strings.ToLower(i) == "true" { utils.Debugf("SUCH FILTERS: %#v", imageFilters)
filt_tagged = false if i, ok := imageFilters["untagged"]; ok {
for _, value := range i {
if strings.ToLower(value) == "true" {
filt_tagged = false
}
}
} }
if job.GetenvBool("all") && !filt_tagged { if job.GetenvBool("all") && !filt_tagged {

View file

@ -2,46 +2,51 @@ package filters
import ( import (
"errors" "errors"
"github.com/dotcloud/docker/pkg/beam/data"
"strings" "strings"
) )
type Args map[string][]string
/* /*
Parse the argument to the filter flag. Like Parse the argument to the filter flag. Like
`docker ps -f 'created=today;image.name=ubuntu*'` `docker ps -f 'created=today' -f 'image.name=ubuntu*'`
Filters delimited by ';', and expected to be 'name=value'
If prev map is provided, then it is appended to, and returned. By default a new If prev map is provided, then it is appended to, and returned. By default a new
map is created. map is created.
*/ */
func ParseFlag(arg string, prev map[string]string) (map[string]string, error) { func ParseFlag(arg string, prev Args) (Args, error) {
var filters map[string]string var filters Args = prev
if prev != nil { if prev == nil {
filters = prev filters = Args{}
} else {
filters = map[string]string{}
} }
if len(arg) == 0 { if len(arg) == 0 {
return filters, nil return filters, nil
} }
for _, chunk := range strings.Split(arg, ";") { if !strings.Contains(arg, "=") {
if !strings.Contains(chunk, "=") { return filters, ErrorBadFormat
return filters, ErrorBadFormat
}
f := strings.SplitN(chunk, "=", 2)
filters[f[0]] = f[1]
} }
f := strings.SplitN(arg, "=", 2)
filters[f[0]] = append(filters[f[0]], f[1])
return filters, nil return filters, nil
} }
var ErrorBadFormat = errors.New("bad format of filter (expected name=value)") var ErrorBadFormat = errors.New("bad format of filter (expected name=value)")
func ToParam(f map[string]string) string { /*
fs := []string{} packs the Args into an string for easy transport from client to server
for k, v := range f { */
fs = append(fs, k+"="+v) func ToParam(a Args) string {
} return data.Encode(a)
return strings.Join(fs, ";") }
/*
unpacks the filter Args
*/
func FromParam(p string) (Args, error) {
return data.Decode(p)
} }

View file

@ -0,0 +1,60 @@
package filters
import (
"sort"
"testing"
)
func TestParseArgs(t *testing.T) {
// equivalent of `docker ps -f 'created=today' -f 'image.name=ubuntu*' -f 'image.name=*untu'`
flagArgs := []string{
"created=today",
"image.name=ubuntu*",
"image.name=*untu",
}
var (
args = Args{}
err error
)
for i := range flagArgs {
args, err = ParseFlag(flagArgs[i], args)
if err != nil {
t.Errorf("failed to parse %s: %s", flagArgs[i], err)
}
}
if len(args["created"]) != 1 {
t.Errorf("failed to set this arg")
}
if len(args["image.name"]) != 2 {
t.Errorf("the args should have collapsed")
}
}
func TestParam(t *testing.T) {
a := Args{
"created": []string{"today"},
"image.name": []string{"ubuntu*", "*untu"},
}
v := ToParam(a)
v1, err := FromParam(v)
if err != nil {
t.Errorf("%s", err)
}
for key, vals := range v1 {
if _, ok := a[key]; !ok {
t.Errorf("could not find key %s in original set", key)
}
sort.Strings(vals)
sort.Strings(a[key])
if len(vals) != len(a[key]) {
t.Errorf("value lengths ought to match")
continue
}
for i := range vals {
if vals[i] != a[key][i] {
t.Errorf("expected %s, but got %s", a[key][i], vals[i])
}
}
}
}