cs-codex-dist-tests/CodexDistTestCore/Metrics/K8sPrometheusSpecs.cs

using CodexDistTestCore.Config;
using k8s.Models;

namespace CodexDistTestCore.Metrics
{
    public class K8sPrometheusSpecs
    {
        public const string ContainerName = "dtest-prom";
        public const string ConfigFilepath = "/etc/prometheus/prometheus.yml";
        private const string dockerImage = "thatbenbierens/prometheus-envconf:latest";
        private const string portName = "prom-1";
        private readonly string config;

        public K8sPrometheusSpecs(int servicePort, int prometheusNumber, string config)
        {
            ServicePort = servicePort;
            PrometheusNumber = prometheusNumber;
            this.config = config;
        }

        public int ServicePort { get; }
        public int PrometheusNumber { get; }

        public string GetDeploymentName()
        {
            return "test-prom" + PrometheusNumber;
        }

        public V1Deployment CreatePrometheusDeployment()
        {
            var deploymentSpec = new V1Deployment
            {
                ApiVersion = "apps/v1",
                Metadata = new V1ObjectMeta
                {
                    Name = GetDeploymentName(),
                    NamespaceProperty = K8sCluster.K8sNamespace
                },
                Spec = new V1DeploymentSpec
                {
                    Replicas = 1,
                    Selector = new V1LabelSelector
                    {
                        MatchLabels = CreateSelector()
                    },
                    Template = new V1PodTemplateSpec
                    {
                        Metadata = new V1ObjectMeta
                        {
                            Labels = CreateSelector()
                        },
                        Spec = new V1PodSpec
                        {
                            Containers = new List<V1Container>
                            {
                                new V1Container
                                {
                                    Name = ContainerName,
                                    Image = dockerImage,
                                    Ports = new List<V1ContainerPort>
                                    {
                                        new V1ContainerPort
                                        {
                                            ContainerPort = 9090,
                                            Name = portName
                                        }
                                    },
                                    Env = new List<V1EnvVar>
                                    {
                                        new V1EnvVar
                                        {
                                            Name = "PROM_CONFIG",
                                            Value = config
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            };

            return deploymentSpec;
        }

        public V1Service CreatePrometheusService()
        {
            var serviceSpec = new V1Service
            {
                ApiVersion = "v1",
                Metadata = new V1ObjectMeta
                {
                    Name = "codex-prom-service" + PrometheusNumber,
                    NamespaceProperty = K8sCluster.K8sNamespace
                },
                Spec = new V1ServiceSpec
                {
                    Type = "NodePort",
                    Selector = CreateSelector(),
                    Ports = new List<V1ServicePort>
                    {
                        new V1ServicePort
                        {
                            Name = "prom-service" + PrometheusNumber,
                            Protocol = "TCP",
                            Port = 9090,
                            TargetPort = portName,
                            NodePort = ServicePort
                        }
                    }
                }
            };

            return serviceSpec;
        }

        private Dictionary<string, string> CreateSelector()
        {
            return new Dictionary<string, string> { { "test-prom", "dtest-prom" } };
        }
    }
}