From 34a33d614efcd6c693cb86f6f1d7e4b8df8772ec Mon Sep 17 00:00:00 2001
From: Andy Librian <andylibrian@gmail.com>
Date: Fri, 9 Oct 2020 05:54:16 +0700
Subject: Add support for resource cpu limit to generate kube

fixes #7855

Signed-off-by: Andy Librian <andylibrian@gmail.com>
---
 libpod/kube.go | 40 +++++++++++++++++++++++++++++++---------
 1 file changed, 31 insertions(+), 9 deletions(-)

(limited to 'libpod')

diff --git a/libpod/kube.go b/libpod/kube.go
index 6df79e394..cd5064c84 100644
--- a/libpod/kube.go
+++ b/libpod/kube.go
@@ -307,18 +307,40 @@ func containerToV1Container(c *Container) (v1.Container, []v1.Volume, error) {
 	kubeContainer.StdinOnce = false
 	kubeContainer.TTY = c.config.Spec.Process.Terminal
 
-	// TODO add CPU limit support.
 	if c.config.Spec.Linux != nil &&
-		c.config.Spec.Linux.Resources != nil &&
-		c.config.Spec.Linux.Resources.Memory != nil &&
-		c.config.Spec.Linux.Resources.Memory.Limit != nil {
-		if kubeContainer.Resources.Limits == nil {
-			kubeContainer.Resources.Limits = v1.ResourceList{}
+		c.config.Spec.Linux.Resources != nil {
+		if c.config.Spec.Linux.Resources.Memory != nil &&
+			c.config.Spec.Linux.Resources.Memory.Limit != nil {
+			if kubeContainer.Resources.Limits == nil {
+				kubeContainer.Resources.Limits = v1.ResourceList{}
+			}
+
+			qty := kubeContainer.Resources.Limits.Memory()
+			qty.Set(*c.config.Spec.Linux.Resources.Memory.Limit)
+			kubeContainer.Resources.Limits[v1.ResourceMemory] = *qty
 		}
 
-		qty := kubeContainer.Resources.Limits.Memory()
-		qty.Set(*c.config.Spec.Linux.Resources.Memory.Limit)
-		kubeContainer.Resources.Limits[v1.ResourceMemory] = *qty
+		if c.config.Spec.Linux.Resources.CPU != nil &&
+			c.config.Spec.Linux.Resources.CPU.Quota != nil &&
+			c.config.Spec.Linux.Resources.CPU.Period != nil {
+			quota := *c.config.Spec.Linux.Resources.CPU.Quota
+			period := *c.config.Spec.Linux.Resources.CPU.Period
+
+			if quota > 0 && period > 0 {
+				cpuLimitMilli := int64(1000 * float64(quota) / float64(period))
+
+				// Kubernetes: precision finer than 1m is not allowed
+				if cpuLimitMilli >= 1 {
+					if kubeContainer.Resources.Limits == nil {
+						kubeContainer.Resources.Limits = v1.ResourceList{}
+					}
+
+					qty := kubeContainer.Resources.Limits.Cpu()
+					qty.SetMilli(cpuLimitMilli)
+					kubeContainer.Resources.Limits[v1.ResourceCPU] = *qty
+				}
+			}
+		}
 	}
 
 	return kubeContainer, kubeVolumes, nil
-- 
cgit v1.2.3-54-g00ecf