presidium 0.26.6 → 0.26.8

package/Archive.test.js

@@ -36,7 +36,7 @@ const test = new Test('Archive', Archive)
 })
 
 .case({
-  Dockerfile: 'FROM busybox:1.32'
+  Dockerfile: 'FROM busybox:1.32',
   '.aws/credentials': '[claimyr]\naccessKeyId\nsecretAccessKey',
 }, async archive => {
   const pack = await archive.tar(`${pathResolve(__dirname, 'internal')}/`, {

package/Docker.js

@@ -328,6 +328,7 @@ Docker.prototype.removeImage = function dockerRemoveImage(image, options) {
  *   healthStartPeriod: >=1e6, // nanoseconds to wait on container init before starting first healthcheck
  *   memory: number, // memory limit in bytes
  *   cpus: number, // number of cpus
+ *   gpus?: 'all', // expose gpus
  *   mounts: Array<{
  *     source: string, // name of volume
  *     target: string, // mounted path inside container
@@ -468,6 +469,7 @@ Docker.prototype.createContainer = function dockerCreateContainer(
           })(options.restart.split(':')),
         },
         ...options.rm && { AutoRemove: options.rm },
+
       },
     }),
 
@@ -634,6 +636,9 @@ Docker.prototype.joinSwarm = async function dockerJoinSwarm(
   address, token, options = {}
 ) {
   return this.http.post('/swarm/join', {
+    headers: {
+      'Content-Type': 'application/json',
+    },
     body: stringifyJSON({
       JoinToken: token,
       AdvertiseAddr: get('advertiseAddr', address)(options),
@@ -868,6 +873,15 @@ Docker.prototype.createService = function dockerCreateService(service, options)
             ...options.cpus ? {
               NanoCPUs: Number(options.cpus * 1e9),
             } : {},
+
+            ...options.gpus == 'all' ? {
+              GenericResources: [{
+                DiscreteResourceSpec: {
+                  Kind: 'gpu',
+                  Value: 1,
+                },
+              }],
+            } : {},
           }, // bytes
         },
 

package/Docker.test.js

@@ -335,6 +335,7 @@ EXPOSE 8888`,
         }],
         memory: 512e6, // bytes
         cpus: 2,
+        gpus: 'all',
         restart: 'on-failure:5',
 
         healthCmd: ['wget', '--no-verbose', '--tries=1', '--spider', 'localhost:3000'],
@@ -364,6 +365,10 @@ EXPOSE 8888`,
         assert.equal(data.Spec.Labels.foo, 'bar')
         assert.equal(data.Spec.TaskTemplate.Resources.Reservations.NanoCPUs, 2000000000)
         assert.equal(data.Spec.TaskTemplate.Resources.Reservations.MemoryBytes, 512000000)
+        assert.equal(
+          data.Spec.TaskTemplate.Resources.Reservations.GenericResources[0].DiscreteResourceSpec.Kind,
+          'gpu',
+        )
       })
 
       this.serviceId1 = serviceId

package/DockerService.js

@@ -26,7 +26,7 @@ const dockerServiceOptions = [
   'rollbackParallelism', 'rollbackDelay',
   'rollbackFailureAction', 'rollbackMonitor', 'rollbackMaxFailureRatio',
   'username', 'password', 'email', 'serveraddress', 'identitytoken',
-  'network', 'memory', 'cpus',
+  'network', 'memory', 'cpus', 'gpus',
   'force',
 ]
 

package/internal/createUpdateServiceSpec.js

@@ -130,7 +130,7 @@ const createUpdateServiceSpec = function (options) {
   if (options.restartDelay != null) {
     result.TaskTemplate.RestartPolicy.Delay = Number(options.restartDelay)
   }
-  if (or([has('memory'), has('cpus')])(options)) {
+  if (or(options, [has('memory'), has('cpus'), has('gpus')])) {
     result.TaskTemplate.Resources = {}
     result.TaskTemplate.Resources.Reservations = {}
     result.TaskTemplate.Resources.Limits = {}
@@ -147,6 +147,15 @@ const createUpdateServiceSpec = function (options) {
     result.TaskTemplate.Resources.Limits.NanoCPUs =
       Number(options.cpus * 1e9)
   }
+  if (options.gpus != null) {
+    result.TaskTemplate.Resources.Reservations.GenericResources = []
+    result.TaskTemplate.Resources.Reservations.GenericResources.push({
+      DiscreteResourceSpec: {
+        Kind: 'gpu',
+        Value: 1,
+      },
+    })
+  }
   if (or([has('logDriver'), has('logDriverOptions')])(options)) {
     result.TaskTemplate.LogDriver = {}
   }

package/internal/createUpdateServiceSpec.test.js

@@ -241,6 +241,29 @@ const test = new Test('createUpdateServiceSpec', createUpdateServiceSpec)
   },
 })
 
+.case({
+  serviceName: 'my-service',
+  spec: {
+    TaskTemplate: {},
+  },
+  gpus: 'all',
+}, {
+  Name: 'my-service',
+  TaskTemplate: {
+    Resources: {
+      Limits: {},
+      Reservations: {
+        GenericResources: [{
+          DiscreteResourceSpec: {
+            Kind: 'gpu',
+            Value: 1,
+          },
+        }],
+      },
+    },
+  },
+})
+
 .case({
   serviceName: 'my-service',
   spec: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "presidium",
-  "version": "0.26.6",
+  "version": "0.26.8",
   "description": "A library for creating web services",
   "author": "Richard Tong",
   "license": "MIT",