metrics.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773
  1. package costmodel
  2. import (
  3. "math"
  4. "strconv"
  5. "strings"
  6. "sync"
  7. "time"
  8. "github.com/kubecost/cost-model/pkg/cloud"
  9. "github.com/kubecost/cost-model/pkg/clustercache"
  10. "github.com/kubecost/cost-model/pkg/costmodel/clusters"
  11. "github.com/kubecost/cost-model/pkg/env"
  12. "github.com/kubecost/cost-model/pkg/errors"
  13. "github.com/kubecost/cost-model/pkg/log"
  14. "github.com/kubecost/cost-model/pkg/metrics"
  15. "github.com/kubecost/cost-model/pkg/prom"
  16. "github.com/kubecost/cost-model/pkg/util"
  17. "github.com/kubecost/cost-model/pkg/util/atomic"
  18. promclient "github.com/prometheus/client_golang/api"
  19. "github.com/prometheus/client_golang/prometheus"
  20. dto "github.com/prometheus/client_model/go"
  21. v1 "k8s.io/api/core/v1"
  22. "k8s.io/klog"
  23. )
  24. //--------------------------------------------------------------------------
  25. // ClusterInfoCollector
  26. //--------------------------------------------------------------------------
  27. // ClusterInfoCollector is a prometheus collector that generates ClusterInfoMetrics
  28. type ClusterInfoCollector struct {
  29. ClusterInfo clusters.ClusterInfoProvider
  30. metricsConfig metrics.MetricsConfig
  31. }
  32. // Describe sends the super-set of all possible descriptors of metrics
  33. // collected by this Collector.
  34. func (cic ClusterInfoCollector) Describe(ch chan<- *prometheus.Desc) {
  35. disabledMetrics := cic.metricsConfig.GetDisabledMetricsMap()
  36. if _, disabled := disabledMetrics["kube_pod_annotations"]; disabled {
  37. return
  38. }
  39. ch <- prometheus.NewDesc("kubecost_cluster_info", "Kubecost Cluster Info", []string{}, nil)
  40. }
  41. // Collect is called by the Prometheus registry when collecting metrics.
  42. func (cic ClusterInfoCollector) Collect(ch chan<- prometheus.Metric) {
  43. disabledMetrics := cic.metricsConfig.GetDisabledMetricsMap()
  44. if _, disabled := disabledMetrics["kube_pod_annotations"]; disabled {
  45. return
  46. }
  47. clusterInfo := cic.ClusterInfo.GetClusterInfo()
  48. labels := prom.MapToLabels(clusterInfo)
  49. m := newClusterInfoMetric("kubecost_cluster_info", labels)
  50. ch <- m
  51. }
  52. //--------------------------------------------------------------------------
  53. // ClusterInfoMetric
  54. //--------------------------------------------------------------------------
  55. // ClusterInfoMetric is a prometheus.Metric used to encode the local cluster info
  56. type ClusterInfoMetric struct {
  57. fqName string
  58. help string
  59. labels map[string]string
  60. }
  61. // Creates a new ClusterInfoMetric, implementation of prometheus.Metric
  62. func newClusterInfoMetric(fqName string, labels map[string]string) ClusterInfoMetric {
  63. return ClusterInfoMetric{
  64. fqName: fqName,
  65. labels: labels,
  66. help: "kubecost_cluster_info ClusterInfo",
  67. }
  68. }
  69. // Desc returns the descriptor for the Metric. This method idempotently
  70. // returns the same descriptor throughout the lifetime of the Metric.
  71. func (cim ClusterInfoMetric) Desc() *prometheus.Desc {
  72. l := prometheus.Labels{}
  73. return prometheus.NewDesc(cim.fqName, cim.help, prom.LabelNamesFrom(cim.labels), l)
  74. }
  75. // Write encodes the Metric into a "Metric" Protocol Buffer data
  76. // transmission object.
  77. func (cim ClusterInfoMetric) Write(m *dto.Metric) error {
  78. h := float64(1)
  79. m.Gauge = &dto.Gauge{
  80. Value: &h,
  81. }
  82. var labels []*dto.LabelPair
  83. for k, v := range cim.labels {
  84. labels = append(labels, &dto.LabelPair{
  85. Name: toStringPtr(k),
  86. Value: toStringPtr(v),
  87. })
  88. }
  89. m.Label = labels
  90. return nil
  91. }
  92. // returns a pointer to the string provided
  93. func toStringPtr(s string) *string { return &s }
  94. //--------------------------------------------------------------------------
  95. // Cost Model Metrics Initialization
  96. //--------------------------------------------------------------------------
  97. // Only allow the metrics to be instantiated and registered once
  98. var metricsInit sync.Once
  99. var (
  100. cpuGv *prometheus.GaugeVec
  101. ramGv *prometheus.GaugeVec
  102. gpuGv *prometheus.GaugeVec
  103. gpuCountGv *prometheus.GaugeVec
  104. pvGv *prometheus.GaugeVec
  105. spotGv *prometheus.GaugeVec
  106. totalGv *prometheus.GaugeVec
  107. ramAllocGv *prometheus.GaugeVec
  108. cpuAllocGv *prometheus.GaugeVec
  109. gpuAllocGv *prometheus.GaugeVec
  110. pvAllocGv *prometheus.GaugeVec
  111. networkZoneEgressCostG prometheus.Gauge
  112. networkRegionEgressCostG prometheus.Gauge
  113. networkInternetEgressCostG prometheus.Gauge
  114. clusterManagementCostGv *prometheus.GaugeVec
  115. lbCostGv *prometheus.GaugeVec
  116. )
  117. // initCostModelMetrics uses a sync.Once to ensure that these metrics are only created once
  118. func initCostModelMetrics(clusterCache clustercache.ClusterCache, provider cloud.Provider, clusterInfo clusters.ClusterInfoProvider, metricsConfig *metrics.MetricsConfig) {
  119. disabledMetrics := metricsConfig.GetDisabledMetricsMap()
  120. var toRegisterGV []*prometheus.GaugeVec
  121. var toRegisterGauge []prometheus.Gauge
  122. metricsInit.Do(func() {
  123. if _, disabled := disabledMetrics["node_cpu_hourly_cost"]; !disabled {
  124. cpuGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  125. Name: "node_cpu_hourly_cost",
  126. Help: "node_cpu_hourly_cost hourly cost for each cpu on this node",
  127. }, []string{"instance", "node", "instance_type", "region", "provider_id"})
  128. toRegisterGV = append(toRegisterGV, cpuGv)
  129. }
  130. if _, disabled := disabledMetrics["node_ram_hourly_cost"]; !disabled {
  131. ramGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  132. Name: "node_ram_hourly_cost",
  133. Help: "node_ram_hourly_cost hourly cost for each gb of ram on this node",
  134. }, []string{"instance", "node", "instance_type", "region", "provider_id"})
  135. toRegisterGV = append(toRegisterGV, ramGv)
  136. }
  137. if _, disabled := disabledMetrics["node_gpu_hourly_cost"]; !disabled {
  138. gpuGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  139. Name: "node_gpu_hourly_cost",
  140. Help: "node_gpu_hourly_cost hourly cost for each gpu on this node",
  141. }, []string{"instance", "node", "instance_type", "region", "provider_id"})
  142. toRegisterGV = append(toRegisterGV, gpuGv)
  143. }
  144. if _, disabled := disabledMetrics["node_gpu_count"]; !disabled {
  145. gpuCountGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  146. Name: "node_gpu_count",
  147. Help: "node_gpu_count count of gpu on this node",
  148. }, []string{"instance", "node", "instance_type", "region", "provider_id"})
  149. toRegisterGV = append(toRegisterGV, gpuCountGv)
  150. }
  151. if _, disabled := disabledMetrics["pv_hourly_cost"]; !disabled {
  152. pvGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  153. Name: "pv_hourly_cost",
  154. Help: "pv_hourly_cost Cost per GB per hour on a persistent disk",
  155. }, []string{"volumename", "persistentvolume", "provider_id"})
  156. toRegisterGV = append(toRegisterGV, pvGv)
  157. }
  158. if _, disabled := disabledMetrics["kubecost_node_is_spot"]; !disabled {
  159. spotGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  160. Name: "kubecost_node_is_spot",
  161. Help: "kubecost_node_is_spot Cloud provider info about node preemptibility",
  162. }, []string{"instance", "node", "instance_type", "region", "provider_id"})
  163. toRegisterGV = append(toRegisterGV, spotGv)
  164. }
  165. if _, disabled := disabledMetrics["node_total_hourly_cost"]; !disabled {
  166. totalGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  167. Name: "node_total_hourly_cost",
  168. Help: "node_total_hourly_cost Total node cost per hour",
  169. }, []string{"instance", "node", "instance_type", "region", "provider_id"})
  170. toRegisterGV = append(toRegisterGV, totalGv)
  171. }
  172. if _, disabled := disabledMetrics["container_memory_allocation_bytes"]; !disabled {
  173. ramAllocGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  174. Name: "container_memory_allocation_bytes",
  175. Help: "container_memory_allocation_bytes Bytes of RAM used",
  176. }, []string{"namespace", "pod", "container", "instance", "node"})
  177. toRegisterGV = append(toRegisterGV, ramAllocGv)
  178. }
  179. if _, disabled := disabledMetrics["container_cpu_allocation"]; !disabled {
  180. cpuAllocGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  181. Name: "container_cpu_allocation",
  182. Help: "container_cpu_allocation Percent of a single CPU used in a minute",
  183. }, []string{"namespace", "pod", "container", "instance", "node"})
  184. toRegisterGV = append(toRegisterGV, cpuAllocGv)
  185. }
  186. if _, disabled := disabledMetrics["container_gpu_allocation"]; !disabled {
  187. gpuAllocGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  188. Name: "container_gpu_allocation",
  189. Help: "container_gpu_allocation GPU used",
  190. }, []string{"namespace", "pod", "container", "instance", "node"})
  191. toRegisterGV = append(toRegisterGV, gpuAllocGv)
  192. }
  193. if _, disabled := disabledMetrics["pod_pvc_allocation"]; !disabled {
  194. pvAllocGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  195. Name: "pod_pvc_allocation",
  196. Help: "pod_pvc_allocation Bytes used by a PVC attached to a pod",
  197. }, []string{"namespace", "pod", "persistentvolumeclaim", "persistentvolume"})
  198. toRegisterGV = append(toRegisterGV, pvAllocGv)
  199. }
  200. if _, disabled := disabledMetrics["kubecost_network_zone_egress_cost"]; !disabled {
  201. networkZoneEgressCostG = prometheus.NewGauge(prometheus.GaugeOpts{
  202. Name: "kubecost_network_zone_egress_cost",
  203. Help: "kubecost_network_zone_egress_cost Total cost per GB egress across zones",
  204. })
  205. toRegisterGauge = append(toRegisterGauge, networkZoneEgressCostG)
  206. }
  207. if _, disabled := disabledMetrics["kubecost_network_region_egress_cost"]; !disabled {
  208. networkRegionEgressCostG = prometheus.NewGauge(prometheus.GaugeOpts{
  209. Name: "kubecost_network_region_egress_cost",
  210. Help: "kubecost_network_region_egress_cost Total cost per GB egress across regions",
  211. })
  212. toRegisterGauge = append(toRegisterGauge, networkRegionEgressCostG)
  213. }
  214. if _, disabled := disabledMetrics["kubecost_network_internet_egress_cost"]; !disabled {
  215. networkInternetEgressCostG = prometheus.NewGauge(prometheus.GaugeOpts{
  216. Name: "kubecost_network_internet_egress_cost",
  217. Help: "kubecost_network_internet_egress_cost Total cost per GB of internet egress.",
  218. })
  219. toRegisterGauge = append(toRegisterGauge, networkInternetEgressCostG)
  220. }
  221. if _, disabled := disabledMetrics["kubecost_cluster_management_cost"]; !disabled {
  222. clusterManagementCostGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{
  223. Name: "kubecost_cluster_management_cost",
  224. Help: "kubecost_cluster_management_cost Hourly cost paid as a cluster management fee.",
  225. }, []string{"provisioner_name"})
  226. toRegisterGV = append(toRegisterGV, clusterManagementCostGv)
  227. }
  228. if _, disabled := disabledMetrics["kubecost_load_balancer_cost"]; !disabled {
  229. lbCostGv = prometheus.NewGaugeVec(prometheus.GaugeOpts{ // no differentiation between ELB and ALB right now
  230. Name: "kubecost_load_balancer_cost",
  231. Help: "kubecost_load_balancer_cost Hourly cost of load balancer",
  232. }, []string{"ingress_ip", "namespace", "service_name"}) // assumes one ingress IP per load balancer
  233. toRegisterGV = append(toRegisterGV, lbCostGv)
  234. }
  235. // Register cost-model metrics for emission
  236. for _, gv := range toRegisterGV {
  237. prometheus.MustRegister(gv)
  238. }
  239. for _, g := range toRegisterGauge {
  240. prometheus.MustRegister(g)
  241. }
  242. // General Metric Collectors
  243. prometheus.MustRegister(ClusterInfoCollector{
  244. ClusterInfo: clusterInfo,
  245. metricsConfig: *metricsConfig,
  246. })
  247. })
  248. }
  249. //--------------------------------------------------------------------------
  250. // CostModelMetricsEmitter
  251. //--------------------------------------------------------------------------
  252. // CostModelMetricsEmitter emits all cost-model specific metrics calculated by
  253. // the CostModel.ComputeCostData() method.
  254. type CostModelMetricsEmitter struct {
  255. PrometheusClient promclient.Client
  256. KubeClusterCache clustercache.ClusterCache
  257. CloudProvider cloud.Provider
  258. Model *CostModel
  259. // Metrics
  260. CPUPriceRecorder *prometheus.GaugeVec
  261. RAMPriceRecorder *prometheus.GaugeVec
  262. PersistentVolumePriceRecorder *prometheus.GaugeVec
  263. GPUPriceRecorder *prometheus.GaugeVec
  264. GPUCountRecorder *prometheus.GaugeVec
  265. PVAllocationRecorder *prometheus.GaugeVec
  266. NodeSpotRecorder *prometheus.GaugeVec
  267. NodeTotalPriceRecorder *prometheus.GaugeVec
  268. RAMAllocationRecorder *prometheus.GaugeVec
  269. CPUAllocationRecorder *prometheus.GaugeVec
  270. GPUAllocationRecorder *prometheus.GaugeVec
  271. ClusterManagementCostRecorder *prometheus.GaugeVec
  272. LBCostRecorder *prometheus.GaugeVec
  273. NetworkZoneEgressRecorder prometheus.Gauge
  274. NetworkRegionEgressRecorder prometheus.Gauge
  275. NetworkInternetEgressRecorder prometheus.Gauge
  276. // Concurrent Flow Control - Manages the run state of the metric emitter
  277. runState atomic.AtomicRunState
  278. }
  279. // NewCostModelMetricsEmitter creates a new cost-model metrics emitter. Use Start() to begin metric emission.
  280. func NewCostModelMetricsEmitter(promClient promclient.Client, clusterCache clustercache.ClusterCache, provider cloud.Provider, clusterInfo clusters.ClusterInfoProvider, model *CostModel) *CostModelMetricsEmitter {
  281. // Get metric configurations, if any
  282. metricsConfig, err := metrics.GetMetricsConfig()
  283. if err != nil {
  284. log.Infof("Failed to get metrics config before init: %s", err)
  285. }
  286. if len(metricsConfig.DisabledMetrics) > 0 {
  287. log.Infof("Starting metrics init with disabled metrics: %v", metricsConfig.DisabledMetrics)
  288. }
  289. // init will only actually execute once to register the custom gauges
  290. initCostModelMetrics(clusterCache, provider, clusterInfo, metricsConfig)
  291. metrics.InitKubeMetrics(clusterCache, metricsConfig, &metrics.KubeMetricsOpts{
  292. EmitKubecostControllerMetrics: true,
  293. EmitNamespaceAnnotations: env.IsEmitNamespaceAnnotationsMetric(),
  294. EmitPodAnnotations: env.IsEmitPodAnnotationsMetric(),
  295. EmitKubeStateMetrics: env.IsEmitKsmV1Metrics(),
  296. EmitKubeStateMetricsV1Only: env.IsEmitKsmV1MetricsOnly(),
  297. })
  298. return &CostModelMetricsEmitter{
  299. PrometheusClient: promClient,
  300. KubeClusterCache: clusterCache,
  301. CloudProvider: provider,
  302. Model: model,
  303. CPUPriceRecorder: cpuGv,
  304. RAMPriceRecorder: ramGv,
  305. GPUPriceRecorder: gpuGv,
  306. GPUCountRecorder: gpuCountGv,
  307. PersistentVolumePriceRecorder: pvGv,
  308. NodeSpotRecorder: spotGv,
  309. NodeTotalPriceRecorder: totalGv,
  310. RAMAllocationRecorder: ramAllocGv,
  311. CPUAllocationRecorder: cpuAllocGv,
  312. GPUAllocationRecorder: gpuAllocGv,
  313. PVAllocationRecorder: pvAllocGv,
  314. NetworkZoneEgressRecorder: networkZoneEgressCostG,
  315. NetworkRegionEgressRecorder: networkRegionEgressCostG,
  316. NetworkInternetEgressRecorder: networkInternetEgressCostG,
  317. ClusterManagementCostRecorder: clusterManagementCostGv,
  318. LBCostRecorder: lbCostGv,
  319. }
  320. }
  321. // IsRunning returns true if metric recording is running.
  322. func (cmme *CostModelMetricsEmitter) IsRunning() bool {
  323. return cmme.runState.IsRunning()
  324. }
  325. // NodeCostAverages tracks a running average of a node's cost attributes.
  326. // The averages are used to detect and discard spurrious outliers.
  327. type NodeCostAverages struct {
  328. CpuCostAverage float64
  329. RamCostAverage float64
  330. NumCpuDataPoints float64
  331. NumRamDataPoints float64
  332. }
  333. // StartCostModelMetricRecording starts the go routine that emits metrics used to determine
  334. // cluster costs.
  335. func (cmme *CostModelMetricsEmitter) Start() bool {
  336. // wait for a reset to prevent a race between start and stop calls
  337. cmme.runState.WaitForReset()
  338. // Check to see if we're already recording, and atomically advance the run state to start if we're not
  339. if !cmme.runState.Start() {
  340. log.Errorf("Attempted to start cost model metric recording when it's already running.")
  341. return false
  342. }
  343. go func() {
  344. defer errors.HandlePanic()
  345. containerSeen := make(map[string]bool)
  346. nodeSeen := make(map[string]bool)
  347. loadBalancerSeen := make(map[string]bool)
  348. pvSeen := make(map[string]bool)
  349. pvcSeen := make(map[string]bool)
  350. nodeCostAverages := make(map[string]NodeCostAverages)
  351. getKeyFromLabelStrings := func(labels ...string) string {
  352. return strings.Join(labels, ",")
  353. }
  354. getLabelStringsFromKey := func(key string) []string {
  355. return strings.Split(key, ",")
  356. }
  357. var defaultRegion string = ""
  358. nodeList := cmme.KubeClusterCache.GetAllNodes()
  359. if len(nodeList) > 0 {
  360. var ok bool
  361. defaultRegion, ok = util.GetRegion(nodeList[0].Labels)
  362. if !ok {
  363. log.DedupedWarningf(5, "Failed to locate default region")
  364. }
  365. }
  366. for {
  367. klog.V(4).Info("Recording prices...")
  368. podlist := cmme.KubeClusterCache.GetAllPods()
  369. podStatus := make(map[string]v1.PodPhase)
  370. for _, pod := range podlist {
  371. podStatus[pod.Name] = pod.Status.Phase
  372. }
  373. cfg, _ := cmme.CloudProvider.GetConfig()
  374. provisioner, clusterManagementCost, err := cmme.CloudProvider.ClusterManagementPricing()
  375. if err != nil {
  376. klog.V(1).Infof("Error getting cluster management cost %s", err.Error())
  377. }
  378. cmme.ClusterManagementCostRecorder.WithLabelValues(provisioner).Set(clusterManagementCost)
  379. // Record network pricing at global scope
  380. networkCosts, err := cmme.CloudProvider.NetworkPricing()
  381. if err != nil {
  382. klog.V(4).Infof("Failed to retrieve network costs: %s", err.Error())
  383. } else {
  384. cmme.NetworkZoneEgressRecorder.Set(networkCosts.ZoneNetworkEgressCost)
  385. cmme.NetworkRegionEgressRecorder.Set(networkCosts.RegionNetworkEgressCost)
  386. cmme.NetworkInternetEgressRecorder.Set(networkCosts.InternetNetworkEgressCost)
  387. }
  388. // TODO: Pass PrometheusClient and CloudProvider into CostModel on instantiation so this isn't so awkward
  389. data, err := cmme.Model.ComputeCostData(cmme.PrometheusClient, cmme.CloudProvider, "2m", "", "")
  390. if err != nil {
  391. // For an error collection, we'll just log the length of the errors (ComputeCostData already logs the
  392. // actual errors)
  393. if prom.IsErrorCollection(err) {
  394. if ec, ok := err.(prom.QueryErrorCollection); ok {
  395. log.Errorf("Error in price recording: %d errors occurred", len(ec.Errors()))
  396. }
  397. } else {
  398. log.Errorf("Error in price recording: " + err.Error())
  399. }
  400. // zero the for loop so the time.Sleep will still work
  401. data = map[string]*CostData{}
  402. }
  403. // TODO: Pass CloudProvider into CostModel on instantiation so this isn't so awkward
  404. nodes, err := cmme.Model.GetNodeCost(cmme.CloudProvider)
  405. if err != nil {
  406. log.Warningf("Metric emission: error getting Node cost: %s", err)
  407. }
  408. for nodeName, node := range nodes {
  409. // Emit costs, guarding against NaN inputs for custom pricing.
  410. cpuCost, _ := strconv.ParseFloat(node.VCPUCost, 64)
  411. if math.IsNaN(cpuCost) || math.IsInf(cpuCost, 0) {
  412. cpuCost, _ = strconv.ParseFloat(cfg.CPU, 64)
  413. if math.IsNaN(cpuCost) || math.IsInf(cpuCost, 0) {
  414. cpuCost = 0
  415. }
  416. }
  417. cpu, _ := strconv.ParseFloat(node.VCPU, 64)
  418. if math.IsNaN(cpu) || math.IsInf(cpu, 0) {
  419. cpu = 1 // Assume 1 CPU
  420. }
  421. ramCost, _ := strconv.ParseFloat(node.RAMCost, 64)
  422. if math.IsNaN(ramCost) || math.IsInf(ramCost, 0) {
  423. ramCost, _ = strconv.ParseFloat(cfg.RAM, 64)
  424. if math.IsNaN(ramCost) || math.IsInf(ramCost, 0) {
  425. ramCost = 0
  426. }
  427. }
  428. ram, _ := strconv.ParseFloat(node.RAMBytes, 64)
  429. if math.IsNaN(ram) || math.IsInf(ram, 0) {
  430. ram = 0
  431. }
  432. gpu, _ := strconv.ParseFloat(node.GPU, 64)
  433. if math.IsNaN(gpu) || math.IsInf(gpu, 0) {
  434. gpu = 0
  435. }
  436. gpuCost, _ := strconv.ParseFloat(node.GPUCost, 64)
  437. if math.IsNaN(gpuCost) || math.IsInf(gpuCost, 0) {
  438. gpuCost, _ = strconv.ParseFloat(cfg.GPU, 64)
  439. if math.IsNaN(gpuCost) || math.IsInf(gpuCost, 0) {
  440. gpuCost = 0
  441. }
  442. }
  443. nodeType := node.InstanceType
  444. nodeRegion := node.Region
  445. totalCost := cpu*cpuCost + ramCost*(ram/1024/1024/1024) + gpu*gpuCost
  446. labelKey := getKeyFromLabelStrings(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID)
  447. avgCosts, ok := nodeCostAverages[labelKey]
  448. // initialize average cost tracking for this node if there is none
  449. if !ok {
  450. avgCosts = NodeCostAverages{
  451. CpuCostAverage: cpuCost,
  452. RamCostAverage: ramCost,
  453. NumCpuDataPoints: 1,
  454. NumRamDataPoints: 1,
  455. }
  456. nodeCostAverages[labelKey] = avgCosts
  457. }
  458. cmme.GPUCountRecorder.WithLabelValues(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID).Set(gpu)
  459. cmme.GPUPriceRecorder.WithLabelValues(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID).Set(gpuCost)
  460. const outlierFactor float64 = 30
  461. // don't record cpuCost, ramCost, or gpuCost in the case of wild outliers
  462. // k8s api sometimes causes cost spikes as described here:
  463. // https://github.com/kubecost/cost-model/issues/927
  464. if cpuCost < outlierFactor*avgCosts.CpuCostAverage {
  465. cmme.CPUPriceRecorder.WithLabelValues(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID).Set(cpuCost)
  466. avgCosts.CpuCostAverage = (avgCosts.CpuCostAverage*avgCosts.NumCpuDataPoints + cpuCost) / (avgCosts.NumCpuDataPoints + 1)
  467. avgCosts.NumCpuDataPoints += 1
  468. } else {
  469. log.Warningf("CPU cost outlier detected; skipping data point.")
  470. }
  471. if ramCost < outlierFactor*avgCosts.RamCostAverage {
  472. cmme.RAMPriceRecorder.WithLabelValues(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID).Set(ramCost)
  473. avgCosts.RamCostAverage = (avgCosts.RamCostAverage*avgCosts.NumRamDataPoints + ramCost) / (avgCosts.NumRamDataPoints + 1)
  474. avgCosts.NumRamDataPoints += 1
  475. } else {
  476. log.Warningf("RAM cost outlier detected; skipping data point.")
  477. }
  478. // skip redording totalCost if any constituent costs were outliers
  479. if cpuCost < outlierFactor*avgCosts.CpuCostAverage &&
  480. ramCost < outlierFactor*avgCosts.RamCostAverage {
  481. cmme.NodeTotalPriceRecorder.WithLabelValues(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID).Set(totalCost)
  482. }
  483. nodeCostAverages[labelKey] = avgCosts
  484. if node.IsSpot() {
  485. cmme.NodeSpotRecorder.WithLabelValues(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID).Set(1.0)
  486. } else {
  487. cmme.NodeSpotRecorder.WithLabelValues(nodeName, nodeName, nodeType, nodeRegion, node.ProviderID).Set(0.0)
  488. }
  489. nodeSeen[labelKey] = true
  490. }
  491. // TODO: Pass CloudProvider into CostModel on instantiation so this isn't so awkward
  492. loadBalancers, err := cmme.Model.GetLBCost(cmme.CloudProvider)
  493. if err != nil {
  494. log.Warningf("Metric emission: error getting LoadBalancer cost: %s", err)
  495. }
  496. for lbKey, lb := range loadBalancers {
  497. // TODO: parse (if necessary) and calculate cost associated with loadBalancer based on dynamic cloud prices fetched into each lb struct on GetLBCost() call
  498. keyParts := getLabelStringsFromKey(lbKey)
  499. namespace := keyParts[0]
  500. serviceName := keyParts[1]
  501. ingressIP := ""
  502. if len(lb.IngressIPAddresses) > 0 {
  503. ingressIP = lb.IngressIPAddresses[0] // assumes one ingress IP per load balancer
  504. }
  505. cmme.LBCostRecorder.WithLabelValues(ingressIP, namespace, serviceName).Set(lb.Cost)
  506. labelKey := getKeyFromLabelStrings(ingressIP, namespace, serviceName)
  507. loadBalancerSeen[labelKey] = true
  508. }
  509. for _, costs := range data {
  510. nodeName := costs.NodeName
  511. namespace := costs.Namespace
  512. podName := costs.PodName
  513. containerName := costs.Name
  514. if costs.PVCData != nil {
  515. for _, pvc := range costs.PVCData {
  516. if pvc.Volume != nil {
  517. timesClaimed := pvc.TimesClaimed
  518. if timesClaimed == 0 {
  519. timesClaimed = 1 // unallocated PVs are unclaimed but have a full allocation
  520. }
  521. cmme.PVAllocationRecorder.WithLabelValues(namespace, podName, pvc.Claim, pvc.VolumeName).Set(pvc.Values[0].Value / float64(timesClaimed))
  522. labelKey := getKeyFromLabelStrings(namespace, podName, pvc.Claim, pvc.VolumeName)
  523. pvcSeen[labelKey] = true
  524. }
  525. }
  526. }
  527. if len(costs.RAMAllocation) > 0 {
  528. cmme.RAMAllocationRecorder.WithLabelValues(namespace, podName, containerName, nodeName, nodeName).Set(costs.RAMAllocation[0].Value)
  529. }
  530. if len(costs.CPUAllocation) > 0 {
  531. cmme.CPUAllocationRecorder.WithLabelValues(namespace, podName, containerName, nodeName, nodeName).Set(costs.CPUAllocation[0].Value)
  532. }
  533. if len(costs.GPUReq) > 0 {
  534. // allocation here is set to the request because shared GPU usage not yet supported.
  535. cmme.GPUAllocationRecorder.WithLabelValues(namespace, podName, containerName, nodeName, nodeName).Set(costs.GPUReq[0].Value)
  536. }
  537. labelKey := getKeyFromLabelStrings(namespace, podName, containerName, nodeName, nodeName)
  538. if podStatus[podName] == v1.PodRunning { // Only report data for current pods
  539. containerSeen[labelKey] = true
  540. } else {
  541. containerSeen[labelKey] = false
  542. }
  543. }
  544. storageClasses := cmme.KubeClusterCache.GetAllStorageClasses()
  545. storageClassMap := make(map[string]map[string]string)
  546. for _, storageClass := range storageClasses {
  547. params := storageClass.Parameters
  548. storageClassMap[storageClass.ObjectMeta.Name] = params
  549. if storageClass.GetAnnotations()["storageclass.kubernetes.io/is-default-class"] == "true" || storageClass.GetAnnotations()["storageclass.beta.kubernetes.io/is-default-class"] == "true" {
  550. storageClassMap["default"] = params
  551. storageClassMap[""] = params
  552. }
  553. }
  554. pvs := cmme.KubeClusterCache.GetAllPersistentVolumes()
  555. for _, pv := range pvs {
  556. // Omit pv_hourly_cost if the volume status is failed
  557. if pv.Status.Phase == v1.VolumeFailed {
  558. continue
  559. }
  560. parameters, ok := storageClassMap[pv.Spec.StorageClassName]
  561. if !ok {
  562. klog.V(4).Infof("Unable to find parameters for storage class \"%s\". Does pv \"%s\" have a storageClassName?", pv.Spec.StorageClassName, pv.Name)
  563. }
  564. var region string
  565. if r, ok := util.GetRegion(pv.Labels); ok {
  566. region = r
  567. } else {
  568. region = defaultRegion
  569. }
  570. cacPv := &cloud.PV{
  571. Class: pv.Spec.StorageClassName,
  572. Region: region,
  573. Parameters: parameters,
  574. }
  575. // TODO: GetPVCost should be a method in CostModel?
  576. GetPVCost(cacPv, pv, cmme.CloudProvider, region)
  577. c, _ := strconv.ParseFloat(cacPv.Cost, 64)
  578. cmme.PersistentVolumePriceRecorder.WithLabelValues(pv.Name, pv.Name, cacPv.ProviderID).Set(c)
  579. labelKey := getKeyFromLabelStrings(pv.Name, pv.Name)
  580. pvSeen[labelKey] = true
  581. }
  582. for labelString, seen := range nodeSeen {
  583. if !seen {
  584. klog.V(4).Infof("Removing %s from nodes", labelString)
  585. labels := getLabelStringsFromKey(labelString)
  586. ok := cmme.NodeTotalPriceRecorder.DeleteLabelValues(labels...)
  587. if ok {
  588. klog.V(4).Infof("removed %s from totalprice", labelString)
  589. } else {
  590. klog.Infof("FAILURE TO REMOVE %s from totalprice", labelString)
  591. }
  592. ok = cmme.NodeSpotRecorder.DeleteLabelValues(labels...)
  593. if ok {
  594. klog.V(4).Infof("removed %s from spot records", labelString)
  595. } else {
  596. klog.Infof("FAILURE TO REMOVE %s from spot records", labelString)
  597. }
  598. ok = cmme.CPUPriceRecorder.DeleteLabelValues(labels...)
  599. if ok {
  600. klog.V(4).Infof("removed %s from cpuprice", labelString)
  601. } else {
  602. klog.Infof("FAILURE TO REMOVE %s from cpuprice", labelString)
  603. }
  604. ok = cmme.GPUPriceRecorder.DeleteLabelValues(labels...)
  605. if ok {
  606. klog.V(4).Infof("removed %s from gpuprice", labelString)
  607. } else {
  608. klog.Infof("FAILURE TO REMOVE %s from gpuprice", labelString)
  609. }
  610. ok = cmme.GPUCountRecorder.DeleteLabelValues(labels...)
  611. if ok {
  612. klog.V(4).Infof("removed %s from gpucount", labelString)
  613. } else {
  614. klog.Infof("FAILURE TO REMOVE %s from gpucount", labelString)
  615. }
  616. ok = cmme.RAMPriceRecorder.DeleteLabelValues(labels...)
  617. if ok {
  618. klog.V(4).Infof("removed %s from ramprice", labelString)
  619. } else {
  620. klog.Infof("FAILURE TO REMOVE %s from ramprice", labelString)
  621. }
  622. delete(nodeSeen, labelString)
  623. delete(nodeCostAverages, labelString)
  624. } else {
  625. nodeSeen[labelString] = false
  626. }
  627. }
  628. for labelString, seen := range loadBalancerSeen {
  629. if !seen {
  630. labels := getLabelStringsFromKey(labelString)
  631. ok := cmme.LBCostRecorder.DeleteLabelValues(labels...)
  632. if !ok {
  633. log.Warningf("Metric emission: failed to delete LoadBalancer with labels: %v", labels)
  634. }
  635. delete(loadBalancerSeen, labelString)
  636. } else {
  637. loadBalancerSeen[labelString] = false
  638. }
  639. }
  640. for labelString, seen := range containerSeen {
  641. if !seen {
  642. labels := getLabelStringsFromKey(labelString)
  643. cmme.RAMAllocationRecorder.DeleteLabelValues(labels...)
  644. cmme.CPUAllocationRecorder.DeleteLabelValues(labels...)
  645. cmme.GPUAllocationRecorder.DeleteLabelValues(labels...)
  646. delete(containerSeen, labelString)
  647. } else {
  648. containerSeen[labelString] = false
  649. }
  650. }
  651. for labelString, seen := range pvSeen {
  652. if !seen {
  653. labels := getLabelStringsFromKey(labelString)
  654. cmme.PersistentVolumePriceRecorder.DeleteLabelValues(labels...)
  655. delete(pvSeen, labelString)
  656. } else {
  657. pvSeen[labelString] = false
  658. }
  659. }
  660. for labelString, seen := range pvcSeen {
  661. if !seen {
  662. labels := getLabelStringsFromKey(labelString)
  663. cmme.PVAllocationRecorder.DeleteLabelValues(labels...)
  664. delete(pvcSeen, labelString)
  665. } else {
  666. pvcSeen[labelString] = false
  667. }
  668. }
  669. select {
  670. case <-time.After(time.Minute):
  671. case <-cmme.runState.OnStop():
  672. cmme.runState.Reset()
  673. return
  674. }
  675. }
  676. }()
  677. return true
  678. }
  679. // Stop halts the metrics emission loop after the current emission is completed
  680. // or if the emission is paused.
  681. func (cmme *CostModelMetricsEmitter) Stop() {
  682. cmme.runState.Stop()
  683. }