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

Merge pull request #12505 from ZJU-SEL/remove_job_from_push

remove job from push
This commit is contained in:
Alexander Morozov 2015-04-23 09:57:17 -07:00
commit 2351b87551
3 changed files with 33 additions and 34 deletions

View file

@ -858,25 +858,26 @@ func (s *Server) postImagesPush(eng *engine.Engine, version version.Version, w h
} }
} }
job := eng.Job("push", vars["name"]) useJSON := version.GreaterThan("1.0")
job.SetenvJson("metaHeaders", metaHeaders) name := vars["name"]
job.SetenvJson("authConfig", authConfig)
job.Setenv("tag", r.Form.Get("tag")) imagePushConfig := &graph.ImagePushConfig{
if version.GreaterThan("1.0") { MetaHeaders: metaHeaders,
job.SetenvBool("json", true) AuthConfig: authConfig,
streamJSON(job.Stdout, w, true) Tag: r.Form.Get("tag"),
} else { OutStream: utils.NewWriteFlusher(w),
job.Stdout.Add(utils.NewWriteFlusher(w)) Json: useJSON,
}
if useJSON {
w.Header().Set("Content-Type", "application/json")
} }
if err := job.Run(); err != nil { if err := s.daemon.Repositories().Push(name, imagePushConfig); err != nil {
if !job.Stdout.Used() { sf := streamformatter.NewStreamFormatter(useJSON)
return err return fmt.Errorf(string(sf.FormatError(err)))
}
sf := streamformatter.NewStreamFormatter(version.GreaterThan("1.0"))
w.Write(sf.FormatError(err))
} }
return nil return nil
} }
func (s *Server) getImagesGet(eng *engine.Engine, version version.Version, w http.ResponseWriter, r *http.Request, vars map[string]string) error { func (s *Server) getImagesGet(eng *engine.Engine, version version.Version, w http.ResponseWriter, r *http.Request, vars map[string]string) error {

View file

@ -12,7 +12,6 @@ import (
"github.com/Sirupsen/logrus" "github.com/Sirupsen/logrus"
"github.com/docker/distribution/digest" "github.com/docker/distribution/digest"
"github.com/docker/docker/engine"
"github.com/docker/docker/image" "github.com/docker/docker/image"
"github.com/docker/docker/pkg/progressreader" "github.com/docker/docker/pkg/progressreader"
"github.com/docker/docker/pkg/streamformatter" "github.com/docker/docker/pkg/streamformatter"
@ -25,6 +24,14 @@ import (
var ErrV2RegistryUnavailable = errors.New("error v2 registry unavailable") var ErrV2RegistryUnavailable = errors.New("error v2 registry unavailable")
type ImagePushConfig struct {
MetaHeaders map[string][]string
AuthConfig *registry.AuthConfig
Tag string
Json bool
OutStream io.Writer
}
// Retrieve the all the images to be uploaded in the correct order // Retrieve the all the images to be uploaded in the correct order
func (s *TagStore) getImageList(localRepo map[string]string, requestedTag string) ([]string, map[string][]string, error) { func (s *TagStore) getImageList(localRepo map[string]string, requestedTag string) ([]string, map[string][]string, error) {
var ( var (
@ -486,15 +493,9 @@ func (s *TagStore) pushV2Image(r *registry.Session, img *image.Image, endpoint *
} }
// FIXME: Allow to interrupt current push when new push of same image is done. // FIXME: Allow to interrupt current push when new push of same image is done.
func (s *TagStore) CmdPush(job *engine.Job) error { func (s *TagStore) Push(localName string, imagePushConfig *ImagePushConfig) error {
if n := len(job.Args); n != 1 {
return fmt.Errorf("Usage: %s IMAGE", job.Name)
}
var ( var (
localName = job.Args[0] sf = streamformatter.NewStreamFormatter(imagePushConfig.Json)
sf = streamformatter.NewStreamFormatter(job.GetenvBool("json"))
authConfig = &registry.AuthConfig{}
metaHeaders map[string][]string
) )
// Resolve the Repository name from fqn to RepositoryInfo // Resolve the Repository name from fqn to RepositoryInfo
@ -503,10 +504,6 @@ func (s *TagStore) CmdPush(job *engine.Job) error {
return err return err
} }
tag := job.Getenv("tag")
job.GetenvJson("authConfig", authConfig)
job.GetenvJson("metaHeaders", &metaHeaders)
if _, err := s.poolAdd("push", repoInfo.LocalName); err != nil { if _, err := s.poolAdd("push", repoInfo.LocalName); err != nil {
return err return err
} }
@ -517,16 +514,18 @@ func (s *TagStore) CmdPush(job *engine.Job) error {
return err return err
} }
r, err := registry.NewSession(authConfig, registry.HTTPRequestFactory(metaHeaders), endpoint, false) r, err := registry.NewSession(imagePushConfig.AuthConfig, registry.HTTPRequestFactory(imagePushConfig.MetaHeaders), endpoint, false)
if err != nil { if err != nil {
return err return err
} }
reposLen := 1 reposLen := 1
if tag == "" { if imagePushConfig.Tag == "" {
reposLen = len(s.Repositories[repoInfo.LocalName]) reposLen = len(s.Repositories[repoInfo.LocalName])
} }
job.Stdout.Write(sf.FormatStatus("", "The push refers to a repository [%s] (len: %d)", repoInfo.CanonicalName, reposLen))
imagePushConfig.OutStream.Write(sf.FormatStatus("", "The push refers to a repository [%s] (len: %d)", repoInfo.CanonicalName, reposLen))
// If it fails, try to get the repository // If it fails, try to get the repository
localRepo, exists := s.Repositories[repoInfo.LocalName] localRepo, exists := s.Repositories[repoInfo.LocalName]
if !exists { if !exists {
@ -534,7 +533,7 @@ func (s *TagStore) CmdPush(job *engine.Job) error {
} }
if repoInfo.Index.Official || endpoint.Version == registry.APIVersion2 { if repoInfo.Index.Official || endpoint.Version == registry.APIVersion2 {
err := s.pushV2Repository(r, localRepo, job.Stdout, repoInfo, tag, sf) err := s.pushV2Repository(r, localRepo, imagePushConfig.OutStream, repoInfo, imagePushConfig.Tag, sf)
if err == nil { if err == nil {
s.eventsService.Log("push", repoInfo.LocalName, "") s.eventsService.Log("push", repoInfo.LocalName, "")
return nil return nil
@ -545,7 +544,7 @@ func (s *TagStore) CmdPush(job *engine.Job) error {
} }
} }
if err := s.pushRepository(r, job.Stdout, repoInfo, localRepo, tag, sf); err != nil { if err := s.pushRepository(r, imagePushConfig.OutStream, repoInfo, localRepo, imagePushConfig.Tag, sf); err != nil {
return err return err
} }
s.eventsService.Log("push", repoInfo.LocalName, "") s.eventsService.Log("push", repoInfo.LocalName, "")

View file

@ -13,7 +13,6 @@ func (s *TagStore) Install(eng *engine.Engine) error {
"image_inspect": s.CmdLookup, "image_inspect": s.CmdLookup,
"image_export": s.CmdImageExport, "image_export": s.CmdImageExport,
"viz": s.CmdViz, "viz": s.CmdViz,
"push": s.CmdPush,
} { } {
if err := eng.Register(name, handler); err != nil { if err := eng.Register(name, handler); err != nil {
return fmt.Errorf("Could not register %q: %v", name, err) return fmt.Errorf("Could not register %q: %v", name, err)