convergence.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927
  1. /*
  2. Copyright 2020 Docker Compose CLI authors
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. */
  13. package compose
  14. import (
  15. "context"
  16. "errors"
  17. "fmt"
  18. "slices"
  19. "sort"
  20. "strconv"
  21. "strings"
  22. "sync"
  23. "time"
  24. "github.com/compose-spec/compose-go/v2/types"
  25. "github.com/containerd/platforms"
  26. containerType "github.com/docker/docker/api/types/container"
  27. mmount "github.com/docker/docker/api/types/mount"
  28. "github.com/docker/docker/api/types/versions"
  29. specs "github.com/opencontainers/image-spec/specs-go/v1"
  30. "github.com/sirupsen/logrus"
  31. "go.opentelemetry.io/otel/attribute"
  32. "go.opentelemetry.io/otel/trace"
  33. "golang.org/x/sync/errgroup"
  34. "github.com/docker/compose/v2/internal/tracing"
  35. "github.com/docker/compose/v2/pkg/api"
  36. "github.com/docker/compose/v2/pkg/progress"
  37. "github.com/docker/compose/v2/pkg/utils"
  38. )
  39. const (
  40. doubledContainerNameWarning = "WARNING: The %q service is using the custom container name %q. " +
  41. "Docker requires each container to have a unique name. " +
  42. "Remove the custom name to scale the service"
  43. )
  44. // convergence manages service's container lifecycle.
  45. // Based on initially observed state, it reconciles the existing container with desired state, which might include
  46. // re-creating container, adding or removing replicas, or starting stopped containers.
  47. // Cross services dependencies are managed by creating services in expected order and updating `service:xx` reference
  48. // when a service has converged, so dependent ones can be managed with resolved containers references.
  49. type convergence struct {
  50. service *composeService
  51. services map[string]Containers
  52. networks map[string]string
  53. volumes map[string]string
  54. stateMutex sync.Mutex
  55. }
  56. func (c *convergence) getObservedState(serviceName string) Containers {
  57. c.stateMutex.Lock()
  58. defer c.stateMutex.Unlock()
  59. return c.services[serviceName]
  60. }
  61. func (c *convergence) setObservedState(serviceName string, containers Containers) {
  62. c.stateMutex.Lock()
  63. defer c.stateMutex.Unlock()
  64. c.services[serviceName] = containers
  65. }
  66. func newConvergence(services []string, state Containers, networks map[string]string, volumes map[string]string, s *composeService) *convergence {
  67. observedState := map[string]Containers{}
  68. for _, s := range services {
  69. observedState[s] = Containers{}
  70. }
  71. for _, c := range state.filter(isNotOneOff) {
  72. service := c.Labels[api.ServiceLabel]
  73. observedState[service] = append(observedState[service], c)
  74. }
  75. return &convergence{
  76. service: s,
  77. services: observedState,
  78. networks: networks,
  79. volumes: volumes,
  80. }
  81. }
  82. func (c *convergence) apply(ctx context.Context, project *types.Project, options api.CreateOptions) error {
  83. return InDependencyOrder(ctx, project, func(ctx context.Context, name string) error {
  84. service, err := project.GetService(name)
  85. if err != nil {
  86. return err
  87. }
  88. return tracing.SpanWrapFunc("service/apply", tracing.ServiceOptions(service), func(ctx context.Context) error {
  89. strategy := options.RecreateDependencies
  90. if slices.Contains(options.Services, name) {
  91. strategy = options.Recreate
  92. }
  93. return c.ensureService(ctx, project, service, strategy, options.Inherit, options.Timeout)
  94. })(ctx)
  95. })
  96. }
  97. func (c *convergence) ensureService(ctx context.Context, project *types.Project, service types.ServiceConfig, recreate string, inherit bool, timeout *time.Duration) error { //nolint:gocyclo
  98. if service.Provider != nil {
  99. return c.service.runPlugin(ctx, project, service, "up")
  100. }
  101. expected, err := getScale(service)
  102. if err != nil {
  103. return err
  104. }
  105. containers := c.getObservedState(service.Name)
  106. actual := len(containers)
  107. updated := make(Containers, expected)
  108. eg, _ := errgroup.WithContext(ctx)
  109. err = c.resolveServiceReferences(&service)
  110. if err != nil {
  111. return err
  112. }
  113. sort.Slice(containers, func(i, j int) bool {
  114. // select obsolete containers first, so they get removed as we scale down
  115. if obsolete, _ := c.mustRecreate(service, containers[i], recreate); obsolete {
  116. // i is obsolete, so must be first in the list
  117. return true
  118. }
  119. if obsolete, _ := c.mustRecreate(service, containers[j], recreate); obsolete {
  120. // j is obsolete, so must be first in the list
  121. return false
  122. }
  123. // For up-to-date containers, sort by container number to preserve low-values in container numbers
  124. ni, erri := strconv.Atoi(containers[i].Labels[api.ContainerNumberLabel])
  125. nj, errj := strconv.Atoi(containers[j].Labels[api.ContainerNumberLabel])
  126. if erri == nil && errj == nil {
  127. return ni > nj
  128. }
  129. // If we don't get a container number (?) just sort by creation date
  130. return containers[i].Created < containers[j].Created
  131. })
  132. slices.Reverse(containers)
  133. for i, container := range containers {
  134. if i >= expected {
  135. // Scale Down
  136. // As we sorted containers, obsolete ones and/or highest number will be removed
  137. container := container
  138. traceOpts := append(tracing.ServiceOptions(service), tracing.ContainerOptions(container)...)
  139. eg.Go(tracing.SpanWrapFuncForErrGroup(ctx, "service/scale/down", traceOpts, func(ctx context.Context) error {
  140. return c.service.stopAndRemoveContainer(ctx, container, &service, timeout, false)
  141. }))
  142. continue
  143. }
  144. mustRecreate, err := c.mustRecreate(service, container, recreate)
  145. if err != nil {
  146. return err
  147. }
  148. if mustRecreate {
  149. err := c.stopDependentContainers(ctx, project, service)
  150. if err != nil {
  151. return err
  152. }
  153. i, container := i, container
  154. eg.Go(tracing.SpanWrapFuncForErrGroup(ctx, "container/recreate", tracing.ContainerOptions(container), func(ctx context.Context) error {
  155. recreated, err := c.service.recreateContainer(ctx, project, service, container, inherit, timeout)
  156. updated[i] = recreated
  157. return err
  158. }))
  159. continue
  160. }
  161. // Enforce non-diverged containers are running
  162. w := progress.ContextWriter(ctx)
  163. name := getContainerProgressName(container)
  164. switch container.State {
  165. case ContainerRunning:
  166. w.Event(progress.RunningEvent(name))
  167. case ContainerCreated:
  168. case ContainerRestarting:
  169. case ContainerExited:
  170. default:
  171. container := container
  172. eg.Go(tracing.EventWrapFuncForErrGroup(ctx, "service/start", tracing.ContainerOptions(container), func(ctx context.Context) error {
  173. return c.service.startContainer(ctx, container)
  174. }))
  175. }
  176. updated[i] = container
  177. }
  178. next := nextContainerNumber(containers)
  179. for i := 0; i < expected-actual; i++ {
  180. // Scale UP
  181. number := next + i
  182. name := getContainerName(project.Name, service, number)
  183. eventOpts := tracing.SpanOptions{trace.WithAttributes(attribute.String("container.name", name))}
  184. eg.Go(tracing.EventWrapFuncForErrGroup(ctx, "service/scale/up", eventOpts, func(ctx context.Context) error {
  185. opts := createOptions{
  186. AutoRemove: false,
  187. AttachStdin: false,
  188. UseNetworkAliases: true,
  189. Labels: mergeLabels(service.Labels, service.CustomLabels),
  190. }
  191. container, err := c.service.createContainer(ctx, project, service, name, number, opts)
  192. updated[actual+i] = container
  193. return err
  194. }))
  195. continue
  196. }
  197. err = eg.Wait()
  198. c.setObservedState(service.Name, updated)
  199. return err
  200. }
  201. func (c *convergence) stopDependentContainers(ctx context.Context, project *types.Project, service types.ServiceConfig) error {
  202. // Stop dependent containers, so they will be restarted after service is re-created
  203. dependents := project.GetDependentsForService(service, func(dependency types.ServiceDependency) bool {
  204. return dependency.Restart
  205. })
  206. if len(dependents) == 0 {
  207. return nil
  208. }
  209. err := c.service.stop(ctx, project.Name, api.StopOptions{
  210. Services: dependents,
  211. Project: project,
  212. })
  213. if err != nil {
  214. return err
  215. }
  216. for _, name := range dependents {
  217. dependentStates := c.getObservedState(name)
  218. for i, dependent := range dependentStates {
  219. dependent.State = ContainerExited
  220. dependentStates[i] = dependent
  221. }
  222. c.setObservedState(name, dependentStates)
  223. }
  224. return nil
  225. }
  226. func getScale(config types.ServiceConfig) (int, error) {
  227. scale := config.GetScale()
  228. if scale > 1 && config.ContainerName != "" {
  229. return 0, fmt.Errorf(doubledContainerNameWarning,
  230. config.Name,
  231. config.ContainerName)
  232. }
  233. return scale, nil
  234. }
  235. // resolveServiceReferences replaces reference to another service with reference to an actual container
  236. func (c *convergence) resolveServiceReferences(service *types.ServiceConfig) error {
  237. err := c.resolveVolumeFrom(service)
  238. if err != nil {
  239. return err
  240. }
  241. err = c.resolveSharedNamespaces(service)
  242. if err != nil {
  243. return err
  244. }
  245. return nil
  246. }
  247. func (c *convergence) resolveVolumeFrom(service *types.ServiceConfig) error {
  248. for i, vol := range service.VolumesFrom {
  249. spec := strings.Split(vol, ":")
  250. if len(spec) == 0 {
  251. continue
  252. }
  253. if spec[0] == "container" {
  254. service.VolumesFrom[i] = spec[1]
  255. continue
  256. }
  257. name := spec[0]
  258. dependencies := c.getObservedState(name)
  259. if len(dependencies) == 0 {
  260. return fmt.Errorf("cannot share volume with service %s: container missing", name)
  261. }
  262. service.VolumesFrom[i] = dependencies.sorted()[0].ID
  263. }
  264. return nil
  265. }
  266. func (c *convergence) resolveSharedNamespaces(service *types.ServiceConfig) error {
  267. str := service.NetworkMode
  268. if name := getDependentServiceFromMode(str); name != "" {
  269. dependencies := c.getObservedState(name)
  270. if len(dependencies) == 0 {
  271. return fmt.Errorf("cannot share network namespace with service %s: container missing", name)
  272. }
  273. service.NetworkMode = types.ContainerPrefix + dependencies.sorted()[0].ID
  274. }
  275. str = service.Ipc
  276. if name := getDependentServiceFromMode(str); name != "" {
  277. dependencies := c.getObservedState(name)
  278. if len(dependencies) == 0 {
  279. return fmt.Errorf("cannot share IPC namespace with service %s: container missing", name)
  280. }
  281. service.Ipc = types.ContainerPrefix + dependencies.sorted()[0].ID
  282. }
  283. str = service.Pid
  284. if name := getDependentServiceFromMode(str); name != "" {
  285. dependencies := c.getObservedState(name)
  286. if len(dependencies) == 0 {
  287. return fmt.Errorf("cannot share PID namespace with service %s: container missing", name)
  288. }
  289. service.Pid = types.ContainerPrefix + dependencies.sorted()[0].ID
  290. }
  291. return nil
  292. }
  293. func (c *convergence) mustRecreate(expected types.ServiceConfig, actual containerType.Summary, policy string) (bool, error) {
  294. if policy == api.RecreateNever {
  295. return false, nil
  296. }
  297. if policy == api.RecreateForce {
  298. return true, nil
  299. }
  300. configHash, err := ServiceHash(expected)
  301. if err != nil {
  302. return false, err
  303. }
  304. configChanged := actual.Labels[api.ConfigHashLabel] != configHash
  305. imageUpdated := actual.Labels[api.ImageDigestLabel] != expected.CustomLabels[api.ImageDigestLabel]
  306. if configChanged || imageUpdated {
  307. return true, nil
  308. }
  309. if c.networks != nil && actual.State == "running" {
  310. if checkExpectedNetworks(expected, actual, c.networks) {
  311. return true, nil
  312. }
  313. }
  314. if c.volumes != nil {
  315. if checkExpectedVolumes(expected, actual, c.volumes) {
  316. return true, nil
  317. }
  318. }
  319. return false, nil
  320. }
  321. func checkExpectedNetworks(expected types.ServiceConfig, actual containerType.Summary, networks map[string]string) bool {
  322. // check the networks container is connected to are the expected ones
  323. for net := range expected.Networks {
  324. id := networks[net]
  325. if id == "swarm" {
  326. // corner-case : swarm overlay network isn't visible until a container is attached
  327. continue
  328. }
  329. found := false
  330. for _, settings := range actual.NetworkSettings.Networks {
  331. if settings.NetworkID == id {
  332. found = true
  333. break
  334. }
  335. }
  336. if !found {
  337. // config is up-to-date but container is not connected to network
  338. return true
  339. }
  340. }
  341. return false
  342. }
  343. func checkExpectedVolumes(expected types.ServiceConfig, actual containerType.Summary, volumes map[string]string) bool {
  344. // check container's volume mounts and search for the expected ones
  345. for _, vol := range expected.Volumes {
  346. if vol.Type != string(mmount.TypeVolume) {
  347. continue
  348. }
  349. if vol.Source == "" {
  350. continue
  351. }
  352. id := volumes[vol.Source]
  353. found := false
  354. for _, mount := range actual.Mounts {
  355. if mount.Type != mmount.TypeVolume {
  356. continue
  357. }
  358. if mount.Name == id {
  359. found = true
  360. break
  361. }
  362. }
  363. if !found {
  364. // config is up-to-date but container doesn't have volume mounted
  365. return true
  366. }
  367. }
  368. return false
  369. }
  370. func getContainerName(projectName string, service types.ServiceConfig, number int) string {
  371. name := getDefaultContainerName(projectName, service.Name, strconv.Itoa(number))
  372. if service.ContainerName != "" {
  373. name = service.ContainerName
  374. }
  375. return name
  376. }
  377. func getDefaultContainerName(projectName, serviceName, index string) string {
  378. return strings.Join([]string{projectName, serviceName, index}, api.Separator)
  379. }
  380. func getContainerProgressName(ctr containerType.Summary) string {
  381. return "Container " + getCanonicalContainerName(ctr)
  382. }
  383. func containerEvents(containers Containers, eventFunc func(string) progress.Event) []progress.Event {
  384. events := []progress.Event{}
  385. for _, ctr := range containers {
  386. events = append(events, eventFunc(getContainerProgressName(ctr)))
  387. }
  388. return events
  389. }
  390. func containerReasonEvents(containers Containers, eventFunc func(string, string) progress.Event, reason string) []progress.Event {
  391. events := []progress.Event{}
  392. for _, ctr := range containers {
  393. events = append(events, eventFunc(getContainerProgressName(ctr), reason))
  394. }
  395. return events
  396. }
  397. // ServiceConditionRunningOrHealthy is a service condition on status running or healthy
  398. const ServiceConditionRunningOrHealthy = "running_or_healthy"
  399. //nolint:gocyclo
  400. func (s *composeService) waitDependencies(ctx context.Context, project *types.Project, dependant string, dependencies types.DependsOnConfig, containers Containers, timeout time.Duration) error {
  401. if timeout > 0 {
  402. withTimeout, cancelFunc := context.WithTimeout(ctx, timeout)
  403. defer cancelFunc()
  404. ctx = withTimeout
  405. }
  406. eg, _ := errgroup.WithContext(ctx)
  407. w := progress.ContextWriter(ctx)
  408. for dep, config := range dependencies {
  409. if shouldWait, err := shouldWaitForDependency(dep, config, project); err != nil {
  410. return err
  411. } else if !shouldWait {
  412. continue
  413. }
  414. waitingFor := containers.filter(isService(dep), isNotOneOff)
  415. w.Events(containerEvents(waitingFor, progress.Waiting))
  416. if len(waitingFor) == 0 {
  417. if config.Required {
  418. return fmt.Errorf("%s is missing dependency %s", dependant, dep)
  419. }
  420. logrus.Warnf("%s is missing dependency %s", dependant, dep)
  421. continue
  422. }
  423. eg.Go(func() error {
  424. ticker := time.NewTicker(500 * time.Millisecond)
  425. defer ticker.Stop()
  426. for {
  427. select {
  428. case <-ticker.C:
  429. case <-ctx.Done():
  430. return nil
  431. }
  432. switch config.Condition {
  433. case ServiceConditionRunningOrHealthy:
  434. healthy, err := s.isServiceHealthy(ctx, waitingFor, true)
  435. if err != nil {
  436. if !config.Required {
  437. w.Events(containerReasonEvents(waitingFor, progress.SkippedEvent, fmt.Sprintf("optional dependency %q is not running or is unhealthy", dep)))
  438. logrus.Warnf("optional dependency %q is not running or is unhealthy: %s", dep, err.Error())
  439. return nil
  440. }
  441. return err
  442. }
  443. if healthy {
  444. w.Events(containerEvents(waitingFor, progress.Healthy))
  445. return nil
  446. }
  447. case types.ServiceConditionHealthy:
  448. healthy, err := s.isServiceHealthy(ctx, waitingFor, false)
  449. if err != nil {
  450. if !config.Required {
  451. w.Events(containerReasonEvents(waitingFor, progress.SkippedEvent, fmt.Sprintf("optional dependency %q failed to start", dep)))
  452. logrus.Warnf("optional dependency %q failed to start: %s", dep, err.Error())
  453. return nil
  454. }
  455. w.Events(containerEvents(waitingFor, progress.ErrorEvent))
  456. return fmt.Errorf("dependency failed to start: %w", err)
  457. }
  458. if healthy {
  459. w.Events(containerEvents(waitingFor, progress.Healthy))
  460. return nil
  461. }
  462. case types.ServiceConditionCompletedSuccessfully:
  463. exited, code, err := s.isServiceCompleted(ctx, waitingFor)
  464. if err != nil {
  465. return err
  466. }
  467. if exited {
  468. if code == 0 {
  469. w.Events(containerEvents(waitingFor, progress.Exited))
  470. return nil
  471. }
  472. messageSuffix := fmt.Sprintf("%q didn't complete successfully: exit %d", dep, code)
  473. if !config.Required {
  474. // optional -> mark as skipped & don't propagate error
  475. w.Events(containerReasonEvents(waitingFor, progress.SkippedEvent, fmt.Sprintf("optional dependency %s", messageSuffix)))
  476. logrus.Warnf("optional dependency %s", messageSuffix)
  477. return nil
  478. }
  479. msg := fmt.Sprintf("service %s", messageSuffix)
  480. w.Events(containerReasonEvents(waitingFor, progress.ErrorMessageEvent, msg))
  481. return errors.New(msg)
  482. }
  483. default:
  484. logrus.Warnf("unsupported depends_on condition: %s", config.Condition)
  485. return nil
  486. }
  487. }
  488. })
  489. }
  490. err := eg.Wait()
  491. if errors.Is(err, context.DeadlineExceeded) {
  492. return fmt.Errorf("timeout waiting for dependencies")
  493. }
  494. return err
  495. }
  496. func shouldWaitForDependency(serviceName string, dependencyConfig types.ServiceDependency, project *types.Project) (bool, error) {
  497. if dependencyConfig.Condition == types.ServiceConditionStarted {
  498. // already managed by InDependencyOrder
  499. return false, nil
  500. }
  501. if service, err := project.GetService(serviceName); err != nil {
  502. for _, ds := range project.DisabledServices {
  503. if ds.Name == serviceName {
  504. // don't wait for disabled service (--no-deps)
  505. return false, nil
  506. }
  507. }
  508. return false, err
  509. } else if service.GetScale() == 0 {
  510. // don't wait for the dependency which configured to have 0 containers running
  511. return false, nil
  512. }
  513. return true, nil
  514. }
  515. func nextContainerNumber(containers []containerType.Summary) int {
  516. maxNumber := 0
  517. for _, c := range containers {
  518. s, ok := c.Labels[api.ContainerNumberLabel]
  519. if !ok {
  520. logrus.Warnf("container %s is missing %s label", c.ID, api.ContainerNumberLabel)
  521. }
  522. n, err := strconv.Atoi(s)
  523. if err != nil {
  524. logrus.Warnf("container %s has invalid %s label: %s", c.ID, api.ContainerNumberLabel, s)
  525. continue
  526. }
  527. if n > maxNumber {
  528. maxNumber = n
  529. }
  530. }
  531. return maxNumber + 1
  532. }
  533. func (s *composeService) createContainer(ctx context.Context, project *types.Project, service types.ServiceConfig,
  534. name string, number int, opts createOptions,
  535. ) (ctr containerType.Summary, err error) {
  536. w := progress.ContextWriter(ctx)
  537. eventName := "Container " + name
  538. w.Event(progress.CreatingEvent(eventName))
  539. ctr, err = s.createMobyContainer(ctx, project, service, name, number, nil, opts, w)
  540. if err != nil {
  541. if ctx.Err() == nil {
  542. w.Event(progress.Event{
  543. ID: eventName,
  544. Status: progress.Error,
  545. StatusText: err.Error(),
  546. })
  547. }
  548. return
  549. }
  550. w.Event(progress.CreatedEvent(eventName))
  551. return
  552. }
  553. func (s *composeService) recreateContainer(ctx context.Context, project *types.Project, service types.ServiceConfig,
  554. replaced containerType.Summary, inherit bool, timeout *time.Duration,
  555. ) (created containerType.Summary, err error) {
  556. w := progress.ContextWriter(ctx)
  557. eventName := getContainerProgressName(replaced)
  558. w.Event(progress.NewEvent(eventName, progress.Working, "Recreate"))
  559. defer func() {
  560. if err != nil && ctx.Err() == nil {
  561. w.Event(progress.Event{
  562. ID: eventName,
  563. Status: progress.Error,
  564. StatusText: err.Error(),
  565. })
  566. }
  567. }()
  568. number, err := strconv.Atoi(replaced.Labels[api.ContainerNumberLabel])
  569. if err != nil {
  570. return created, err
  571. }
  572. var inherited *containerType.Summary
  573. if inherit {
  574. inherited = &replaced
  575. }
  576. name := getContainerName(project.Name, service, number)
  577. tmpName := fmt.Sprintf("%s_%s", replaced.ID[:12], name)
  578. opts := createOptions{
  579. AutoRemove: false,
  580. AttachStdin: false,
  581. UseNetworkAliases: true,
  582. Labels: mergeLabels(service.Labels, service.CustomLabels).Add(api.ContainerReplaceLabel, replaced.ID),
  583. }
  584. created, err = s.createMobyContainer(ctx, project, service, tmpName, number, inherited, opts, w)
  585. if err != nil {
  586. return created, err
  587. }
  588. timeoutInSecond := utils.DurationSecondToInt(timeout)
  589. err = s.apiClient().ContainerStop(ctx, replaced.ID, containerType.StopOptions{Timeout: timeoutInSecond})
  590. if err != nil {
  591. return created, err
  592. }
  593. err = s.apiClient().ContainerRemove(ctx, replaced.ID, containerType.RemoveOptions{})
  594. if err != nil {
  595. return created, err
  596. }
  597. err = s.apiClient().ContainerRename(ctx, created.ID, name)
  598. if err != nil {
  599. return created, err
  600. }
  601. w.Event(progress.NewEvent(eventName, progress.Done, "Recreated"))
  602. return created, err
  603. }
  604. // force sequential calls to ContainerStart to prevent race condition in engine assigning ports from ranges
  605. var startMx sync.Mutex
  606. func (s *composeService) startContainer(ctx context.Context, ctr containerType.Summary) error {
  607. w := progress.ContextWriter(ctx)
  608. w.Event(progress.NewEvent(getContainerProgressName(ctr), progress.Working, "Restart"))
  609. startMx.Lock()
  610. defer startMx.Unlock()
  611. err := s.apiClient().ContainerStart(ctx, ctr.ID, containerType.StartOptions{})
  612. if err != nil {
  613. return err
  614. }
  615. w.Event(progress.NewEvent(getContainerProgressName(ctr), progress.Done, "Restarted"))
  616. return nil
  617. }
  618. func (s *composeService) createMobyContainer(ctx context.Context,
  619. project *types.Project,
  620. service types.ServiceConfig,
  621. name string,
  622. number int,
  623. inherit *containerType.Summary,
  624. opts createOptions,
  625. w progress.Writer,
  626. ) (containerType.Summary, error) {
  627. var created containerType.Summary
  628. cfgs, err := s.getCreateConfigs(ctx, project, service, number, inherit, opts)
  629. if err != nil {
  630. return created, err
  631. }
  632. platform := service.Platform
  633. if platform == "" {
  634. platform = project.Environment["DOCKER_DEFAULT_PLATFORM"]
  635. }
  636. var plat *specs.Platform
  637. if platform != "" {
  638. var p specs.Platform
  639. p, err = platforms.Parse(platform)
  640. if err != nil {
  641. return created, err
  642. }
  643. plat = &p
  644. }
  645. response, err := s.apiClient().ContainerCreate(ctx, cfgs.Container, cfgs.Host, cfgs.Network, plat, name)
  646. if err != nil {
  647. return created, err
  648. }
  649. for _, warning := range response.Warnings {
  650. w.Event(progress.Event{
  651. ID: service.Name,
  652. Status: progress.Warning,
  653. Text: warning,
  654. })
  655. }
  656. inspectedContainer, err := s.apiClient().ContainerInspect(ctx, response.ID)
  657. if err != nil {
  658. return created, err
  659. }
  660. created = containerType.Summary{
  661. ID: inspectedContainer.ID,
  662. Labels: inspectedContainer.Config.Labels,
  663. Names: []string{inspectedContainer.Name},
  664. NetworkSettings: &containerType.NetworkSettingsSummary{
  665. Networks: inspectedContainer.NetworkSettings.Networks,
  666. },
  667. }
  668. apiVersion, err := s.RuntimeVersion(ctx)
  669. if err != nil {
  670. return created, err
  671. }
  672. // Starting API version 1.44, the ContainerCreate API call takes multiple networks
  673. // so we include all the configurations there and can skip the one-by-one calls here
  674. if versions.LessThan(apiVersion, "1.44") {
  675. // the highest-priority network is the primary and is included in the ContainerCreate API
  676. // call via container.NetworkMode & network.NetworkingConfig
  677. // any remaining networks are connected one-by-one here after creation (but before start)
  678. serviceNetworks := service.NetworksByPriority()
  679. for _, networkKey := range serviceNetworks {
  680. mobyNetworkName := project.Networks[networkKey].Name
  681. if string(cfgs.Host.NetworkMode) == mobyNetworkName {
  682. // primary network already configured as part of ContainerCreate
  683. continue
  684. }
  685. epSettings := createEndpointSettings(project, service, number, networkKey, cfgs.Links, opts.UseNetworkAliases)
  686. if err := s.apiClient().NetworkConnect(ctx, mobyNetworkName, created.ID, epSettings); err != nil {
  687. return created, err
  688. }
  689. }
  690. }
  691. err = s.injectSecrets(ctx, project, service, created.ID)
  692. if err != nil {
  693. return created, err
  694. }
  695. err = s.injectConfigs(ctx, project, service, created.ID)
  696. return created, err
  697. }
  698. // getLinks mimics V1 compose/service.py::Service::_get_links()
  699. func (s *composeService) getLinks(ctx context.Context, projectName string, service types.ServiceConfig, number int) ([]string, error) {
  700. var links []string
  701. format := func(k, v string) string {
  702. return fmt.Sprintf("%s:%s", k, v)
  703. }
  704. getServiceContainers := func(serviceName string) (Containers, error) {
  705. return s.getContainers(ctx, projectName, oneOffExclude, true, serviceName)
  706. }
  707. for _, rawLink := range service.Links {
  708. linkSplit := strings.Split(rawLink, ":")
  709. linkServiceName := linkSplit[0]
  710. linkName := linkServiceName
  711. if len(linkSplit) == 2 {
  712. linkName = linkSplit[1] // linkName if informed like in: "serviceName:linkName"
  713. }
  714. cnts, err := getServiceContainers(linkServiceName)
  715. if err != nil {
  716. return nil, err
  717. }
  718. for _, c := range cnts {
  719. containerName := getCanonicalContainerName(c)
  720. links = append(links,
  721. format(containerName, linkName),
  722. format(containerName, linkServiceName+api.Separator+strconv.Itoa(number)),
  723. format(containerName, strings.Join([]string{projectName, linkServiceName, strconv.Itoa(number)}, api.Separator)),
  724. )
  725. }
  726. }
  727. if service.Labels[api.OneoffLabel] == "True" {
  728. cnts, err := getServiceContainers(service.Name)
  729. if err != nil {
  730. return nil, err
  731. }
  732. for _, c := range cnts {
  733. containerName := getCanonicalContainerName(c)
  734. links = append(links,
  735. format(containerName, service.Name),
  736. format(containerName, strings.TrimPrefix(containerName, projectName+api.Separator)),
  737. format(containerName, containerName),
  738. )
  739. }
  740. }
  741. for _, rawExtLink := range service.ExternalLinks {
  742. extLinkSplit := strings.Split(rawExtLink, ":")
  743. externalLink := extLinkSplit[0]
  744. linkName := externalLink
  745. if len(extLinkSplit) == 2 {
  746. linkName = extLinkSplit[1]
  747. }
  748. links = append(links, format(externalLink, linkName))
  749. }
  750. return links, nil
  751. }
  752. func (s *composeService) isServiceHealthy(ctx context.Context, containers Containers, fallbackRunning bool) (bool, error) {
  753. for _, c := range containers {
  754. container, err := s.apiClient().ContainerInspect(ctx, c.ID)
  755. if err != nil {
  756. return false, err
  757. }
  758. name := container.Name[1:]
  759. if container.State.Status == "exited" {
  760. return false, fmt.Errorf("container %s exited (%d)", name, container.State.ExitCode)
  761. }
  762. if container.Config.Healthcheck == nil && fallbackRunning {
  763. // Container does not define a health check, but we can fall back to "running" state
  764. return container.State != nil && container.State.Status == "running", nil
  765. }
  766. if container.State == nil || container.State.Health == nil {
  767. return false, fmt.Errorf("container %s has no healthcheck configured", name)
  768. }
  769. switch container.State.Health.Status {
  770. case containerType.Healthy:
  771. // Continue by checking the next container.
  772. case containerType.Unhealthy:
  773. return false, fmt.Errorf("container %s is unhealthy", name)
  774. case containerType.Starting:
  775. return false, nil
  776. default:
  777. return false, fmt.Errorf("container %s had unexpected health status %q", name, container.State.Health.Status)
  778. }
  779. }
  780. return true, nil
  781. }
  782. func (s *composeService) isServiceCompleted(ctx context.Context, containers Containers) (bool, int, error) {
  783. for _, c := range containers {
  784. container, err := s.apiClient().ContainerInspect(ctx, c.ID)
  785. if err != nil {
  786. return false, 0, err
  787. }
  788. if container.State != nil && container.State.Status == "exited" {
  789. return true, container.State.ExitCode, nil
  790. }
  791. }
  792. return false, 0, nil
  793. }
  794. func (s *composeService) startService(ctx context.Context,
  795. project *types.Project, service types.ServiceConfig,
  796. containers Containers, listener api.ContainerEventListener,
  797. timeout time.Duration,
  798. ) error {
  799. if service.Deploy != nil && service.Deploy.Replicas != nil && *service.Deploy.Replicas == 0 {
  800. return nil
  801. }
  802. err := s.waitDependencies(ctx, project, service.Name, service.DependsOn, containers, timeout)
  803. if err != nil {
  804. return err
  805. }
  806. if len(containers) == 0 {
  807. if service.GetScale() == 0 {
  808. return nil
  809. }
  810. return fmt.Errorf("service %q has no container to start", service.Name)
  811. }
  812. w := progress.ContextWriter(ctx)
  813. for _, ctr := range containers.filter(isService(service.Name)) {
  814. if ctr.State == ContainerRunning {
  815. continue
  816. }
  817. eventName := getContainerProgressName(ctr)
  818. w.Event(progress.StartingEvent(eventName))
  819. err = s.apiClient().ContainerStart(ctx, ctr.ID, containerType.StartOptions{})
  820. if err != nil {
  821. return err
  822. }
  823. for _, hook := range service.PostStart {
  824. err = s.runHook(ctx, ctr, service, hook, listener)
  825. if err != nil {
  826. return err
  827. }
  828. }
  829. w.Event(progress.StartedEvent(eventName))
  830. }
  831. return nil
  832. }
  833. func mergeLabels(ls ...types.Labels) types.Labels {
  834. merged := types.Labels{}
  835. for _, l := range ls {
  836. for k, v := range l {
  837. merged[k] = v
  838. }
  839. }
  840. return merged
  841. }