| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075 |
- /*
- Copyright 2020 Docker Compose CLI authors
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
- package local
- import (
- "context"
- "encoding/base64"
- "encoding/json"
- "fmt"
- "io"
- "path/filepath"
- "sort"
- "strconv"
- "strings"
- "github.com/compose-spec/compose-go/cli"
- "github.com/compose-spec/compose-go/types"
- "github.com/docker/buildx/build"
- cliconfig "github.com/docker/cli/cli/config"
- "github.com/docker/distribution/reference"
- moby "github.com/docker/docker/api/types"
- "github.com/docker/docker/api/types/container"
- "github.com/docker/docker/api/types/filters"
- "github.com/docker/docker/api/types/mount"
- "github.com/docker/docker/api/types/network"
- "github.com/docker/docker/api/types/strslice"
- mobyvolume "github.com/docker/docker/api/types/volume"
- "github.com/docker/docker/client"
- "github.com/docker/docker/errdefs"
- "github.com/docker/docker/pkg/jsonmessage"
- "github.com/docker/docker/pkg/stdcopy"
- "github.com/docker/docker/registry"
- "github.com/docker/go-connections/nat"
- "github.com/pkg/errors"
- "github.com/sanathkr/go-yaml"
- "golang.org/x/sync/errgroup"
- "github.com/docker/compose-cli/api/compose"
- "github.com/docker/compose-cli/config"
- errdefs2 "github.com/docker/compose-cli/errdefs"
- "github.com/docker/compose-cli/formatter"
- "github.com/docker/compose-cli/progress"
- )
- type composeService struct {
- apiClient *client.Client
- }
- func (s *composeService) Build(ctx context.Context, project *types.Project) error {
- opts := map[string]build.Options{}
- for _, service := range project.Services {
- if service.Build != nil {
- imageName := service.Image
- if imageName == "" {
- imageName = project.Name + "_" + service.Name
- }
- opts[imageName] = s.toBuildOptions(service, project.WorkingDir, imageName)
- }
- }
- return s.build(ctx, project, opts)
- }
- func (s *composeService) Push(ctx context.Context, project *types.Project) error {
- configFile, err := cliconfig.Load(config.Dir(ctx))
- if err != nil {
- return err
- }
- eg, ctx := errgroup.WithContext(ctx)
- info, err := s.apiClient.Info(ctx)
- if err != nil {
- return err
- }
- if info.IndexServerAddress == "" {
- info.IndexServerAddress = registry.IndexServer
- }
- for _, service := range project.Services {
- if service.Build == nil {
- continue
- }
- service := service
- eg.Go(func() error {
- w := progress.ContextWriter(ctx)
- ref, err := reference.ParseNormalizedNamed(service.Image)
- if err != nil {
- return err
- }
- repoInfo, err := registry.ParseRepositoryInfo(ref)
- if err != nil {
- return err
- }
- key := repoInfo.Index.Name
- if repoInfo.Index.Official {
- key = info.IndexServerAddress
- }
- authConfig, err := configFile.GetAuthConfig(key)
- if err != nil {
- return err
- }
- buf, err := json.Marshal(authConfig)
- if err != nil {
- return err
- }
- stream, err := s.apiClient.ImagePush(ctx, service.Image, moby.ImagePushOptions{
- RegistryAuth: base64.URLEncoding.EncodeToString(buf),
- })
- if err != nil {
- return err
- }
- dec := json.NewDecoder(stream)
- for {
- var jm jsonmessage.JSONMessage
- if err := dec.Decode(&jm); err != nil {
- if err == io.EOF {
- break
- }
- return err
- }
- if jm.Error != nil {
- return errors.New(jm.Error.Message)
- }
- toProgressEvent("Pushing "+service.Name, jm, w)
- }
- return nil
- })
- }
- return eg.Wait()
- }
- func (s *composeService) Pull(ctx context.Context, project *types.Project) error {
- configFile, err := cliconfig.Load(config.Dir(ctx))
- if err != nil {
- return err
- }
- info, err := s.apiClient.Info(ctx)
- if err != nil {
- return err
- }
- if info.IndexServerAddress == "" {
- info.IndexServerAddress = registry.IndexServer
- }
- w := progress.ContextWriter(ctx)
- eg, ctx := errgroup.WithContext(ctx)
- for _, srv := range project.Services {
- service := srv
- eg.Go(func() error {
- w.Event(progress.Event{
- ID: service.Name,
- Status: progress.Working,
- Text: "Pulling",
- })
- ref, err := reference.ParseNormalizedNamed(service.Image)
- if err != nil {
- return err
- }
- repoInfo, err := registry.ParseRepositoryInfo(ref)
- if err != nil {
- return err
- }
- key := repoInfo.Index.Name
- if repoInfo.Index.Official {
- key = info.IndexServerAddress
- }
- authConfig, err := configFile.GetAuthConfig(key)
- if err != nil {
- return err
- }
- buf, err := json.Marshal(authConfig)
- if err != nil {
- return err
- }
- stream, err := s.apiClient.ImagePull(ctx, service.Image, moby.ImagePullOptions{
- RegistryAuth: base64.URLEncoding.EncodeToString(buf),
- })
- if err != nil {
- w.Event(progress.Event{
- ID: service.Name,
- Status: progress.Error,
- Text: "Error",
- })
- return err
- }
- dec := json.NewDecoder(stream)
- for {
- var jm jsonmessage.JSONMessage
- if err := dec.Decode(&jm); err != nil {
- if err == io.EOF {
- break
- }
- return err
- }
- if jm.Error != nil {
- return errors.New(jm.Error.Message)
- }
- toPullProgressEvent(service.Name, jm, w)
- }
- w.Event(progress.Event{
- ID: service.Name,
- Status: progress.Done,
- Text: "Pulled",
- })
- return nil
- })
- }
- return eg.Wait()
- }
- func toPullProgressEvent(parent string, jm jsonmessage.JSONMessage, w progress.Writer) {
- if jm.ID == "" || jm.Progress == nil {
- return
- }
- var (
- text string
- status = progress.Working
- )
- text = jm.Progress.String()
- if jm.Status == "Pull complete" ||
- jm.Status == "Already exists" ||
- strings.Contains(jm.Status, "Image is up to date") ||
- strings.Contains(jm.Status, "Downloaded newer image") {
- status = progress.Done
- }
- if jm.Error != nil {
- status = progress.Error
- text = jm.Error.Message
- }
- w.Event(progress.Event{
- ID: jm.ID,
- ParentID: parent,
- Text: jm.Status,
- Status: status,
- StatusText: text,
- })
- }
- func toProgressEvent(prefix string, jm jsonmessage.JSONMessage, w progress.Writer) {
- if jm.ID == "" {
- // skipped
- return
- }
- var (
- text string
- status = progress.Working
- )
- if jm.Status == "Pull complete" || jm.Status == "Already exists" {
- status = progress.Done
- }
- if jm.Error != nil {
- status = progress.Error
- text = jm.Error.Message
- }
- if jm.Progress != nil {
- text = jm.Progress.String()
- }
- w.Event(progress.Event{
- ID: fmt.Sprintf("Pushing %s: %s", prefix, jm.ID),
- Text: jm.Status,
- Status: status,
- StatusText: text,
- })
- }
- func (s *composeService) Up(ctx context.Context, project *types.Project, detach bool) error {
- return errdefs2.ErrNotImplemented
- }
- func (s *composeService) Create(ctx context.Context, project *types.Project) error {
- err := s.ensureImagesExists(ctx, project)
- if err != nil {
- return err
- }
- for k, network := range project.Networks {
- if !network.External.External && network.Name != "" {
- network.Name = fmt.Sprintf("%s_%s", project.Name, k)
- project.Networks[k] = network
- }
- network.Labels = network.Labels.Add(networkLabel, k)
- network.Labels = network.Labels.Add(projectLabel, project.Name)
- network.Labels = network.Labels.Add(versionLabel, ComposeVersion)
- err := s.ensureNetwork(ctx, network)
- if err != nil {
- return err
- }
- }
- for k, volume := range project.Volumes {
- if !volume.External.External && volume.Name != "" {
- volume.Name = fmt.Sprintf("%s_%s", project.Name, k)
- project.Volumes[k] = volume
- }
- volume.Labels = volume.Labels.Add(volumeLabel, k)
- volume.Labels = volume.Labels.Add(projectLabel, project.Name)
- volume.Labels = volume.Labels.Add(versionLabel, ComposeVersion)
- err := s.ensureVolume(ctx, volume)
- if err != nil {
- return err
- }
- }
- return InDependencyOrder(ctx, project, func(c context.Context, service types.ServiceConfig) error {
- return s.ensureService(c, project, service)
- })
- }
- func (s *composeService) Start(ctx context.Context, project *types.Project, w io.Writer) error {
- var group *errgroup.Group
- if w != nil {
- eg, err := s.attach(ctx, project, w)
- if err != nil {
- return err
- }
- group = eg
- }
- err := InDependencyOrder(ctx, project, func(c context.Context, service types.ServiceConfig) error {
- return s.startService(ctx, project, service)
- })
- if err != nil {
- return err
- }
- if group != nil {
- return group.Wait()
- }
- return nil
- }
- func (s *composeService) attach(ctx context.Context, project *types.Project, w io.Writer) (*errgroup.Group, error) {
- consumer := formatter.NewLogConsumer(ctx, w)
- containers, err := s.apiClient.ContainerList(ctx, moby.ContainerListOptions{
- Filters: filters.NewArgs(
- projectFilter(project.Name),
- ),
- All: true,
- })
- if err != nil {
- return nil, err
- }
- var names []string
- for _, c := range containers {
- names = append(names, getContainerName(c))
- }
- fmt.Printf("Attaching to %s\n", strings.Join(names, ", "))
- eg, ctx := errgroup.WithContext(ctx)
- for _, c := range containers {
- container := c
- eg.Go(func() error {
- return s.attachContainer(ctx, container, consumer, project)
- })
- }
- return eg, nil
- }
- func (s *composeService) attachContainer(ctx context.Context, container moby.Container, consumer formatter.LogConsumer, project *types.Project) error {
- serviceName := container.Labels[serviceLabel]
- w := consumer.GetWriter(serviceName, container.ID)
- service, err := project.GetService(serviceName)
- if err != nil {
- return err
- }
- reader, err := s.getContainerStdout(ctx, container)
- if err != nil {
- return err
- }
- go func() {
- <-ctx.Done()
- reader.Close() //nolint:errcheck
- }()
- if service.Tty {
- _, err = io.Copy(w, reader)
- } else {
- _, err = stdcopy.StdCopy(w, w, reader)
- }
- return err
- }
- func (s *composeService) getContainerStdout(ctx context.Context, container moby.Container) (io.ReadCloser, error) {
- var reader io.ReadCloser
- if container.State == containerRunning {
- logs, err := s.apiClient.ContainerLogs(ctx, container.ID, moby.ContainerLogsOptions{
- ShowStdout: true,
- ShowStderr: true,
- Follow: true,
- })
- if err != nil {
- return nil, err
- }
- reader = logs
- } else {
- cnx, err := s.apiClient.ContainerAttach(ctx, container.ID, moby.ContainerAttachOptions{
- Stream: true,
- Stdin: true,
- Stdout: true,
- Stderr: true,
- })
- if err != nil {
- return nil, err
- }
- reader = containerStdout{cnx}
- err = s.apiClient.ContainerStart(ctx, container.ID, moby.ContainerStartOptions{})
- if err != nil {
- return nil, err
- }
- }
- return reader, nil
- }
- func getContainerName(c moby.Container) string {
- // Names return container canonical name /foo + link aliases /linked_by/foo
- for _, name := range c.Names {
- if strings.LastIndex(name, "/") == 0 {
- return name[1:]
- }
- }
- return c.Names[0][1:]
- }
- func (s *composeService) Down(ctx context.Context, projectName string) error {
- eg, _ := errgroup.WithContext(ctx)
- w := progress.ContextWriter(ctx)
- project, err := s.projectFromContainerLabels(ctx, projectName)
- if err != nil || project == nil {
- return err
- }
- err = InReverseDependencyOrder(ctx, project, func(c context.Context, service types.ServiceConfig) error {
- filter := filters.NewArgs(projectFilter(project.Name), serviceFilter(service.Name))
- return s.removeContainers(ctx, w, eg, filter)
- })
- if err != nil {
- return err
- }
- err = eg.Wait()
- if err != nil {
- return err
- }
- networks, err := s.apiClient.NetworkList(ctx, moby.NetworkListOptions{
- Filters: filters.NewArgs(
- projectFilter(projectName),
- ),
- })
- if err != nil {
- return err
- }
- for _, n := range networks {
- networkID := n.ID
- networkName := n.Name
- eg.Go(func() error {
- return s.ensureNetworkDown(ctx, networkID, networkName)
- })
- }
- return eg.Wait()
- }
- func (s *composeService) removeContainers(ctx context.Context, w progress.Writer, eg *errgroup.Group, filter filters.Args) error {
- containers, err := s.apiClient.ContainerList(ctx, moby.ContainerListOptions{
- Filters: filter,
- All: true,
- })
- if err != nil {
- return err
- }
- for _, container := range containers {
- eg.Go(func() error {
- eventName := "Container " + getContainerName(container)
- w.Event(progress.StoppingEvent(eventName))
- err := s.apiClient.ContainerStop(ctx, container.ID, nil)
- if err != nil {
- w.Event(progress.ErrorMessageEvent(eventName, "Error while Stopping"))
- return err
- }
- w.Event(progress.RemovingEvent(eventName))
- err = s.apiClient.ContainerRemove(ctx, container.ID, moby.ContainerRemoveOptions{})
- if err != nil {
- w.Event(progress.ErrorMessageEvent(eventName, "Error while Removing"))
- return err
- }
- w.Event(progress.RemovedEvent(eventName))
- return nil
- })
- }
- return nil
- }
- func (s *composeService) projectFromContainerLabels(ctx context.Context, projectName string) (*types.Project, error) {
- containers, err := s.apiClient.ContainerList(ctx, moby.ContainerListOptions{
- Filters: filters.NewArgs(
- projectFilter(projectName),
- ),
- All: true,
- })
- if err != nil {
- return nil, err
- }
- if len(containers) == 0 {
- return nil, nil
- }
- options, err := loadProjectOptionsFromLabels(containers[0])
- if err != nil {
- return nil, err
- }
- if options.ConfigPaths[0] == "-" {
- fakeProject := &types.Project{
- Name: projectName,
- }
- for _, container := range containers {
- fakeProject.Services = append(fakeProject.Services, types.ServiceConfig{
- Name: container.Labels[serviceLabel],
- })
- }
- return fakeProject, nil
- }
- project, err := cli.ProjectFromOptions(options)
- if err != nil {
- return nil, err
- }
- return project, nil
- }
- func loadProjectOptionsFromLabels(c moby.Container) (*cli.ProjectOptions, error) {
- var configFiles []string
- relativePathConfigFiles := strings.Split(c.Labels[configFilesLabel], ",")
- for _, c := range relativePathConfigFiles {
- configFiles = append(configFiles, filepath.Base(c))
- }
- return cli.NewProjectOptions(configFiles,
- cli.WithOsEnv,
- cli.WithWorkingDirectory(c.Labels[workingDirLabel]),
- cli.WithName(c.Labels[projectLabel]))
- }
- func (s *composeService) Logs(ctx context.Context, projectName string, w io.Writer) error {
- list, err := s.apiClient.ContainerList(ctx, moby.ContainerListOptions{
- Filters: filters.NewArgs(
- projectFilter(projectName),
- ),
- })
- if err != nil {
- return err
- }
- consumer := formatter.NewLogConsumer(ctx, w)
- eg, ctx := errgroup.WithContext(ctx)
- for _, c := range list {
- service := c.Labels[serviceLabel]
- container, err := s.apiClient.ContainerInspect(ctx, c.ID)
- if err != nil {
- return err
- }
- eg.Go(func() error {
- r, err := s.apiClient.ContainerLogs(ctx, container.ID, moby.ContainerLogsOptions{
- ShowStdout: true,
- ShowStderr: true,
- Follow: true,
- })
- defer r.Close() // nolint errcheck
- if err != nil {
- return err
- }
- w := consumer.GetWriter(service, container.ID)
- if container.Config.Tty {
- _, err = io.Copy(w, r)
- } else {
- _, err = stdcopy.StdCopy(w, w, r)
- }
- return err
- })
- }
- return eg.Wait()
- }
- func (s *composeService) Ps(ctx context.Context, projectName string) ([]compose.ServiceStatus, error) {
- list, err := s.apiClient.ContainerList(ctx, moby.ContainerListOptions{
- Filters: filters.NewArgs(
- projectFilter(projectName),
- ),
- })
- if err != nil {
- return nil, err
- }
- return containersToServiceStatus(list)
- }
- func containersToServiceStatus(containers []moby.Container) ([]compose.ServiceStatus, error) {
- containersByLabel, keys, err := groupContainerByLabel(containers, serviceLabel)
- if err != nil {
- return nil, err
- }
- var services []compose.ServiceStatus
- for _, service := range keys {
- containers := containersByLabel[service]
- runnningContainers := []moby.Container{}
- for _, container := range containers {
- if container.State == containerRunning {
- runnningContainers = append(runnningContainers, container)
- }
- }
- services = append(services, compose.ServiceStatus{
- ID: service,
- Name: service,
- Desired: len(containers),
- Replicas: len(runnningContainers),
- })
- }
- return services, nil
- }
- func groupContainerByLabel(containers []moby.Container, labelName string) (map[string][]moby.Container, []string, error) {
- containersByLabel := map[string][]moby.Container{}
- keys := []string{}
- for _, c := range containers {
- label, ok := c.Labels[labelName]
- if !ok {
- return nil, nil, fmt.Errorf("No label %q set on container %q of compose project", labelName, c.ID)
- }
- labelContainers, ok := containersByLabel[label]
- if !ok {
- labelContainers = []moby.Container{}
- keys = append(keys, label)
- }
- labelContainers = append(labelContainers, c)
- containersByLabel[label] = labelContainers
- }
- sort.Strings(keys)
- return containersByLabel, keys, nil
- }
- func (s *composeService) List(ctx context.Context, projectName string) ([]compose.Stack, error) {
- list, err := s.apiClient.ContainerList(ctx, moby.ContainerListOptions{
- Filters: filters.NewArgs(hasProjectLabelFilter()),
- })
- if err != nil {
- return nil, err
- }
- return containersToStacks(list)
- }
- func containersToStacks(containers []moby.Container) ([]compose.Stack, error) {
- containersByLabel, keys, err := groupContainerByLabel(containers, projectLabel)
- if err != nil {
- return nil, err
- }
- var projects []compose.Stack
- for _, project := range keys {
- projects = append(projects, compose.Stack{
- ID: project,
- Name: project,
- Status: combinedStatus(containerToState(containersByLabel[project])),
- })
- }
- return projects, nil
- }
- func containerToState(containers []moby.Container) []string {
- statuses := []string{}
- for _, c := range containers {
- statuses = append(statuses, c.State)
- }
- return statuses
- }
- func combinedStatus(statuses []string) string {
- nbByStatus := map[string]int{}
- keys := []string{}
- for _, status := range statuses {
- nb, ok := nbByStatus[status]
- if !ok {
- nb = 0
- keys = append(keys, status)
- }
- nbByStatus[status] = nb + 1
- }
- sort.Strings(keys)
- result := ""
- for _, status := range keys {
- nb := nbByStatus[status]
- if result != "" {
- result = result + ", "
- }
- result = result + fmt.Sprintf("%s(%d)", status, nb)
- }
- return result
- }
- func (s *composeService) Convert(ctx context.Context, project *types.Project, format string) ([]byte, error) {
- switch format {
- case "json":
- return json.MarshalIndent(project, "", " ")
- case "yaml":
- return yaml.Marshal(project)
- default:
- return nil, fmt.Errorf("unsupported format %q", format)
- }
- }
- func getContainerCreateOptions(p *types.Project, s types.ServiceConfig, number int, inherit *moby.Container) (*container.Config, *container.HostConfig, *network.NetworkingConfig, error) {
- hash, err := jsonHash(s)
- if err != nil {
- return nil, nil, nil, err
- }
- // TODO: change oneoffLabel value for containers started with `docker compose run`
- labels := map[string]string{
- projectLabel: p.Name,
- serviceLabel: s.Name,
- versionLabel: ComposeVersion,
- oneoffLabel: "False",
- configHashLabel: hash,
- workingDirLabel: p.WorkingDir,
- configFilesLabel: strings.Join(p.ComposeFiles, ","),
- containerNumberLabel: strconv.Itoa(number),
- }
- var (
- runCmd strslice.StrSlice
- entrypoint strslice.StrSlice
- )
- if len(s.Command) > 0 {
- runCmd = strslice.StrSlice(s.Command)
- }
- if len(s.Entrypoint) > 0 {
- entrypoint = strslice.StrSlice(s.Entrypoint)
- }
- image := s.Image
- if s.Image == "" {
- image = fmt.Sprintf("%s_%s", p.Name, s.Name)
- }
- var (
- tty = s.Tty
- stdinOpen = s.StdinOpen
- attachStdin = false
- )
- containerConfig := container.Config{
- Hostname: s.Hostname,
- Domainname: s.DomainName,
- User: s.User,
- ExposedPorts: buildContainerPorts(s),
- Tty: tty,
- OpenStdin: stdinOpen,
- StdinOnce: true,
- AttachStdin: attachStdin,
- AttachStderr: true,
- AttachStdout: true,
- Cmd: runCmd,
- Image: image,
- WorkingDir: s.WorkingDir,
- Entrypoint: entrypoint,
- NetworkDisabled: s.NetworkMode == "disabled",
- MacAddress: s.MacAddress,
- Labels: labels,
- StopSignal: s.StopSignal,
- Env: toMobyEnv(s.Environment),
- Healthcheck: toMobyHealthCheck(s.HealthCheck),
- // Volumes: // FIXME unclear to me the overlap with HostConfig.Mounts
- StopTimeout: toSeconds(s.StopGracePeriod),
- }
- mountOptions, err := buildContainerMountOptions(s, inherit)
- if err != nil {
- return nil, nil, nil, err
- }
- bindings := buildContainerBindingOptions(s)
- networkMode := getNetworkMode(p, s)
- hostConfig := container.HostConfig{
- Mounts: mountOptions,
- CapAdd: strslice.StrSlice(s.CapAdd),
- CapDrop: strslice.StrSlice(s.CapDrop),
- NetworkMode: networkMode,
- Init: s.Init,
- ReadonlyRootfs: s.ReadOnly,
- // ShmSize: , TODO
- Sysctls: s.Sysctls,
- PortBindings: bindings,
- }
- networkConfig := buildDefaultNetworkConfig(s, networkMode)
- return &containerConfig, &hostConfig, networkConfig, nil
- }
- func buildContainerPorts(s types.ServiceConfig) nat.PortSet {
- ports := nat.PortSet{}
- for _, p := range s.Ports {
- p := nat.Port(fmt.Sprintf("%d/%s", p.Target, p.Protocol))
- ports[p] = struct{}{}
- }
- return ports
- }
- func buildContainerBindingOptions(s types.ServiceConfig) nat.PortMap {
- bindings := nat.PortMap{}
- for _, port := range s.Ports {
- p := nat.Port(fmt.Sprintf("%d/%s", port.Target, port.Protocol))
- bind := []nat.PortBinding{}
- binding := nat.PortBinding{}
- if port.Published > 0 {
- binding.HostPort = fmt.Sprint(port.Published)
- }
- bind = append(bind, binding)
- bindings[p] = bind
- }
- return bindings
- }
- func buildContainerMountOptions(s types.ServiceConfig, inherit *moby.Container) ([]mount.Mount, error) {
- mounts := []mount.Mount{}
- var inherited []string
- if inherit != nil {
- for _, m := range inherit.Mounts {
- if m.Type == "tmpfs" {
- continue
- }
- src := m.Source
- if m.Type == "volume" {
- src = m.Name
- }
- mounts = append(mounts, mount.Mount{
- Type: m.Type,
- Source: src,
- Target: m.Destination,
- ReadOnly: !m.RW,
- })
- inherited = append(inherited, m.Destination)
- }
- }
- for _, v := range s.Volumes {
- if contains(inherited, v.Target) {
- continue
- }
- mount, err := buildMount(v)
- if err != nil {
- return nil, err
- }
- mounts = append(mounts, mount)
- }
- return mounts, nil
- }
- func buildMount(volume types.ServiceVolumeConfig) (mount.Mount, error) {
- source := volume.Source
- if volume.Type == "bind" && !filepath.IsAbs(source) {
- // volume source has already been prefixed with workdir if required, by compose-go project loader
- var err error
- source, err = filepath.Abs(source)
- if err != nil {
- return mount.Mount{}, err
- }
- }
- return mount.Mount{
- Type: mount.Type(volume.Type),
- Source: source,
- Target: volume.Target,
- ReadOnly: volume.ReadOnly,
- Consistency: mount.Consistency(volume.Consistency),
- BindOptions: buildBindOption(volume.Bind),
- VolumeOptions: buildVolumeOptions(volume.Volume),
- TmpfsOptions: buildTmpfsOptions(volume.Tmpfs),
- }, nil
- }
- func buildBindOption(bind *types.ServiceVolumeBind) *mount.BindOptions {
- if bind == nil {
- return nil
- }
- return &mount.BindOptions{
- Propagation: mount.Propagation(bind.Propagation),
- // NonRecursive: false, FIXME missing from model ?
- }
- }
- func buildVolumeOptions(vol *types.ServiceVolumeVolume) *mount.VolumeOptions {
- if vol == nil {
- return nil
- }
- return &mount.VolumeOptions{
- NoCopy: vol.NoCopy,
- // Labels: , // FIXME missing from model ?
- // DriverConfig: , // FIXME missing from model ?
- }
- }
- func buildTmpfsOptions(tmpfs *types.ServiceVolumeTmpfs) *mount.TmpfsOptions {
- if tmpfs == nil {
- return nil
- }
- return &mount.TmpfsOptions{
- SizeBytes: tmpfs.Size,
- // Mode: , // FIXME missing from model ?
- }
- }
- func buildDefaultNetworkConfig(s types.ServiceConfig, networkMode container.NetworkMode) *network.NetworkingConfig {
- config := map[string]*network.EndpointSettings{}
- net := string(networkMode)
- config[net] = &network.EndpointSettings{
- Aliases: getAliases(s, s.Networks[net]),
- }
- return &network.NetworkingConfig{
- EndpointsConfig: config,
- }
- }
- func getAliases(s types.ServiceConfig, c *types.ServiceNetworkConfig) []string {
- aliases := []string{s.Name}
- if c != nil {
- aliases = append(aliases, c.Aliases...)
- }
- return aliases
- }
- func getNetworkMode(p *types.Project, service types.ServiceConfig) container.NetworkMode {
- mode := service.NetworkMode
- if mode == "" {
- if len(p.Networks) > 0 {
- for name := range getNetworksForService(service) {
- return container.NetworkMode(p.Networks[name].Name)
- }
- }
- return container.NetworkMode("none")
- }
- // FIXME incomplete implementation
- if strings.HasPrefix(mode, "service:") {
- panic("Not yet implemented")
- }
- if strings.HasPrefix(mode, "container:") {
- panic("Not yet implemented")
- }
- return container.NetworkMode(mode)
- }
- func getNetworksForService(s types.ServiceConfig) map[string]*types.ServiceNetworkConfig {
- if len(s.Networks) > 0 {
- return s.Networks
- }
- return map[string]*types.ServiceNetworkConfig{"default": nil}
- }
- func (s *composeService) ensureNetwork(ctx context.Context, n types.NetworkConfig) error {
- _, err := s.apiClient.NetworkInspect(ctx, n.Name, moby.NetworkInspectOptions{})
- if err != nil {
- if errdefs.IsNotFound(err) {
- if n.External.External {
- return fmt.Errorf("network %s declared as external, but could not be found", n.Name)
- }
- createOpts := moby.NetworkCreate{
- // TODO NameSpace Labels
- Labels: n.Labels,
- Driver: n.Driver,
- Options: n.DriverOpts,
- Internal: n.Internal,
- Attachable: n.Attachable,
- }
- if n.Ipam.Driver != "" || len(n.Ipam.Config) > 0 {
- createOpts.IPAM = &network.IPAM{}
- }
- if n.Ipam.Driver != "" {
- createOpts.IPAM.Driver = n.Ipam.Driver
- }
- for _, ipamConfig := range n.Ipam.Config {
- config := network.IPAMConfig{
- Subnet: ipamConfig.Subnet,
- }
- createOpts.IPAM.Config = append(createOpts.IPAM.Config, config)
- }
- networkEventName := fmt.Sprintf("Network %q", n.Name)
- w := progress.ContextWriter(ctx)
- w.Event(progress.CreatingEvent(networkEventName))
- if _, err := s.apiClient.NetworkCreate(ctx, n.Name, createOpts); err != nil {
- w.Event(progress.ErrorEvent(networkEventName))
- return errors.Wrapf(err, "failed to create network %s", n.Name)
- }
- w.Event(progress.CreatedEvent(networkEventName))
- return nil
- }
- return err
- }
- return nil
- }
- func (s *composeService) ensureNetworkDown(ctx context.Context, networkID string, networkName string) error {
- w := progress.ContextWriter(ctx)
- eventName := fmt.Sprintf("Network %q", networkName)
- w.Event(progress.RemovingEvent(eventName))
- if err := s.apiClient.NetworkRemove(ctx, networkID); err != nil {
- w.Event(progress.ErrorEvent(eventName))
- return errors.Wrapf(err, fmt.Sprintf("failed to create network %s", networkID))
- }
- w.Event(progress.RemovedEvent(eventName))
- return nil
- }
- func (s *composeService) ensureVolume(ctx context.Context, volume types.VolumeConfig) error {
- // TODO could identify volume by label vs name
- _, err := s.apiClient.VolumeInspect(ctx, volume.Name)
- if err != nil {
- if errdefs.IsNotFound(err) {
- eventName := fmt.Sprintf("Volume %q", volume.Name)
- w := progress.ContextWriter(ctx)
- w.Event(progress.CreatingEvent(eventName))
- // TODO we miss support for driver_opts and labels
- _, err := s.apiClient.VolumeCreate(ctx, mobyvolume.VolumeCreateBody{
- Labels: volume.Labels,
- Name: volume.Name,
- Driver: volume.Driver,
- DriverOpts: volume.DriverOpts,
- })
- if err != nil {
- w.Event(progress.ErrorEvent(eventName))
- return err
- }
- w.Event(progress.CreatedEvent(eventName))
- }
- return err
- }
- return nil
- }
|