dgrove-oss commented on a change in pull request #3338: implement suspend/resume for KubernetesContainer URL: https://github.com/apache/incubator-openwhisk/pull/3338#discussion_r170916241
########## File path: core/invoker/src/main/scala/whisk/core/containerpool/kubernetes/KubernetesClient.scala ########## @@ -99,43 +116,139 @@ class KubernetesClient( } protected val kubectlCmd = Seq(findKubectlCmd) - def run(name: String, image: String, args: Seq[String] = Seq.empty[String])( - implicit transid: TransactionId): Future[ContainerId] = { - runCmd(Seq("run", name, s"--image=$image") ++ args, timeouts.run) - .map(_ => ContainerId(name)) - } + def run(name: String, + image: String, + memory: ByteSize = 256.MB, + environment: Map[String, String] = Map(), + labels: Map[String, String] = Map())(implicit transid: TransactionId): Future[KubernetesContainer] = { + + val envVars = environment.map { + case (key, value) => new EnvVarBuilder().withName(key).withValue(value).build() + }.toSeq + + val pod = new PodBuilder() + .withNewMetadata() + .withName(name) + .addToLabels("name", name) + .addToLabels(labels.asJava) + .endMetadata() + .withNewSpec() + .withRestartPolicy("Always") + .addNewContainer() + .withNewResources() + .withLimits(Map("memory" -> new Quantity(memory.toMB + "Mi")).asJava) + .endResources() + .withName("user-action") + .withImage(image) + .withEnv(envVars.asJava) + .addNewPort() + .withContainerPort(8080) + .withName("action") + .endPort() + .endContainer() + .endSpec() + .build() + + kubeRestClient.pods.inNamespace("openwhisk").create(pod) - def inspectIPAddress(id: ContainerId)(implicit transid: TransactionId): Future[ContainerAddress] = { Future { blocking { - val pod = - kubeRestClient.pods().withName(id.asString).waitUntilReady(timeouts.inspect.length, timeouts.inspect.unit) - ContainerAddress(pod.getStatus().getPodIP()) + val createdPod = kubeRestClient.pods + .inNamespace("openwhisk") + .withName(name) + .waitUntilReady(config.timeouts.run.length, config.timeouts.run.unit) + toContainer(createdPod) } }.recoverWith { case e => - log.error(this, s"Failed to get IP of Pod '${id.asString}' within timeout: ${e.getClass} - ${e.getMessage}") - Future.failed(new Exception(s"Failed to get IP of Pod '${id.asString}'")) + log.error(this, s"Failed create pod for '$name': ${e.getClass} - ${e.getMessage}") + Future.failed(new Exception(s"Failed to create pod '$name'")) + } + } + + def rm(container: KubernetesContainer)(implicit transid: TransactionId): Future[Unit] = { + runCmd(Seq("delete", "--now", "pod", container.id.asString), config.timeouts.rm).map(_ => ()) + } + + def rm(key: String, value: String)(implicit transid: TransactionId): Future[Unit] = { + if (config.invokerAgent.enabled) { + Future { + blocking { + kubeRestClient + .inNamespace("openwhisk") Review comment: good point. Moved the namespace value into application.conf. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services