introducing pids_limit, fix for #4178

onorua · shin- · commit 6a151aac046b · 2017-03-02T16:44:26.000-08:00
Signed-off-by: Yaroslav Molochko &lt;y.molochko@anchorfree.com&gt;
diff --git a/compose/config/config.py b/compose/config/config.py
@@ -87,6 +87,7 @@
     'secrets',
     'security_opt',
     'shm_size',
+    'pids_limit',
     'stdin_open',
     'stop_signal',
     'sysctls',
diff --git a/compose/config/config_schema_v2.1.json b/compose/config/config_schema_v2.1.json
@@ -216,6 +216,7 @@
         "security_opt": {"type": "array", "items": {"type": "string"}, "uniqueItems": true},
         "shm_size": {"type": ["number", "string"]},
         "sysctls": {"$ref": "#/definitions/list_or_dict"},
+        "pids_limit": {"type": ["number", "string"]},
         "stdin_open": {"type": "boolean"},
         "stop_grace_period": {"type": "string", "format": "duration"},
         "stop_signal": {"type": "string"},
diff --git a/compose/config/config_schema_v3.0.json b/compose/config/config_schema_v3.0.json
@@ -168,6 +168,7 @@
         "security_opt": {"type": "array", "items": {"type": "string"}, "uniqueItems": true},
         "shm_size": {"type": ["number", "string"]},
         "sysctls": {"$ref": "#/definitions/list_or_dict"},
+        "pids_limit": {"type": ["number", "string"]},
         "stdin_open": {"type": "boolean"},
         "stop_grace_period": {"type": "string", "format": "duration"},
         "stop_signal": {"type": "string"},
diff --git a/compose/service.py b/compose/service.py
@@ -66,6 +66,7 @@
     'oom_score_adj',
     'mem_swappiness',
     'pid',
+    'pids_limit',
     'privileged',
     'restart',
     'security_opt',
@@ -772,6 +773,7 @@ def _get_container_host_config(self, override_options, one_off=False):
             cpu_quota=options.get('cpu_quota'),
             shm_size=options.get('shm_size'),
             sysctls=options.get('sysctls'),
+            pids_limit=options.get('pids_limit'),
             tmpfs=options.get('tmpfs'),
             oom_score_adj=options.get('oom_score_adj'),
             mem_swappiness=options.get('mem_swappiness'),
diff --git a/tests/integration/service_test.py b/tests/integration/service_test.py
@@ -115,6 +115,13 @@ def test_create_container_with_shm_size(self):
         service.start_container(container)
         self.assertEqual(container.get('HostConfig.ShmSize'), 67108864)
 
+    def test_create_container_with_pids_limit(self):
+        self.require_api_version('1.23')
+        service = self.create_service('db', pids_limit=10)
+        container = service.create_container()
+        service.start_container(container)
+        self.assertEqual(container.get('HostConfig.PidsLimit'), 10)
+
     def test_create_container_with_extra_hosts_list(self):
         extra_hosts = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
         service = self.create_service('db', extra_hosts=extra_hosts)
