mirror of
https://github.com/docker/compose.git
synced 2025-04-08 17:05:13 +02:00
365 lines
12 KiB
Python
365 lines
12 KiB
Python
from __future__ import unicode_literals
|
|
from __future__ import absolute_import
|
|
from functools import reduce
|
|
import logging
|
|
|
|
from docker.errors import APIError
|
|
|
|
from .config import get_service_name_from_net, ConfigurationError
|
|
from .const import DEFAULT_TIMEOUT, LABEL_PROJECT, LABEL_SERVICE, LABEL_ONE_OFF
|
|
from .container import Container
|
|
from .legacy import check_for_legacy_containers
|
|
from .service import Service
|
|
from .utils import parallel_execute
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def sort_service_dicts(services):
|
|
# Topological sort (Cormen/Tarjan algorithm).
|
|
unmarked = services[:]
|
|
temporary_marked = set()
|
|
sorted_services = []
|
|
|
|
def get_service_names(links):
|
|
return [link.split(':')[0] for link in links]
|
|
|
|
def get_service_dependents(service_dict, services):
|
|
name = service_dict['name']
|
|
return [
|
|
service for service in services
|
|
if (name in get_service_names(service.get('links', [])) or
|
|
name in service.get('volumes_from', []) or
|
|
name == get_service_name_from_net(service.get('net')))
|
|
]
|
|
|
|
def visit(n):
|
|
if n['name'] in temporary_marked:
|
|
if n['name'] in get_service_names(n.get('links', [])):
|
|
raise DependencyError('A service can not link to itself: %s' % n['name'])
|
|
if n['name'] in n.get('volumes_from', []):
|
|
raise DependencyError('A service can not mount itself as volume: %s' % n['name'])
|
|
else:
|
|
raise DependencyError('Circular import between %s' % ' and '.join(temporary_marked))
|
|
if n in unmarked:
|
|
temporary_marked.add(n['name'])
|
|
for m in get_service_dependents(n, services):
|
|
visit(m)
|
|
temporary_marked.remove(n['name'])
|
|
unmarked.remove(n)
|
|
sorted_services.insert(0, n)
|
|
|
|
while unmarked:
|
|
visit(unmarked[-1])
|
|
|
|
return sorted_services
|
|
|
|
|
|
class Project(object):
|
|
"""
|
|
A collection of services.
|
|
"""
|
|
def __init__(self, name, services, client):
|
|
self.name = name
|
|
self.services = services
|
|
self.client = client
|
|
|
|
def labels(self, one_off=False):
|
|
return [
|
|
'{0}={1}'.format(LABEL_PROJECT, self.name),
|
|
'{0}={1}'.format(LABEL_ONE_OFF, "True" if one_off else "False"),
|
|
]
|
|
|
|
@classmethod
|
|
def from_dicts(cls, name, service_dicts, client):
|
|
"""
|
|
Construct a ServiceCollection from a list of dicts representing services.
|
|
"""
|
|
project = cls(name, [], client)
|
|
for service_dict in sort_service_dicts(service_dicts):
|
|
links = project.get_links(service_dict)
|
|
volumes_from = project.get_volumes_from(service_dict)
|
|
net = project.get_net(service_dict)
|
|
|
|
project.services.append(Service(client=client, project=name, links=links, net=net,
|
|
volumes_from=volumes_from, **service_dict))
|
|
return project
|
|
|
|
@property
|
|
def service_names(self):
|
|
return [service.name for service in self.services]
|
|
|
|
def get_service(self, name):
|
|
"""
|
|
Retrieve a service by name. Raises NoSuchService
|
|
if the named service does not exist.
|
|
"""
|
|
for service in self.services:
|
|
if service.name == name:
|
|
return service
|
|
|
|
raise NoSuchService(name)
|
|
|
|
def validate_service_names(self, service_names):
|
|
"""
|
|
Validate that the given list of service names only contains valid
|
|
services. Raises NoSuchService if one of the names is invalid.
|
|
"""
|
|
valid_names = self.service_names
|
|
for name in service_names:
|
|
if name not in valid_names:
|
|
raise NoSuchService(name)
|
|
|
|
def get_services(self, service_names=None, include_deps=False):
|
|
"""
|
|
Returns a list of this project's services filtered
|
|
by the provided list of names, or all services if service_names is None
|
|
or [].
|
|
|
|
If include_deps is specified, returns a list including the dependencies for
|
|
service_names, in order of dependency.
|
|
|
|
Preserves the original order of self.services where possible,
|
|
reordering as needed to resolve dependencies.
|
|
|
|
Raises NoSuchService if any of the named services do not exist.
|
|
"""
|
|
if service_names is None or len(service_names) == 0:
|
|
return self.get_services(
|
|
service_names=self.service_names,
|
|
include_deps=include_deps
|
|
)
|
|
else:
|
|
unsorted = [self.get_service(name) for name in service_names]
|
|
services = [s for s in self.services if s in unsorted]
|
|
|
|
if include_deps:
|
|
services = reduce(self._inject_deps, services, [])
|
|
|
|
uniques = []
|
|
[uniques.append(s) for s in services if s not in uniques]
|
|
return uniques
|
|
|
|
def get_links(self, service_dict):
|
|
links = []
|
|
if 'links' in service_dict:
|
|
for link in service_dict.get('links', []):
|
|
if ':' in link:
|
|
service_name, link_name = link.split(':', 1)
|
|
else:
|
|
service_name, link_name = link, None
|
|
try:
|
|
links.append((self.get_service(service_name), link_name))
|
|
except NoSuchService:
|
|
raise ConfigurationError('Service "%s" has a link to service "%s" which does not exist.' % (service_dict['name'], service_name))
|
|
del service_dict['links']
|
|
return links
|
|
|
|
def get_volumes_from(self, service_dict):
|
|
volumes_from = []
|
|
if 'volumes_from' in service_dict:
|
|
for volume_name in service_dict.get('volumes_from', []):
|
|
try:
|
|
service = self.get_service(volume_name)
|
|
volumes_from.append(service)
|
|
except NoSuchService:
|
|
try:
|
|
container = Container.from_id(self.client, volume_name)
|
|
volumes_from.append(container)
|
|
except APIError:
|
|
raise ConfigurationError('Service "%s" mounts volumes from "%s", which is not the name of a service or container.' % (service_dict['name'], volume_name))
|
|
del service_dict['volumes_from']
|
|
return volumes_from
|
|
|
|
def get_net(self, service_dict):
|
|
if 'net' in service_dict:
|
|
net_name = get_service_name_from_net(service_dict.get('net'))
|
|
|
|
if net_name:
|
|
try:
|
|
net = self.get_service(net_name)
|
|
except NoSuchService:
|
|
try:
|
|
net = Container.from_id(self.client, net_name)
|
|
except APIError:
|
|
raise ConfigurationError('Service "%s" is trying to use the network of "%s", which is not the name of a service or container.' % (service_dict['name'], net_name))
|
|
else:
|
|
net = service_dict['net']
|
|
|
|
del service_dict['net']
|
|
|
|
else:
|
|
net = None
|
|
|
|
return net
|
|
|
|
def start(self, service_names=None, **options):
|
|
for service in self.get_services(service_names):
|
|
service.start(**options)
|
|
|
|
def stop(self, service_names=None, **options):
|
|
parallel_execute(
|
|
objects=self.containers(service_names),
|
|
obj_callable=lambda c: c.stop(**options),
|
|
msg_index=lambda c: c.name,
|
|
msg="Stopping"
|
|
)
|
|
|
|
def pause(self, service_names=None, **options):
|
|
for service in reversed(self.get_services(service_names)):
|
|
service.pause(**options)
|
|
|
|
def unpause(self, service_names=None, **options):
|
|
for service in self.get_services(service_names):
|
|
service.unpause(**options)
|
|
|
|
def kill(self, service_names=None, **options):
|
|
parallel_execute(
|
|
objects=self.containers(service_names),
|
|
obj_callable=lambda c: c.kill(**options),
|
|
msg_index=lambda c: c.name,
|
|
msg="Killing"
|
|
)
|
|
|
|
def remove_stopped(self, service_names=None, **options):
|
|
all_containers = self.containers(service_names, stopped=True)
|
|
stopped_containers = [c for c in all_containers if not c.is_running]
|
|
parallel_execute(
|
|
objects=stopped_containers,
|
|
obj_callable=lambda c: c.remove(**options),
|
|
msg_index=lambda c: c.name,
|
|
msg="Removing"
|
|
)
|
|
|
|
def restart(self, service_names=None, **options):
|
|
for service in self.get_services(service_names):
|
|
service.restart(**options)
|
|
|
|
def build(self, service_names=None, no_cache=False):
|
|
for service in self.get_services(service_names):
|
|
if service.can_be_built():
|
|
service.build(no_cache)
|
|
else:
|
|
log.info('%s uses an image, skipping' % service.name)
|
|
|
|
def up(self,
|
|
service_names=None,
|
|
start_deps=True,
|
|
allow_recreate=True,
|
|
force_recreate=False,
|
|
do_build=True,
|
|
timeout=DEFAULT_TIMEOUT):
|
|
|
|
if force_recreate and not allow_recreate:
|
|
raise ValueError("force_recreate and allow_recreate are in conflict")
|
|
|
|
services = self.get_services(service_names, include_deps=start_deps)
|
|
|
|
for service in services:
|
|
service.remove_duplicate_containers()
|
|
|
|
plans = self._get_convergence_plans(
|
|
services,
|
|
allow_recreate=allow_recreate,
|
|
force_recreate=force_recreate,
|
|
)
|
|
|
|
return [
|
|
container
|
|
for service in services
|
|
for container in service.execute_convergence_plan(
|
|
plans[service.name],
|
|
do_build=do_build,
|
|
timeout=timeout
|
|
)
|
|
]
|
|
|
|
def _get_convergence_plans(self,
|
|
services,
|
|
allow_recreate=True,
|
|
force_recreate=False):
|
|
|
|
plans = {}
|
|
|
|
for service in services:
|
|
updated_dependencies = [
|
|
name
|
|
for name in service.get_dependency_names()
|
|
if name in plans
|
|
and plans[name].action == 'recreate'
|
|
]
|
|
|
|
if updated_dependencies and allow_recreate:
|
|
log.debug(
|
|
'%s has upstream changes (%s)',
|
|
service.name, ", ".join(updated_dependencies),
|
|
)
|
|
plan = service.convergence_plan(
|
|
allow_recreate=allow_recreate,
|
|
force_recreate=True,
|
|
)
|
|
else:
|
|
plan = service.convergence_plan(
|
|
allow_recreate=allow_recreate,
|
|
force_recreate=force_recreate,
|
|
)
|
|
|
|
plans[service.name] = plan
|
|
|
|
return plans
|
|
|
|
def pull(self, service_names=None):
|
|
for service in self.get_services(service_names, include_deps=True):
|
|
service.pull()
|
|
|
|
def containers(self, service_names=None, stopped=False, one_off=False):
|
|
if service_names:
|
|
self.validate_service_names(service_names)
|
|
else:
|
|
service_names = self.service_names
|
|
|
|
containers = filter(None, [
|
|
Container.from_ps(self.client, container)
|
|
for container in self.client.containers(
|
|
all=stopped,
|
|
filters={'label': self.labels(one_off=one_off)})])
|
|
|
|
def matches_service_names(container):
|
|
return container.labels.get(LABEL_SERVICE) in service_names
|
|
|
|
if not containers:
|
|
check_for_legacy_containers(
|
|
self.client,
|
|
self.name,
|
|
self.service_names,
|
|
)
|
|
|
|
return filter(matches_service_names, containers)
|
|
|
|
def _inject_deps(self, acc, service):
|
|
dep_names = service.get_dependency_names()
|
|
|
|
if len(dep_names) > 0:
|
|
dep_services = self.get_services(
|
|
service_names=list(set(dep_names)),
|
|
include_deps=True
|
|
)
|
|
else:
|
|
dep_services = []
|
|
|
|
dep_services.append(service)
|
|
return acc + dep_services
|
|
|
|
|
|
class NoSuchService(Exception):
|
|
def __init__(self, name):
|
|
self.name = name
|
|
self.msg = "No such service: %s" % self.name
|
|
|
|
def __str__(self):
|
|
return self.msg
|
|
|
|
|
|
class DependencyError(ConfigurationError):
|
|
pass
|