metricsquerier.go 94 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921
  1. package prom
  2. import (
  3. "fmt"
  4. "time"
  5. "github.com/opencost/opencost/core/pkg/log"
  6. "github.com/opencost/opencost/core/pkg/source"
  7. "github.com/opencost/opencost/core/pkg/util/timeutil"
  8. prometheus "github.com/prometheus/client_golang/api"
  9. )
  10. //--------------------------------------------------------------------------
  11. // PrometheusMetricsQuerier
  12. //--------------------------------------------------------------------------
  13. // PrometheusMetricsQueryLogFormat is the log format used to log metric queries before being sent to the prometheus
  14. // instance
  15. const PrometheusMetricsQueryLogFormat = `[PrometheusMetricsQuerier][%s][At Time: %d]: %s`
  16. // PrometheusMetricsQuerier is the implementation of the data source's MetricsQuerier interface for Prometheus.
  17. type PrometheusMetricsQuerier struct {
  18. promConfig *OpenCostPrometheusConfig
  19. promClient prometheus.Client
  20. promContexts *ContextFactory
  21. }
  22. func newPrometheusMetricsQuerier(
  23. promConfig *OpenCostPrometheusConfig,
  24. promClient prometheus.Client,
  25. promContexts *ContextFactory,
  26. ) *PrometheusMetricsQuerier {
  27. return &PrometheusMetricsQuerier{
  28. promConfig: promConfig,
  29. promClient: promClient,
  30. promContexts: promContexts,
  31. }
  32. }
  33. func (pds *PrometheusMetricsQuerier) QueryPVPricePerGiBHour(start, end time.Time) *source.Future[source.PVPricePerGiBHourResult] {
  34. const queryName = "QueryPVPricePerGiBHour"
  35. const pvCostQuery = `avg(avg_over_time(pv_hourly_cost{%s}[%s])) by (%s, persistentvolume, volumename, uid, provider_id)`
  36. cfg := pds.promConfig
  37. durStr := timeutil.DurationString(end.Sub(start))
  38. if durStr == "" {
  39. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  40. }
  41. queryPVCost := fmt.Sprintf(pvCostQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  42. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVCost)
  43. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  44. return source.NewFuture(source.DecodePVPricePerGiBHourResult, ctx.QueryAtTime(queryPVCost, end))
  45. }
  46. func (pds *PrometheusMetricsQuerier) QueryPVUsedAverage(start, end time.Time) *source.Future[source.PVUsedAvgResult] {
  47. const queryName = "QueryPVUsedAverage"
  48. const pvUsedAverageQuery = `avg(avg_over_time(kubelet_volume_stats_used_bytes{%s}[%s])) by (%s, persistentvolumeclaim, namespace, uid)`
  49. cfg := pds.promConfig
  50. durStr := timeutil.DurationString(end.Sub(start))
  51. if durStr == "" {
  52. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  53. }
  54. queryPVUsedAvg := fmt.Sprintf(pvUsedAverageQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  55. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVUsedAvg)
  56. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  57. return source.NewFuture(source.DecodePVUsedAvgResult, ctx.QueryAtTime(queryPVUsedAvg, end))
  58. }
  59. func (pds *PrometheusMetricsQuerier) QueryPVUsedMax(start, end time.Time) *source.Future[source.PVUsedMaxResult] {
  60. const queryName = "QueryPVUsedMax"
  61. const pvUsedMaxQuery = `max(max_over_time(kubelet_volume_stats_used_bytes{%s}[%s])) by (%s, persistentvolumeclaim, namespace, uid)`
  62. cfg := pds.promConfig
  63. durStr := timeutil.DurationString(end.Sub(start))
  64. if durStr == "" {
  65. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  66. }
  67. queryPVUsedMax := fmt.Sprintf(pvUsedMaxQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  68. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVUsedMax)
  69. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  70. return source.NewFuture(source.DecodePVUsedMaxResult, ctx.QueryAtTime(queryPVUsedMax, end))
  71. }
  72. func (pds *PrometheusMetricsQuerier) QueryPVCInfo(start, end time.Time) *source.Future[source.PVCInfoResult] {
  73. const queryName = "QueryPVCInfo"
  74. const queryFmtPVCInfo = `avg(kube_persistentvolumeclaim_info{volumename != "", %s}) by (persistentvolumeclaim, storageclass, volumename, namespace, uid, %s)[%s:%dm]`
  75. cfg := pds.promConfig
  76. minsPerResolution := cfg.DataResolutionMinutes
  77. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  78. if durStr == "" {
  79. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  80. }
  81. queryPVCInfo := fmt.Sprintf(queryFmtPVCInfo, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  82. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVCInfo)
  83. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  84. return source.NewFuture(source.DecodePVCInfoResult, ctx.QueryAtTime(queryPVCInfo, end))
  85. }
  86. func (pds *PrometheusMetricsQuerier) QueryPVActiveMinutes(start, end time.Time) *source.Future[source.PVActiveMinutesResult] {
  87. const queryName = "QueryPVActiveMinutes"
  88. const pvActiveMinsQuery = `avg(kube_persistentvolume_capacity_bytes{%s}) by (%s, persistentvolume, uid)[%s:%dm]`
  89. cfg := pds.promConfig
  90. minsPerResolution := cfg.DataResolutionMinutes
  91. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  92. if durStr == "" {
  93. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  94. }
  95. queryPVActiveMins := fmt.Sprintf(pvActiveMinsQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  96. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVActiveMins)
  97. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  98. return source.NewFuture(source.DecodePVActiveMinutesResult, ctx.QueryAtTime(queryPVActiveMins, end))
  99. }
  100. func (pds *PrometheusMetricsQuerier) QueryLocalStorageCost(start, end time.Time) *source.Future[source.LocalStorageCostResult] {
  101. const queryName = "QueryLocalStorageCost"
  102. const localStorageCostQuery = `sum_over_time(sum(container_fs_limit_bytes{device=~"/dev/(nvme|sda).*", id="/", %s}) by (instance, device, uid, %s)[%s:%dm]) / 1024 / 1024 / 1024 * %f * %f`
  103. cfg := pds.promConfig
  104. minsPerResolution := cfg.DataResolutionMinutes
  105. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  106. if durStr == "" {
  107. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  108. }
  109. // hourlyToCumulative is a scaling factor that, when multiplied by an
  110. // hourly value, converts it to a cumulative value; i.e. [$/hr] *
  111. // [min/res]*[hr/min] = [$/res]
  112. hourlyToCumulative := float64(minsPerResolution) * (1.0 / 60.0)
  113. costPerGBHr := 0.04 / 730.0
  114. queryLocalStorageCost := fmt.Sprintf(localStorageCostQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution, hourlyToCumulative, costPerGBHr)
  115. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLocalStorageCost)
  116. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  117. return source.NewFuture(source.DecodeLocalStorageCostResult, ctx.QueryAtTime(queryLocalStorageCost, end))
  118. }
  119. func (pds *PrometheusMetricsQuerier) QueryLocalStorageUsedCost(start, end time.Time) *source.Future[source.LocalStorageUsedCostResult] {
  120. const queryName = "QueryLocalStorageUsedCost"
  121. const localStorageUsedCostQuery = `sum_over_time(sum(container_fs_usage_bytes{device=~"/dev/(nvme|sda).*", id="/", %s}) by (instance, device, uid, %s)[%s:%dm]) / 1024 / 1024 / 1024 * %f * %f`
  122. cfg := pds.promConfig
  123. minsPerResolution := cfg.DataResolutionMinutes
  124. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  125. if durStr == "" {
  126. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  127. }
  128. // hourlyToCumulative is a scaling factor that, when multiplied by an
  129. // hourly value, converts it to a cumulative value; i.e. [$/hr] *
  130. // [min/res]*[hr/min] = [$/res]
  131. hourlyToCumulative := float64(minsPerResolution) * (1.0 / 60.0)
  132. costPerGBHr := 0.04 / 730.0
  133. queryLocalStorageUsedCost := fmt.Sprintf(localStorageUsedCostQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution, hourlyToCumulative, costPerGBHr)
  134. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLocalStorageUsedCost)
  135. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  136. return source.NewFuture(source.DecodeLocalStorageUsedCostResult, ctx.QueryAtTime(queryLocalStorageUsedCost, end))
  137. }
  138. func (pds *PrometheusMetricsQuerier) QueryLocalStorageUsedAvg(start, end time.Time) *source.Future[source.LocalStorageUsedAvgResult] {
  139. const queryName = "QueryLocalStorageUsedAvg"
  140. const localStorageUsedAvgQuery = `avg(sum(avg_over_time(container_fs_usage_bytes{device=~"/dev/(nvme|sda).*", id="/", %s}[%s])) by (instance, device, uid, %s, job)) by (instance, device, uid, %s)`
  141. cfg := pds.promConfig
  142. durStr := timeutil.DurationString(end.Sub(start))
  143. if durStr == "" {
  144. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  145. }
  146. queryLocalStorageUsedAvg := fmt.Sprintf(localStorageUsedAvgQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel, cfg.ClusterLabel)
  147. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLocalStorageUsedAvg)
  148. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  149. return source.NewFuture(source.DecodeLocalStorageUsedAvgResult, ctx.QueryAtTime(queryLocalStorageUsedAvg, end))
  150. }
  151. func (pds *PrometheusMetricsQuerier) QueryLocalStorageUsedMax(start, end time.Time) *source.Future[source.LocalStorageUsedMaxResult] {
  152. const queryName = "QueryLocalStorageUsedMax"
  153. const localStorageUsedMaxQuery = `max(sum(max_over_time(container_fs_usage_bytes{device=~"/dev/(nvme|sda).*", id="/", %s}[%s])) by (instance, device, uid, %s, job)) by (instance, device, uid, %s)`
  154. cfg := pds.promConfig
  155. durStr := timeutil.DurationString(end.Sub(start))
  156. if durStr == "" {
  157. panic("failed to parse duration string passed to QueryLocalStorageUsedMax")
  158. }
  159. queryLocalStorageUsedMax := fmt.Sprintf(localStorageUsedMaxQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel, cfg.ClusterLabel)
  160. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLocalStorageUsedMax)
  161. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  162. return source.NewFuture(source.DecodeLocalStorageUsedMaxResult, ctx.QueryAtTime(queryLocalStorageUsedMax, end))
  163. }
  164. func (pds *PrometheusMetricsQuerier) QueryLocalStorageBytes(start, end time.Time) *source.Future[source.LocalStorageBytesResult] {
  165. const queryName = "QueryLocalStorageBytes"
  166. const localStorageBytesQuery = `avg_over_time(sum(container_fs_limit_bytes{device=~"/dev/(nvme|sda).*", id="/", %s}) by (instance, device, uid, %s)[%s:%dm])`
  167. cfg := pds.promConfig
  168. minsPerResolution := cfg.DataResolutionMinutes
  169. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  170. if durStr == "" {
  171. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  172. }
  173. queryLocalStorageBytes := fmt.Sprintf(localStorageBytesQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  174. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLocalStorageBytes)
  175. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  176. return source.NewFuture(source.DecodeLocalStorageBytesResult, ctx.QueryAtTime(queryLocalStorageBytes, end))
  177. }
  178. func (pds *PrometheusMetricsQuerier) QueryLocalStorageActiveMinutes(start, end time.Time) *source.Future[source.LocalStorageActiveMinutesResult] {
  179. const queryName = "QueryLocalStorageActiveMinutes"
  180. const localStorageActiveMinutesQuery = `count(node_total_hourly_cost{%s}) by (%s, node, uid, instance, provider_id)[%s:%dm]`
  181. cfg := pds.promConfig
  182. minsPerResolution := cfg.DataResolutionMinutes
  183. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  184. if durStr == "" {
  185. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  186. }
  187. queryLocalStorageActiveMins := fmt.Sprintf(localStorageActiveMinutesQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  188. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLocalStorageActiveMins)
  189. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  190. return source.NewFuture(source.DecodeLocalStorageActiveMinutesResult, ctx.QueryAtTime(queryLocalStorageActiveMins, end))
  191. }
  192. func (pds *PrometheusMetricsQuerier) QueryNodeCPUCoresCapacity(start, end time.Time) *source.Future[source.NodeCPUCoresCapacityResult] {
  193. const queryName = "QueryNodeCPUCoresCapacity"
  194. const nodeCPUCoresCapacityQuery = `avg(avg_over_time(kube_node_status_capacity_cpu_cores{%s}[%s])) by (%s, node, uid)`
  195. cfg := pds.promConfig
  196. durStr := timeutil.DurationString(end.Sub(start))
  197. if durStr == "" {
  198. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  199. }
  200. queryNodeCPUCoresCapacity := fmt.Sprintf(nodeCPUCoresCapacityQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  201. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeCPUCoresCapacity)
  202. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  203. return source.NewFuture(source.DecodeNodeCPUCoresCapacityResult, ctx.QueryAtTime(queryNodeCPUCoresCapacity, end))
  204. }
  205. func (pds *PrometheusMetricsQuerier) QueryNodeCPUCoresAllocatable(start, end time.Time) *source.Future[source.NodeCPUCoresAllocatableResult] {
  206. const queryName = "QueryNodeCPUCoresAllocatable"
  207. const nodeCPUCoresAllocatableQuery = `avg(avg_over_time(kube_node_status_allocatable_cpu_cores{%s}[%s])) by (%s, node, uid)`
  208. // `avg(avg_over_time(container_cpu_allocation{container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, %s)`
  209. cfg := pds.promConfig
  210. durStr := timeutil.DurationString(end.Sub(start))
  211. if durStr == "" {
  212. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  213. }
  214. queryNodeCPUCoresAllocatable := fmt.Sprintf(nodeCPUCoresAllocatableQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  215. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeCPUCoresAllocatable)
  216. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  217. return source.NewFuture(source.DecodeNodeCPUCoresAllocatableResult, ctx.QueryAtTime(queryNodeCPUCoresAllocatable, end))
  218. }
  219. func (pds *PrometheusMetricsQuerier) QueryNodeRAMBytesCapacity(start, end time.Time) *source.Future[source.NodeRAMBytesCapacityResult] {
  220. const queryName = "QueryNodeRAMBytesCapacity"
  221. const nodeRAMBytesCapacityQuery = `avg(avg_over_time(kube_node_status_capacity_memory_bytes{%s}[%s])) by (%s, node, uid)`
  222. cfg := pds.promConfig
  223. durStr := timeutil.DurationString(end.Sub(start))
  224. if durStr == "" {
  225. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  226. }
  227. queryNodeRAMBytesCapacity := fmt.Sprintf(nodeRAMBytesCapacityQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  228. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeRAMBytesCapacity)
  229. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  230. return source.NewFuture(source.DecodeNodeRAMBytesCapacityResult, ctx.QueryAtTime(queryNodeRAMBytesCapacity, end))
  231. }
  232. func (pds *PrometheusMetricsQuerier) QueryNodeRAMBytesAllocatable(start, end time.Time) *source.Future[source.NodeRAMBytesAllocatableResult] {
  233. const queryName = "QueryNodeRAMBytesAllocatable"
  234. const nodeRAMBytesAllocatableQuery = `avg(avg_over_time(kube_node_status_allocatable_memory_bytes{%s}[%s])) by (%s, node, uid)`
  235. cfg := pds.promConfig
  236. durStr := timeutil.DurationString(end.Sub(start))
  237. if durStr == "" {
  238. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  239. }
  240. queryNodeRAMBytesAllocatable := fmt.Sprintf(nodeRAMBytesAllocatableQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  241. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeRAMBytesAllocatable)
  242. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  243. return source.NewFuture(source.DecodeNodeRAMBytesAllocatableResult, ctx.QueryAtTime(queryNodeRAMBytesAllocatable, end))
  244. }
  245. func (pds *PrometheusMetricsQuerier) QueryNodeGPUCount(start, end time.Time) *source.Future[source.NodeGPUCountResult] {
  246. const queryName = "QueryNodeGPUCount"
  247. const nodeGPUCountQuery = `avg(avg_over_time(node_gpu_count{%s}[%s])) by (%s, node, uid, provider_id)`
  248. cfg := pds.promConfig
  249. durStr := timeutil.DurationString(end.Sub(start))
  250. if durStr == "" {
  251. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  252. }
  253. queryNodeGPUCount := fmt.Sprintf(nodeGPUCountQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  254. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeGPUCount)
  255. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  256. return source.NewFuture(source.DecodeNodeGPUCountResult, ctx.QueryAtTime(queryNodeGPUCount, end))
  257. }
  258. func (pds *PrometheusMetricsQuerier) QueryNodeLabels(start, end time.Time) *source.Future[source.NodeLabelsResult] {
  259. const queryName = "QueryNodeLabels"
  260. const labelsQuery = `avg_over_time(kube_node_labels{%s}[%s])`
  261. cfg := pds.promConfig
  262. durStr := timeutil.DurationString(end.Sub(start))
  263. if durStr == "" {
  264. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  265. }
  266. queryLabels := fmt.Sprintf(labelsQuery, cfg.ClusterFilter, durStr)
  267. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLabels)
  268. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  269. return source.NewFuture(source.DecodeNodeLabelsResult, ctx.QueryAtTime(queryLabels, end))
  270. }
  271. func (pds *PrometheusMetricsQuerier) QueryNodeActiveMinutes(start, end time.Time) *source.Future[source.NodeActiveMinutesResult] {
  272. const queryName = "QueryNodeActiveMinutes"
  273. const activeMinsQuery = `avg(node_total_hourly_cost{%s}) by (node, uid, %s, provider_id)[%s:%dm]`
  274. cfg := pds.promConfig
  275. minsPerResolution := cfg.DataResolutionMinutes
  276. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  277. if durStr == "" {
  278. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  279. }
  280. queryActiveMins := fmt.Sprintf(activeMinsQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  281. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryActiveMins)
  282. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  283. return source.NewFuture(source.DecodeNodeActiveMinutesResult, ctx.QueryAtTime(queryActiveMins, end))
  284. }
  285. func (pds *PrometheusMetricsQuerier) QueryNodeCPUModeTotal(start, end time.Time) *source.Future[source.NodeCPUModeTotalResult] {
  286. const queryName = "QueryNodeCPUModeTotal"
  287. const nodeCPUModeTotalQuery = `sum(rate(node_cpu_seconds_total{%s}[%s:%dm])) by (kubernetes_node, uid, %s, mode)`
  288. cfg := pds.promConfig
  289. minsPerResolution := cfg.DataResolutionMinutes
  290. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  291. if durStr == "" {
  292. panic("failed to parse duration string passed to QueryNodeCPUModeTotal")
  293. }
  294. queryCPUModeTotal := fmt.Sprintf(nodeCPUModeTotalQuery, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  295. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryCPUModeTotal)
  296. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  297. return source.NewFuture(source.DecodeNodeCPUModeTotalResult, ctx.QueryAtTime(queryCPUModeTotal, end))
  298. }
  299. func (pds *PrometheusMetricsQuerier) QueryNodeRAMSystemPercent(start, end time.Time) *source.Future[source.NodeRAMSystemPercentResult] {
  300. const queryName = "QueryNodeRAMSystemPercent"
  301. const nodeRAMSystemPctQuery = `sum(sum_over_time(container_memory_working_set_bytes{container_name!="POD",container_name!="",namespace="kube-system", %s}[%s:%dm])) by (instance, uid, %s) / avg(label_replace(sum(sum_over_time(kube_node_status_capacity_memory_bytes{%s}[%s:%dm])) by (node, uid, %s), "instance", "$1", "node", "(.*)")) by (instance, uid, %s)`
  302. cfg := pds.promConfig
  303. minsPerResolution := cfg.DataResolutionMinutes
  304. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  305. if durStr == "" {
  306. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  307. }
  308. queryRAMSystemPct := fmt.Sprintf(nodeRAMSystemPctQuery, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel, cfg.ClusterLabel)
  309. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryRAMSystemPct)
  310. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  311. return source.NewFuture(source.DecodeNodeRAMSystemPercentResult, ctx.QueryAtTime(queryRAMSystemPct, end))
  312. }
  313. func (pds *PrometheusMetricsQuerier) QueryNodeRAMUserPercent(start, end time.Time) *source.Future[source.NodeRAMUserPercentResult] {
  314. const queryName = "QueryNodeRAMUserPercent"
  315. const nodeRAMUserPctQuery = `sum(sum_over_time(container_memory_working_set_bytes{container_name!="POD",container_name!="",namespace!="kube-system", %s}[%s:%dm])) by (instance, uid, %s) / avg(label_replace(sum(sum_over_time(kube_node_status_capacity_memory_bytes{%s}[%s:%dm])) by (node, uid, %s), "instance", "$1", "node", "(.*)")) by (instance, uid, %s)`
  316. cfg := pds.promConfig
  317. minsPerResolution := cfg.DataResolutionMinutes
  318. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  319. if durStr == "" {
  320. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  321. }
  322. queryRAMUserPct := fmt.Sprintf(nodeRAMUserPctQuery, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel, cfg.ClusterLabel)
  323. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryRAMUserPct)
  324. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  325. return source.NewFuture(source.DecodeNodeRAMUserPercentResult, ctx.QueryAtTime(queryRAMUserPct, end))
  326. }
  327. func (pds *PrometheusMetricsQuerier) QueryLBPricePerHr(start, end time.Time) *source.Future[source.LBPricePerHrResult] {
  328. const queryName = "QueryLBPricePerHr"
  329. const queryFmtLBCostPerHr = `avg(avg_over_time(kubecost_load_balancer_cost{%s}[%s])) by (namespace, service_name, ingress_ip, uid, %s)`
  330. cfg := pds.promConfig
  331. durStr := timeutil.DurationString(end.Sub(start))
  332. if durStr == "" {
  333. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  334. }
  335. queryLBCostPerHr := fmt.Sprintf(queryFmtLBCostPerHr, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  336. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLBCostPerHr)
  337. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  338. return source.NewFuture(source.DecodeLBPricePerHrResult, ctx.QueryAtTime(queryLBCostPerHr, end))
  339. }
  340. func (pds *PrometheusMetricsQuerier) QueryLBActiveMinutes(start, end time.Time) *source.Future[source.LBActiveMinutesResult] {
  341. const queryName = "QueryLBActiveMinutes"
  342. const lbActiveMinutesQuery = `avg(kubecost_load_balancer_cost{%s}) by (namespace, service_name, uid, %s, ingress_ip)[%s:%dm]`
  343. cfg := pds.promConfig
  344. minsPerResolution := cfg.DataResolutionMinutes
  345. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  346. if durStr == "" {
  347. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  348. }
  349. queryLBActiveMins := fmt.Sprintf(lbActiveMinutesQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  350. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryLBActiveMins)
  351. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  352. return source.NewFuture(source.DecodeLBActiveMinutesResult, ctx.QueryAtTime(queryLBActiveMins, end))
  353. }
  354. // Note: cluster_info is not currently emitted
  355. func (pds *PrometheusMetricsQuerier) QueryClusterUptime(start, end time.Time) *source.Future[source.UptimeResult] {
  356. const queryName = "QueryClusterUptime"
  357. const queryFmtClusterUptime = `avg(cluster_info{%s}) by (%s, uid)[%s:%dm]`
  358. cfg := pds.promConfig
  359. minsPerResolution := cfg.DataResolutionMinutes
  360. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  361. if durStr == "" {
  362. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  363. }
  364. queryClusterUptime := fmt.Sprintf(queryFmtClusterUptime, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  365. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryFmtClusterUptime)
  366. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  367. return source.NewFuture(source.DecodeUptimeResult, ctx.QueryAtTime(queryClusterUptime, end))
  368. }
  369. func (pds *PrometheusMetricsQuerier) QueryClusterManagementDuration(start, end time.Time) *source.Future[source.ClusterManagementDurationResult] {
  370. const queryName = "QueryClusterManagementDuration"
  371. const clusterManagementDurationQuery = `avg(kubecost_cluster_management_cost{%s}) by (%s, provisioner_name)[%s:%dm]`
  372. cfg := pds.promConfig
  373. minsPerResolution := cfg.DataResolutionMinutes
  374. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  375. if durStr == "" {
  376. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  377. }
  378. queryClusterManagementDuration := fmt.Sprintf(clusterManagementDurationQuery, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  379. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryClusterManagementDuration)
  380. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  381. return source.NewFuture(source.DecodeClusterManagementDurationResult, ctx.QueryAtTime(queryClusterManagementDuration, end))
  382. }
  383. func (pds *PrometheusMetricsQuerier) QueryClusterManagementPricePerHr(start, end time.Time) *source.Future[source.ClusterManagementPricePerHrResult] {
  384. const queryName = "QueryClusterManagementPricePerHr"
  385. const clusterManagementCostQuery = `avg(avg_over_time(kubecost_cluster_management_cost{%s}[%s])) by (%s, provisioner_name)`
  386. cfg := pds.promConfig
  387. durStr := timeutil.DurationString(end.Sub(start))
  388. if durStr == "" {
  389. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  390. }
  391. queryClusterManagementCost := fmt.Sprintf(clusterManagementCostQuery, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  392. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryClusterManagementCost)
  393. ctx := pds.promContexts.NewNamedContext(ClusterContextName)
  394. return source.NewFuture(source.DecodeClusterManagementPricePerHrResult, ctx.QueryAtTime(queryClusterManagementCost, end))
  395. }
  396. // AllocationMetricQuerier
  397. func (pds *PrometheusMetricsQuerier) QueryPods(start, end time.Time) *source.Future[source.PodsResult] {
  398. const queryName = "QueryPods"
  399. const queryFmtPods = `avg(kube_pod_container_status_running{%s} != 0) by (pod, namespace, uid, %s)[%s:%dm]`
  400. cfg := pds.promConfig
  401. minsPerResolution := cfg.DataResolutionMinutes
  402. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  403. if durStr == "" {
  404. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  405. }
  406. queryPods := fmt.Sprintf(queryFmtPods, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  407. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPods)
  408. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  409. return source.NewFuture(source.DecodePodsResult, ctx.QueryAtTime(queryPods, end))
  410. }
  411. func (pds *PrometheusMetricsQuerier) QueryPodsUID(start, end time.Time) *source.Future[source.PodsResult] {
  412. const queryName = "QueryPodsUID"
  413. const queryFmtPodsUID = `avg(kube_pod_container_status_running{%s} != 0) by (pod, namespace, uid, %s)[%s:%dm]`
  414. cfg := pds.promConfig
  415. minsPerResolution := cfg.DataResolutionMinutes
  416. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  417. if durStr == "" {
  418. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  419. }
  420. queryPodsUID := fmt.Sprintf(queryFmtPodsUID, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  421. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPodsUID)
  422. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  423. return source.NewFuture(source.DecodePodsResult, ctx.QueryAtTime(queryPodsUID, end))
  424. }
  425. func (pds *PrometheusMetricsQuerier) QueryRAMBytesAllocated(start, end time.Time) *source.Future[source.RAMBytesAllocatedResult] {
  426. const queryName = "QueryRAMBytesAllocated"
  427. const queryFmtRAMBytesAllocated = `avg(avg_over_time(container_memory_allocation_bytes{container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, uid, %s, provider_id)`
  428. cfg := pds.promConfig
  429. durStr := timeutil.DurationString(end.Sub(start))
  430. if durStr == "" {
  431. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  432. }
  433. queryRAMBytesAllocated := fmt.Sprintf(queryFmtRAMBytesAllocated, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  434. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryRAMBytesAllocated)
  435. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  436. return source.NewFuture(source.DecodeRAMBytesAllocatedResult, ctx.QueryAtTime(queryRAMBytesAllocated, end))
  437. }
  438. func (pds *PrometheusMetricsQuerier) QueryRAMRequests(start, end time.Time) *source.Future[source.RAMRequestsResult] {
  439. const queryName = "QueryRAMRequests"
  440. const queryFmtRAMRequests = `avg(avg_over_time(kube_pod_container_resource_requests{resource="memory", unit="byte", container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, uid, %s)`
  441. cfg := pds.promConfig
  442. durStr := timeutil.DurationString(end.Sub(start))
  443. if durStr == "" {
  444. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  445. }
  446. queryRAMRequests := fmt.Sprintf(queryFmtRAMRequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  447. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryRAMRequests)
  448. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  449. return source.NewFuture(source.DecodeRAMRequestsResult, ctx.QueryAtTime(queryRAMRequests, end))
  450. }
  451. func (pds *PrometheusMetricsQuerier) QueryRAMLimits(start, end time.Time) *source.Future[source.RAMLimitsResult] {
  452. const queryName = "QueryRAMLimits"
  453. const queryFmtRAMLimits = `avg(avg_over_time(kube_pod_container_resource_limits{resource="memory", unit="byte", container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, %s)`
  454. cfg := pds.promConfig
  455. durStr := timeutil.DurationString(end.Sub(start))
  456. if durStr == "" {
  457. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  458. }
  459. queryRAMLimits := fmt.Sprintf(queryFmtRAMLimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  460. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryRAMLimits)
  461. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  462. return source.NewFuture(source.DecodeRAMLimitsResult, ctx.QueryAtTime(queryRAMLimits, end))
  463. }
  464. func (pds *PrometheusMetricsQuerier) QueryRAMUsageAvg(start, end time.Time) *source.Future[source.RAMUsageAvgResult] {
  465. const queryName = "QueryRAMUsageAvg"
  466. const queryFmtRAMUsageAvg = `avg(avg_over_time(container_memory_working_set_bytes{container!="", container_name!="POD", container!="POD", %s}[%s])) by (container_name, container, pod_name, pod, namespace, node, instance, uid, %s)`
  467. cfg := pds.promConfig
  468. durStr := timeutil.DurationString(end.Sub(start))
  469. if durStr == "" {
  470. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  471. }
  472. queryRAMUsageAvg := fmt.Sprintf(queryFmtRAMUsageAvg, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  473. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryRAMUsageAvg)
  474. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  475. return source.NewFuture(source.DecodeRAMUsageAvgResult, ctx.QueryAtTime(queryRAMUsageAvg, end))
  476. }
  477. func (pds *PrometheusMetricsQuerier) QueryRAMUsageMax(start, end time.Time) *source.Future[source.RAMUsageMaxResult] {
  478. const queryName = "QueryRAMUsageMax"
  479. const queryFmtRAMUsageMax = `max(max_over_time(container_memory_working_set_bytes{container!="", container_name!="POD", container!="POD", %s}[%s])) by (container_name, container, pod_name, pod, namespace, node, instance, uid, %s)`
  480. cfg := pds.promConfig
  481. durStr := timeutil.DurationString(end.Sub(start))
  482. if durStr == "" {
  483. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  484. }
  485. queryRAMUsageMax := fmt.Sprintf(queryFmtRAMUsageMax, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  486. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryRAMUsageMax)
  487. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  488. return source.NewFuture(source.DecodeRAMUsageMaxResult, ctx.QueryAtTime(queryRAMUsageMax, end))
  489. }
  490. func (pds *PrometheusMetricsQuerier) QueryCPUCoresAllocated(start, end time.Time) *source.Future[source.CPUCoresAllocatedResult] {
  491. const queryName = "QueryCPUCoresAllocated"
  492. const queryFmtCPUCoresAllocated = `avg(avg_over_time(container_cpu_allocation{container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, uid, %s)`
  493. cfg := pds.promConfig
  494. durStr := timeutil.DurationString(end.Sub(start))
  495. if durStr == "" {
  496. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  497. }
  498. queryCPUCoresAllocated := fmt.Sprintf(queryFmtCPUCoresAllocated, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  499. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryCPUCoresAllocated)
  500. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  501. return source.NewFuture(source.DecodeCPUCoresAllocatedResult, ctx.QueryAtTime(queryCPUCoresAllocated, end))
  502. }
  503. func (pds *PrometheusMetricsQuerier) QueryCPURequests(start, end time.Time) *source.Future[source.CPURequestsResult] {
  504. const queryName = "QueryCPURequests"
  505. const queryFmtCPURequests = `avg(avg_over_time(kube_pod_container_resource_requests{resource="cpu", unit="core", container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, uid, %s)`
  506. cfg := pds.promConfig
  507. durStr := timeutil.DurationString(end.Sub(start))
  508. if durStr == "" {
  509. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  510. }
  511. queryCPURequests := fmt.Sprintf(queryFmtCPURequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  512. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryCPURequests)
  513. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  514. return source.NewFuture(source.DecodeCPURequestsResult, ctx.QueryAtTime(queryCPURequests, end))
  515. }
  516. func (pds *PrometheusMetricsQuerier) QueryCPULimits(start, end time.Time) *source.Future[source.CPULimitsResult] {
  517. const queryName = "QueryCPULimits"
  518. const queryFmtCPULimits = `avg(avg_over_time(kube_pod_container_resource_limits{resource="cpu", unit="core", container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, %s)`
  519. cfg := pds.promConfig
  520. durStr := timeutil.DurationString(end.Sub(start))
  521. if durStr == "" {
  522. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  523. }
  524. queryCPULimits := fmt.Sprintf(queryFmtCPULimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  525. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryCPULimits)
  526. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  527. return source.NewFuture(source.DecodeCPULimitsResult, ctx.QueryAtTime(queryCPULimits, end))
  528. }
  529. func (pds *PrometheusMetricsQuerier) QueryCPUUsageAvg(start, end time.Time) *source.Future[source.CPUUsageAvgResult] {
  530. const queryName = "QueryCPUUsageAvg"
  531. const queryFmtCPUUsageAvg = `avg(rate(container_cpu_usage_seconds_total{container!="", container_name!="POD", container!="POD", %s}[%s])) by (container_name, container, pod_name, pod, namespace, node, instance, uid, %s)`
  532. cfg := pds.promConfig
  533. durStr := timeutil.DurationString(end.Sub(start))
  534. if durStr == "" {
  535. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  536. }
  537. queryCPUUsageAvg := fmt.Sprintf(queryFmtCPUUsageAvg, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  538. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryCPUUsageAvg)
  539. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  540. return source.NewFuture(source.DecodeCPUUsageAvgResult, ctx.QueryAtTime(queryCPUUsageAvg, end))
  541. }
  542. func (pds *PrometheusMetricsQuerier) QueryCPUUsageMax(start, end time.Time) *source.Future[source.CPUUsageMaxResult] {
  543. const queryName = "QueryCPUUsageMax"
  544. // Because we use container_cpu_usage_seconds_total to calculate CPU usage
  545. // at any given "instant" of time, we need to use an irate or rate. To then
  546. // calculate a max (or any aggregation) we have to perform an aggregation
  547. // query on top of an instant-by-instant maximum. Prometheus supports this
  548. // type of query with a "subquery" [1], however it is reportedly expensive
  549. // to make such a query. By default, Kubecost's Prometheus config includes
  550. // a recording rule that keeps track of the instant-by-instant irate for CPU
  551. // usage. The metric in this query is created by that recording rule.
  552. //
  553. // [1] https://prometheus.io/blog/2019/01/28/subquery-support/
  554. //
  555. // If changing the name of the recording rule, make sure to update the
  556. // corresponding diagnostic query to avoid confusion.
  557. const queryFmtCPUUsageMaxRecordingRule = `max(max_over_time(kubecost_container_cpu_usage_irate{%s}[%s])) by (container_name, container, pod_name, pod, namespace, node, instance, uid, %s)`
  558. // This is the subquery equivalent of the above recording rule query. It is
  559. // more expensive, but does not require the recording rule. It should be
  560. // used as a fallback query if the recording rule data does not exist.
  561. //
  562. // The parameter after the colon [:<thisone>] in the subquery affects the
  563. // resolution of the subquery.
  564. // The parameter after the metric ...{}[<thisone>] should be set to 2x
  565. // the resolution, to make sure the irate always has two points to query
  566. // in case the Prom scrape duration has been reduced to be equal to the
  567. // query resolution.
  568. const queryFmtCPUUsageMaxSubquery = `max(max_over_time(irate(container_cpu_usage_seconds_total{container!="POD", container!="", %s}[%dm])[%s:%dm])) by (container, pod_name, pod, namespace, node, instance, uid, %s)`
  569. cfg := pds.promConfig
  570. durStr := timeutil.DurationString(end.Sub(start))
  571. if durStr == "" {
  572. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  573. }
  574. queryCPUUsageMaxRecordingRule := fmt.Sprintf(queryFmtCPUUsageMaxRecordingRule, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  575. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryCPUUsageMaxRecordingRule)
  576. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  577. resCPUUsageMaxRR := ctx.QueryAtTime(queryCPUUsageMaxRecordingRule, end)
  578. resCPUUsageMax, _ := resCPUUsageMaxRR.Await()
  579. if len(resCPUUsageMax) > 0 {
  580. return source.NewFutureFrom(source.DecodeAll(resCPUUsageMax, source.DecodeCPUUsageMaxResult))
  581. }
  582. minsPerResolution := cfg.DataResolutionMinutes
  583. durStr = pds.durationStringFor(start, end, minsPerResolution, false)
  584. if durStr == "" {
  585. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  586. }
  587. queryCPUUsageMaxSubquery := fmt.Sprintf(queryFmtCPUUsageMaxSubquery, cfg.ClusterFilter, 2*minsPerResolution, durStr, minsPerResolution, cfg.ClusterLabel)
  588. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryCPUUsageMaxSubquery)
  589. return source.NewFuture(source.DecodeCPUUsageMaxResult, ctx.QueryAtTime(queryCPUUsageMaxSubquery, end))
  590. }
  591. func (pds *PrometheusMetricsQuerier) QueryGPUsRequested(start, end time.Time) *source.Future[source.GPUsRequestedResult] {
  592. const queryName = "QueryGPUsRequested"
  593. const queryFmtGPUsRequested = `avg(avg_over_time(kube_pod_container_resource_requests{resource="nvidia_com_gpu", container!="",container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, uid, %s)`
  594. cfg := pds.promConfig
  595. durStr := timeutil.DurationString(end.Sub(start))
  596. if durStr == "" {
  597. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  598. }
  599. queryGPUsRequested := fmt.Sprintf(queryFmtGPUsRequested, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  600. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryGPUsRequested)
  601. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  602. return source.NewFuture(source.DecodeGPUsRequestedResult, ctx.QueryAtTime(queryGPUsRequested, end))
  603. }
  604. func (pds *PrometheusMetricsQuerier) QueryGPUsUsageAvg(start, end time.Time) *source.Future[source.GPUsUsageAvgResult] {
  605. const queryName = "QueryGPUsUsageAvg"
  606. const queryFmtGPUsUsageAvg = `avg(avg_over_time(DCGM_FI_PROF_GR_ENGINE_ACTIVE{container!=""}[%s])) by (container, pod, namespace, uid, %s)`
  607. cfg := pds.promConfig
  608. durStr := timeutil.DurationString(end.Sub(start))
  609. if durStr == "" {
  610. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  611. }
  612. queryGPUsUsageAvg := fmt.Sprintf(queryFmtGPUsUsageAvg, durStr, cfg.ClusterLabel)
  613. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryGPUsUsageAvg)
  614. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  615. return source.NewFuture(source.DecodeGPUsUsageAvgResult, ctx.QueryAtTime(queryGPUsUsageAvg, end))
  616. }
  617. func (pds *PrometheusMetricsQuerier) QueryGPUsUsageMax(start, end time.Time) *source.Future[source.GPUsUsageMaxResult] {
  618. const queryName = "QueryGPUsUsageMax"
  619. const queryFmtGPUsUsageMax = `max(max_over_time(DCGM_FI_PROF_GR_ENGINE_ACTIVE{container!=""}[%s])) by (container, pod, namespace, uid, %s)`
  620. cfg := pds.promConfig
  621. durStr := timeutil.DurationString(end.Sub(start))
  622. if durStr == "" {
  623. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  624. }
  625. queryGPUsUsageMax := fmt.Sprintf(queryFmtGPUsUsageMax, durStr, cfg.ClusterLabel)
  626. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryGPUsUsageMax)
  627. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  628. return source.NewFuture(source.DecodeGPUsUsageMaxResult, ctx.QueryAtTime(queryGPUsUsageMax, end))
  629. }
  630. func (pds *PrometheusMetricsQuerier) QueryGPUsAllocated(start, end time.Time) *source.Future[source.GPUsAllocatedResult] {
  631. const queryName = "QueryGPUsAllocated"
  632. const queryFmtGPUsAllocated = `avg(avg_over_time(container_gpu_allocation{container!="", container!="POD", node!="", %s}[%s])) by (container, pod, namespace, node, uid, %s)`
  633. cfg := pds.promConfig
  634. durStr := timeutil.DurationString(end.Sub(start))
  635. if durStr == "" {
  636. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  637. }
  638. queryGPUsAllocated := fmt.Sprintf(queryFmtGPUsAllocated, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  639. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryGPUsAllocated)
  640. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  641. return source.NewFuture(source.DecodeGPUsAllocatedResult, ctx.QueryAtTime(queryGPUsAllocated, end))
  642. }
  643. func (pds *PrometheusMetricsQuerier) QueryIsGPUShared(start, end time.Time) *source.Future[source.IsGPUSharedResult] {
  644. const queryName = "QueryIsGPUShared"
  645. const queryFmtIsGPUShared = `avg(avg_over_time(kube_pod_container_resource_requests{container!="", node != "", pod != "", container!= "", unit = "integer", %s}[%s])) by (container, pod, namespace, node, resource, uid, %s)`
  646. cfg := pds.promConfig
  647. durStr := timeutil.DurationString(end.Sub(start))
  648. if durStr == "" {
  649. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  650. }
  651. queryIsGPUShared := fmt.Sprintf(queryFmtIsGPUShared, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  652. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryIsGPUShared)
  653. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  654. return source.NewFuture(source.DecodeIsGPUSharedResult, ctx.QueryAtTime(queryIsGPUShared, end))
  655. }
  656. func (pds *PrometheusMetricsQuerier) QueryGPUInfo(start, end time.Time) *source.Future[source.GPUInfoResult] {
  657. const queryName = "QueryGPUInfo"
  658. const queryFmtGetGPUInfo = `avg(avg_over_time(DCGM_FI_DEV_DEC_UTIL{container!="",%s}[%s])) by (container, pod, namespace, device, modelName, UUID, uid, %s)`
  659. cfg := pds.promConfig
  660. durStr := timeutil.DurationString(end.Sub(start))
  661. if durStr == "" {
  662. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  663. }
  664. queryGetGPUInfo := fmt.Sprintf(queryFmtGetGPUInfo, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  665. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryGetGPUInfo)
  666. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  667. return source.NewFuture(source.DecodeGPUInfoResult, ctx.QueryAtTime(queryGetGPUInfo, end))
  668. }
  669. func (pds *PrometheusMetricsQuerier) QueryNodeCPUPricePerHr(start, end time.Time) *source.Future[source.NodeCPUPricePerHrResult] {
  670. const queryName = "QueryNodeCPUPricePerHr"
  671. const queryFmtNodeCostPerCPUHr = `avg(avg_over_time(node_cpu_hourly_cost{%s}[%s])) by (node, uid, %s, instance_type, provider_id)`
  672. cfg := pds.promConfig
  673. durStr := timeutil.DurationString(end.Sub(start))
  674. if durStr == "" {
  675. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  676. }
  677. queryNodeCostPerCPUHr := fmt.Sprintf(queryFmtNodeCostPerCPUHr, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  678. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeCostPerCPUHr)
  679. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  680. return source.NewFuture(source.DecodeNodeCPUPricePerHrResult, ctx.QueryAtTime(queryNodeCostPerCPUHr, end))
  681. }
  682. func (pds *PrometheusMetricsQuerier) QueryNodeRAMPricePerGiBHr(start, end time.Time) *source.Future[source.NodeRAMPricePerGiBHrResult] {
  683. const queryName = "QueryNodeRAMPricePerGiBHr"
  684. const queryFmtNodeCostPerRAMGiBHr = `avg(avg_over_time(node_ram_hourly_cost{%s}[%s])) by (node, uid, %s, instance_type, provider_id)`
  685. cfg := pds.promConfig
  686. durStr := timeutil.DurationString(end.Sub(start))
  687. if durStr == "" {
  688. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  689. }
  690. queryNodeCostPerRAMGiBHr := fmt.Sprintf(queryFmtNodeCostPerRAMGiBHr, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  691. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeCostPerRAMGiBHr)
  692. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  693. return source.NewFuture(source.DecodeNodeRAMPricePerGiBHrResult, ctx.QueryAtTime(queryNodeCostPerRAMGiBHr, end))
  694. }
  695. func (pds *PrometheusMetricsQuerier) QueryNodeGPUPricePerHr(start, end time.Time) *source.Future[source.NodeGPUPricePerHrResult] {
  696. const queryName = "QueryNodeGPUPricePerHr"
  697. const queryFmtNodeCostPerGPUHr = `avg(avg_over_time(node_gpu_hourly_cost{%s}[%s])) by (node, uid, %s, instance_type, provider_id)`
  698. cfg := pds.promConfig
  699. durStr := timeutil.DurationString(end.Sub(start))
  700. if durStr == "" {
  701. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  702. }
  703. queryNodeCostPerGPUHr := fmt.Sprintf(queryFmtNodeCostPerGPUHr, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  704. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeCostPerGPUHr)
  705. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  706. return source.NewFuture(source.DecodeNodeGPUPricePerHrResult, ctx.QueryAtTime(queryNodeCostPerGPUHr, end))
  707. }
  708. func (pds *PrometheusMetricsQuerier) QueryNodeIsSpot(start, end time.Time) *source.Future[source.NodeIsSpotResult] {
  709. const queryName = "QueryNodeIsSpot"
  710. const queryFmtNodeIsSpot = `avg_over_time(kubecost_node_is_spot{%s}[%s])`
  711. cfg := pds.promConfig
  712. durStr := timeutil.DurationString(end.Sub(start))
  713. if durStr == "" {
  714. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  715. }
  716. queryNodeIsSpot := fmt.Sprintf(queryFmtNodeIsSpot, cfg.ClusterFilter, durStr)
  717. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNodeIsSpot)
  718. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  719. return source.NewFuture(source.DecodeNodeIsSpotResult, ctx.QueryAtTime(queryNodeIsSpot, end))
  720. }
  721. func (pds *PrometheusMetricsQuerier) QueryPodPVCAllocation(start, end time.Time) *source.Future[source.PodPVCAllocationResult] {
  722. const queryName = "QueryPodPVCAllocation"
  723. const queryFmtPodPVCAllocation = `avg(avg_over_time(pod_pvc_allocation{%s}[%s])) by (persistentvolume, persistentvolumeclaim, pod, namespace, uid, %s)`
  724. cfg := pds.promConfig
  725. durStr := timeutil.DurationString(end.Sub(start))
  726. if durStr == "" {
  727. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  728. }
  729. queryPodPVCAllocation := fmt.Sprintf(queryFmtPodPVCAllocation, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  730. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPodPVCAllocation)
  731. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  732. return source.NewFuture(source.DecodePodPVCAllocationResult, ctx.QueryAtTime(queryPodPVCAllocation, end))
  733. }
  734. func (pds *PrometheusMetricsQuerier) QueryPVCBytesRequested(start, end time.Time) *source.Future[source.PVCBytesRequestedResult] {
  735. const queryName = "QueryPVCBytesRequested"
  736. const queryFmtPVCBytesRequested = `avg(avg_over_time(kube_persistentvolumeclaim_resource_requests_storage_bytes{%s}[%s])) by (persistentvolumeclaim, namespace, uid, %s)`
  737. cfg := pds.promConfig
  738. durStr := timeutil.DurationString(end.Sub(start))
  739. if durStr == "" {
  740. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  741. }
  742. queryPVCBytesRequested := fmt.Sprintf(queryFmtPVCBytesRequested, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  743. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVCBytesRequested)
  744. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  745. return source.NewFuture(source.DecodePVCBytesRequestedResult, ctx.QueryAtTime(queryPVCBytesRequested, end))
  746. }
  747. func (pds *PrometheusMetricsQuerier) QueryPVBytes(start, end time.Time) *source.Future[source.PVBytesResult] {
  748. const queryName = "QueryPVBytes"
  749. const queryFmtPVBytes = `avg(avg_over_time(kube_persistentvolume_capacity_bytes{%s}[%s])) by (persistentvolume, uid, %s)`
  750. cfg := pds.promConfig
  751. durStr := timeutil.DurationString(end.Sub(start))
  752. if durStr == "" {
  753. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  754. }
  755. queryPVBytes := fmt.Sprintf(queryFmtPVBytes, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  756. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVBytes)
  757. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  758. return source.NewFuture(source.DecodePVBytesResult, ctx.QueryAtTime(queryPVBytes, end))
  759. }
  760. func (pds *PrometheusMetricsQuerier) QueryPVInfo(start, end time.Time) *source.Future[source.PVInfoResult] {
  761. const queryName = "QueryPVInfo"
  762. const queryFmtPVMeta = `avg(avg_over_time(kubecost_pv_info{%s}[%s])) by (%s, storageclass, persistentvolume, uid, provider_id)`
  763. cfg := pds.promConfig
  764. durStr := timeutil.DurationString(end.Sub(start))
  765. if durStr == "" {
  766. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  767. }
  768. queryPVMeta := fmt.Sprintf(queryFmtPVMeta, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  769. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPVMeta)
  770. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  771. return source.NewFuture(source.DecodePVInfoResult, ctx.QueryAtTime(queryPVMeta, end))
  772. }
  773. func (pds *PrometheusMetricsQuerier) QueryNetZoneGiB(start, end time.Time) *source.Future[source.NetZoneGiBResult] {
  774. const queryName = "QueryNetZoneGiB"
  775. const queryFmtNetZoneGiB = `sum(increase(kubecost_pod_network_egress_bytes_total{internet="false", same_zone="false", same_region="true", %s}[%s:%dm])) by (pod_name, namespace, uid, %s) / 1024 / 1024 / 1024`
  776. cfg := pds.promConfig
  777. minsPerResolution := cfg.DataResolutionMinutes
  778. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  779. if durStr == "" {
  780. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  781. }
  782. queryNetZoneGiB := fmt.Sprintf(queryFmtNetZoneGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  783. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetZoneGiB)
  784. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  785. return source.NewFuture(source.DecodeNetZoneGiBResult, ctx.QueryAtTime(queryNetZoneGiB, end))
  786. }
  787. func (pds *PrometheusMetricsQuerier) QueryNetZonePricePerGiB(start, end time.Time) *source.Future[source.NetZonePricePerGiBResult] {
  788. const queryName = "QueryNetZonePricePerGiB"
  789. const queryFmtNetZoneCostPerGiB = `avg(avg_over_time(kubecost_network_zone_egress_cost{%s}[%s])) by (%s)`
  790. cfg := pds.promConfig
  791. durStr := timeutil.DurationString(end.Sub(start))
  792. if durStr == "" {
  793. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  794. }
  795. queryNetZoneCostPerGiB := fmt.Sprintf(queryFmtNetZoneCostPerGiB, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  796. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetZoneCostPerGiB)
  797. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  798. return source.NewFuture(source.DecodeNetZonePricePerGiBResult, ctx.QueryAtTime(queryNetZoneCostPerGiB, end))
  799. }
  800. func (pds *PrometheusMetricsQuerier) QueryNetRegionGiB(start, end time.Time) *source.Future[source.NetRegionGiBResult] {
  801. const queryName = "QueryNetRegionGiB"
  802. const queryFmtNetRegionGiB = `sum(increase(kubecost_pod_network_egress_bytes_total{internet="false", same_zone="false", same_region="false", %s}[%s:%dm])) by (pod_name, namespace, uid, %s) / 1024 / 1024 / 1024`
  803. cfg := pds.promConfig
  804. minsPerResolution := cfg.DataResolutionMinutes
  805. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  806. if durStr == "" {
  807. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  808. }
  809. queryNetRegionGiB := fmt.Sprintf(queryFmtNetRegionGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  810. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetRegionGiB)
  811. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  812. return source.NewFuture(source.DecodeNetRegionGiBResult, ctx.QueryAtTime(queryNetRegionGiB, end))
  813. }
  814. func (pds *PrometheusMetricsQuerier) QueryNetRegionPricePerGiB(start, end time.Time) *source.Future[source.NetRegionPricePerGiBResult] {
  815. const queryName = "QueryNetRegionPricePerGiB"
  816. const queryFmtNetRegionCostPerGiB = `avg(avg_over_time(kubecost_network_region_egress_cost{%s}[%s])) by (%s)`
  817. cfg := pds.promConfig
  818. durStr := timeutil.DurationString(end.Sub(start))
  819. if durStr == "" {
  820. panic("failed to parse duration string passed to QueryNetRegionPricePerGiB")
  821. }
  822. queryNetRegionCostPerGiB := fmt.Sprintf(queryFmtNetRegionCostPerGiB, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  823. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetRegionCostPerGiB)
  824. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  825. return source.NewFuture(source.DecodeNetRegionPricePerGiBResult, ctx.QueryAtTime(queryNetRegionCostPerGiB, end))
  826. }
  827. func (pds *PrometheusMetricsQuerier) QueryNetInternetGiB(start, end time.Time) *source.Future[source.NetInternetGiBResult] {
  828. const queryName = "QueryNetInternetGiB"
  829. const queryFmtNetInternetGiB = `sum(increase(kubecost_pod_network_egress_bytes_total{internet="true", %s}[%s:%dm])) by (pod_name, namespace, uid, %s) / 1024 / 1024 / 1024`
  830. cfg := pds.promConfig
  831. minsPerResolution := cfg.DataResolutionMinutes
  832. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  833. if durStr == "" {
  834. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  835. }
  836. queryNetInternetGiB := fmt.Sprintf(queryFmtNetInternetGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  837. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetInternetGiB)
  838. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  839. return source.NewFuture(source.DecodeNetInternetGiBResult, ctx.QueryAtTime(queryNetInternetGiB, end))
  840. }
  841. func (pds *PrometheusMetricsQuerier) QueryNetInternetPricePerGiB(start, end time.Time) *source.Future[source.NetInternetPricePerGiBResult] {
  842. const queryName = "QueryNetInternetPricePerGiB"
  843. const queryFmtNetInternetCostPerGiB = `avg(avg_over_time(kubecost_network_internet_egress_cost{%s}[%s])) by (%s)`
  844. cfg := pds.promConfig
  845. durStr := timeutil.DurationString(end.Sub(start))
  846. if durStr == "" {
  847. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  848. }
  849. queryNetInternetCostPerGiB := fmt.Sprintf(queryFmtNetInternetCostPerGiB, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  850. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetInternetCostPerGiB)
  851. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  852. return source.NewFuture(source.DecodeNetInternetPricePerGiBResult, ctx.QueryAtTime(queryNetInternetCostPerGiB, end))
  853. }
  854. func (pds *PrometheusMetricsQuerier) QueryNetInternetServiceGiB(start, end time.Time) *source.Future[source.NetInternetServiceGiBResult] {
  855. const queryName = "QueryNetInternetServiceGiB"
  856. const queryFmtNetInternetGiB = `sum(increase(kubecost_pod_network_egress_bytes_total{internet="true", %s}[%s:%dm])) by (pod_name, namespace, service, uid, %s) / 1024 / 1024 / 1024`
  857. cfg := pds.promConfig
  858. minsPerResolution := cfg.DataResolutionMinutes
  859. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  860. if durStr == "" {
  861. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  862. }
  863. queryNetInternetGiB := fmt.Sprintf(queryFmtNetInternetGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  864. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetInternetGiB)
  865. ctx := pds.promContexts.NewNamedContext(NetworkInsightsContextName)
  866. return source.NewFuture(source.DecodeNetInternetServiceGiBResult, ctx.QueryAtTime(queryNetInternetGiB, end))
  867. }
  868. func (pds *PrometheusMetricsQuerier) QueryNetTransferBytes(start, end time.Time) *source.Future[source.NetTransferBytesResult] {
  869. const queryName = "QueryNetTransferBytes"
  870. const queryFmtNetTransferBytes = `sum(increase(container_network_transmit_bytes_total{pod!="", %s}[%s:%dm])) by (pod_name, pod, namespace, uid, %s)`
  871. cfg := pds.promConfig
  872. minsPerResolution := cfg.DataResolutionMinutes
  873. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  874. if durStr == "" {
  875. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  876. }
  877. queryNetTransferBytes := fmt.Sprintf(queryFmtNetTransferBytes, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  878. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetTransferBytes)
  879. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  880. return source.NewFuture(source.DecodeNetTransferBytesResult, ctx.QueryAtTime(queryNetTransferBytes, end))
  881. }
  882. func (pds *PrometheusMetricsQuerier) QueryNetZoneIngressGiB(start, end time.Time) *source.Future[source.NetZoneIngressGiBResult] {
  883. const queryName = "QueryNetZoneIngressGiB"
  884. const queryFmtIngNetZoneGiB = `sum(increase(kubecost_pod_network_ingress_bytes_total{internet="false", same_zone="false", same_region="true", %s}[%s:%dm])) by (pod_name, namespace, uid, %s) / 1024 / 1024 / 1024`
  885. cfg := pds.promConfig
  886. minsPerResolution := cfg.DataResolutionMinutes
  887. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  888. if durStr == "" {
  889. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  890. }
  891. queryNetZoneCostPerGiB := fmt.Sprintf(queryFmtIngNetZoneGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  892. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetZoneCostPerGiB)
  893. ctx := pds.promContexts.NewNamedContext(NetworkInsightsContextName)
  894. return source.NewFuture(source.DecodeNetZoneIngressGiBResult, ctx.QueryAtTime(queryNetZoneCostPerGiB, end))
  895. }
  896. func (pds *PrometheusMetricsQuerier) QueryNetRegionIngressGiB(start, end time.Time) *source.Future[source.NetRegionIngressGiBResult] {
  897. const queryName = "QueryNetRegionIngressGiB"
  898. const queryFmtIngNetRegionGiB = `sum(increase(kubecost_pod_network_ingress_bytes_total{internet="false", same_zone="false", same_region="false", %s}[%s:%dm])) by (pod_name, namespace, uid, %s) / 1024 / 1024 / 1024`
  899. cfg := pds.promConfig
  900. minsPerResolution := cfg.DataResolutionMinutes
  901. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  902. if durStr == "" {
  903. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  904. }
  905. queryNetRegionIngGiB := fmt.Sprintf(queryFmtIngNetRegionGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  906. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetRegionIngGiB)
  907. ctx := pds.promContexts.NewNamedContext(NetworkInsightsContextName)
  908. return source.NewFuture(source.DecodeNetRegionIngressGiBResult, ctx.QueryAtTime(queryNetRegionIngGiB, end))
  909. }
  910. func (pds *PrometheusMetricsQuerier) QueryNetInternetIngressGiB(start, end time.Time) *source.Future[source.NetInternetIngressGiBResult] {
  911. const queryName = "QueryNetInternetIngressGiB"
  912. const queryFmtNetIngInternetGiB = `sum(increase(kubecost_pod_network_ingress_bytes_total{internet="true", %s}[%s:%dm])) by (pod_name, namespace, uid, %s) / 1024 / 1024 / 1024`
  913. cfg := pds.promConfig
  914. minsPerResolution := cfg.DataResolutionMinutes
  915. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  916. if durStr == "" {
  917. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  918. }
  919. queryNetIngInternetGiB := fmt.Sprintf(queryFmtNetIngInternetGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  920. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetIngInternetGiB)
  921. ctx := pds.promContexts.NewNamedContext(NetworkInsightsContextName)
  922. return source.NewFuture(source.DecodeNetInternetIngressGiBResult, ctx.QueryAtTime(queryNetIngInternetGiB, end))
  923. }
  924. func (pds *PrometheusMetricsQuerier) QueryNetInternetServiceIngressGiB(start, end time.Time) *source.Future[source.NetInternetServiceIngressGiBResult] {
  925. const queryName = "QueryNetInternetServiceIngressGiB"
  926. const queryFmtIngNetInternetGiB = `sum(increase(kubecost_pod_network_ingress_bytes_total{internet="true", %s}[%s:%dm])) by (pod_name, namespace, service, uid, %s) / 1024 / 1024 / 1024`
  927. cfg := pds.promConfig
  928. minsPerResolution := cfg.DataResolutionMinutes
  929. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  930. if durStr == "" {
  931. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  932. }
  933. queryNetIngInternetGiB := fmt.Sprintf(queryFmtIngNetInternetGiB, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  934. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetIngInternetGiB)
  935. ctx := pds.promContexts.NewNamedContext(NetworkInsightsContextName)
  936. return source.NewFuture(source.DecodeNetInternetServiceIngressGiBResult, ctx.QueryAtTime(queryNetIngInternetGiB, end))
  937. }
  938. func (pds *PrometheusMetricsQuerier) QueryNetReceiveBytes(start, end time.Time) *source.Future[source.NetReceiveBytesResult] {
  939. const queryName = "QueryNetReceiveBytes"
  940. const queryFmtNetReceiveBytes = `sum(increase(container_network_receive_bytes_total{pod!="", %s}[%s:%dm])) by (pod_name, pod, namespace, uid, %s)`
  941. cfg := pds.promConfig
  942. minsPerResolution := cfg.DataResolutionMinutes
  943. durStr := pds.durationStringFor(start, end, minsPerResolution, true)
  944. if durStr == "" {
  945. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  946. }
  947. queryNetReceiveBytes := fmt.Sprintf(queryFmtNetReceiveBytes, cfg.ClusterFilter, durStr, minsPerResolution, cfg.ClusterLabel)
  948. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNetReceiveBytes)
  949. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  950. return source.NewFuture(source.DecodeNetReceiveBytesResult, ctx.QueryAtTime(queryNetReceiveBytes, end))
  951. }
  952. // Note: namespace_info is not currently emitted
  953. func (pds *PrometheusMetricsQuerier) QueryNamespaceUptime(start, end time.Time) *source.Future[source.UptimeResult] {
  954. const queryName = "QueryNamespaceUptime"
  955. const queryFmtNamespaceUptime = `avg(namespace_info{%s}) by (%s, uid)[%s:%dm]`
  956. cfg := pds.promConfig
  957. minsPerResolution := cfg.DataResolutionMinutes
  958. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  959. if durStr == "" {
  960. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  961. }
  962. queryNamespaceUptime := fmt.Sprintf(queryFmtNamespaceUptime, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  963. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryFmtNamespaceUptime)
  964. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  965. return source.NewFuture(source.DecodeUptimeResult, ctx.QueryAtTime(queryNamespaceUptime, end))
  966. }
  967. func (pds *PrometheusMetricsQuerier) QueryNamespaceLabels(start, end time.Time) *source.Future[source.NamespaceLabelsResult] {
  968. const queryName = "QueryNamespaceLabels"
  969. const queryFmtNamespaceLabels = `avg_over_time(kube_namespace_labels{%s}[%s])`
  970. cfg := pds.promConfig
  971. durStr := timeutil.DurationString(end.Sub(start))
  972. if durStr == "" {
  973. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  974. }
  975. queryNamespaceLabels := fmt.Sprintf(queryFmtNamespaceLabels, cfg.ClusterFilter, durStr)
  976. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNamespaceLabels)
  977. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  978. return source.NewFuture(source.DecodeNamespaceLabelsResult, ctx.QueryAtTime(queryNamespaceLabels, end))
  979. }
  980. func (pds *PrometheusMetricsQuerier) QueryNamespaceAnnotations(start, end time.Time) *source.Future[source.NamespaceAnnotationsResult] {
  981. const queryName = "QueryNamespaceAnnotations"
  982. const queryFmtNamespaceAnnotations = `avg_over_time(kube_namespace_annotations{%s}[%s])`
  983. cfg := pds.promConfig
  984. durStr := timeutil.DurationString(end.Sub(start))
  985. if durStr == "" {
  986. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  987. }
  988. queryNamespaceAnnotations := fmt.Sprintf(queryFmtNamespaceAnnotations, cfg.ClusterFilter, durStr)
  989. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryNamespaceAnnotations)
  990. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  991. return source.NewFuture(source.DecodeNamespaceAnnotationsResult, ctx.QueryAtTime(queryNamespaceAnnotations, end))
  992. }
  993. func (pds *PrometheusMetricsQuerier) QueryPodLabels(start, end time.Time) *source.Future[source.PodLabelsResult] {
  994. const queryName = "QueryPodLabels"
  995. const queryFmtPodLabels = `avg_over_time(kube_pod_labels{%s}[%s])`
  996. cfg := pds.promConfig
  997. durStr := timeutil.DurationString(end.Sub(start))
  998. if durStr == "" {
  999. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1000. }
  1001. queryPodLabels := fmt.Sprintf(queryFmtPodLabels, cfg.ClusterFilter, durStr)
  1002. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPodLabels)
  1003. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1004. return source.NewFuture(source.DecodePodLabelsResult, ctx.QueryAtTime(queryPodLabels, end))
  1005. }
  1006. func (pds *PrometheusMetricsQuerier) QueryPodAnnotations(start, end time.Time) *source.Future[source.PodAnnotationsResult] {
  1007. const queryName = "QueryPodAnnotations"
  1008. const queryFmtPodAnnotations = `avg_over_time(kube_pod_annotations{%s}[%s])`
  1009. cfg := pds.promConfig
  1010. durStr := timeutil.DurationString(end.Sub(start))
  1011. if durStr == "" {
  1012. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1013. }
  1014. queryPodAnnotations := fmt.Sprintf(queryFmtPodAnnotations, cfg.ClusterFilter, durStr)
  1015. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPodAnnotations)
  1016. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1017. return source.NewFuture(source.DecodePodAnnotationsResult, ctx.QueryAtTime(queryPodAnnotations, end))
  1018. }
  1019. func (pds *PrometheusMetricsQuerier) QueryServiceLabels(start, end time.Time) *source.Future[source.ServiceLabelsResult] {
  1020. const queryName = "QueryServiceLabels"
  1021. const queryFmtServiceLabels = `avg_over_time(service_selector_labels{%s}[%s])`
  1022. cfg := pds.promConfig
  1023. durStr := timeutil.DurationString(end.Sub(start))
  1024. if durStr == "" {
  1025. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1026. }
  1027. queryServiceLabels := fmt.Sprintf(queryFmtServiceLabels, cfg.ClusterFilter, durStr)
  1028. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryServiceLabels)
  1029. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1030. return source.NewFuture(source.DecodeServiceLabelsResult, ctx.QueryAtTime(queryServiceLabels, end))
  1031. }
  1032. func (pds *PrometheusMetricsQuerier) QueryDeploymentLabels(start, end time.Time) *source.Future[source.DeploymentLabelsResult] {
  1033. const queryName = "QueryDeploymentLabels"
  1034. const queryFmtDeploymentLabels = `avg_over_time(deployment_match_labels{%s}[%s])`
  1035. cfg := pds.promConfig
  1036. durStr := timeutil.DurationString(end.Sub(start))
  1037. if durStr == "" {
  1038. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1039. }
  1040. queryDeploymentLabels := fmt.Sprintf(queryFmtDeploymentLabels, cfg.ClusterFilter, durStr)
  1041. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryDeploymentLabels)
  1042. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1043. return source.NewFuture(source.DecodeDeploymentLabelsResult, ctx.QueryAtTime(queryDeploymentLabels, end))
  1044. }
  1045. func (pds *PrometheusMetricsQuerier) QueryStatefulSetLabels(start, end time.Time) *source.Future[source.StatefulSetLabelsResult] {
  1046. const queryName = "QueryStatefulSetLabels"
  1047. const queryFmtStatefulSetLabels = `avg_over_time(statefulSet_match_labels{%s}[%s])`
  1048. cfg := pds.promConfig
  1049. durStr := timeutil.DurationString(end.Sub(start))
  1050. if durStr == "" {
  1051. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1052. }
  1053. queryStatefulSetLabels := fmt.Sprintf(queryFmtStatefulSetLabels, cfg.ClusterFilter, durStr)
  1054. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryStatefulSetLabels)
  1055. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1056. return source.NewFuture(source.DecodeStatefulSetLabelsResult, ctx.QueryAtTime(queryStatefulSetLabels, end))
  1057. }
  1058. func (pds *PrometheusMetricsQuerier) QueryDaemonSetLabels(start, end time.Time) *source.Future[source.DaemonSetLabelsResult] {
  1059. const queryName = "QueryDaemonSetLabels"
  1060. const queryFmtDaemonSetLabels = `sum(avg_over_time(kube_pod_owner{owner_kind="DaemonSet", %s}[%s])) by (pod, owner_name, namespace, uid, %s)`
  1061. cfg := pds.promConfig
  1062. durStr := timeutil.DurationString(end.Sub(start))
  1063. if durStr == "" {
  1064. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1065. }
  1066. queryDaemonSetLabels := fmt.Sprintf(queryFmtDaemonSetLabels, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1067. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryDaemonSetLabels)
  1068. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1069. return source.NewFuture(source.DecodeDaemonSetLabelsResult, ctx.QueryAtTime(queryDaemonSetLabels, end))
  1070. }
  1071. func (pds *PrometheusMetricsQuerier) QueryJobLabels(start, end time.Time) *source.Future[source.JobLabelsResult] {
  1072. const queryName = "QueryJobLabels"
  1073. const queryFmtJobLabels = `sum(avg_over_time(kube_pod_owner{owner_kind="Job", %s}[%s])) by (pod, owner_name, namespace, uid, %s)`
  1074. cfg := pds.promConfig
  1075. durStr := timeutil.DurationString(end.Sub(start))
  1076. if durStr == "" {
  1077. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1078. }
  1079. queryJobLabels := fmt.Sprintf(queryFmtJobLabels, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1080. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryJobLabels)
  1081. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1082. return source.NewFuture(source.DecodeJobLabelsResult, ctx.QueryAtTime(queryJobLabels, end))
  1083. }
  1084. func (pds *PrometheusMetricsQuerier) QueryPodsWithReplicaSetOwner(start, end time.Time) *source.Future[source.PodsWithReplicaSetOwnerResult] {
  1085. const queryName = "QueryPodsWithReplicaSetOwner"
  1086. const queryFmtPodsWithReplicaSetOwner = `sum(avg_over_time(kube_pod_owner{owner_kind="ReplicaSet", %s}[%s])) by (pod, owner_name, namespace, uid, %s)`
  1087. cfg := pds.promConfig
  1088. durStr := timeutil.DurationString(end.Sub(start))
  1089. if durStr == "" {
  1090. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1091. }
  1092. queryPodsWithReplicaSetOwner := fmt.Sprintf(queryFmtPodsWithReplicaSetOwner, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1093. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryPodsWithReplicaSetOwner)
  1094. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1095. return source.NewFuture(source.DecodePodsWithReplicaSetOwnerResult, ctx.QueryAtTime(queryPodsWithReplicaSetOwner, end))
  1096. }
  1097. func (pds *PrometheusMetricsQuerier) QueryReplicaSetsWithoutOwners(start, end time.Time) *source.Future[source.ReplicaSetsWithoutOwnersResult] {
  1098. const queryName = "QueryReplicaSetsWithoutOwners"
  1099. const queryFmtReplicaSetsWithoutOwners = `avg(avg_over_time(kube_replicaset_owner{owner_kind="<none>", owner_name="<none>", %s}[%s])) by (replicaset, namespace, uid, %s)`
  1100. cfg := pds.promConfig
  1101. durStr := timeutil.DurationString(end.Sub(start))
  1102. if durStr == "" {
  1103. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1104. }
  1105. queryReplicaSetsWithoutOwners := fmt.Sprintf(queryFmtReplicaSetsWithoutOwners, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1106. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryReplicaSetsWithoutOwners)
  1107. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1108. return source.NewFuture(source.DecodeReplicaSetsWithoutOwnersResult, ctx.QueryAtTime(queryReplicaSetsWithoutOwners, end))
  1109. }
  1110. func (pds *PrometheusMetricsQuerier) QueryReplicaSetsWithRollout(start, end time.Time) *source.Future[source.ReplicaSetsWithRolloutResult] {
  1111. const queryName = "QueryReplicaSetsWithRollout"
  1112. const queryFmtReplicaSetsWithRolloutOwner = `avg(avg_over_time(kube_replicaset_owner{owner_kind="Rollout", %s}[%s])) by (replicaset, namespace, owner_kind, owner_name, uid, %s)`
  1113. cfg := pds.promConfig
  1114. durStr := timeutil.DurationString(end.Sub(start))
  1115. if durStr == "" {
  1116. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1117. }
  1118. queryReplicaSetsWithRolloutOwner := fmt.Sprintf(queryFmtReplicaSetsWithRolloutOwner, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1119. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryReplicaSetsWithRolloutOwner)
  1120. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1121. return source.NewFuture(source.DecodeReplicaSetsWithRolloutResult, ctx.QueryAtTime(queryReplicaSetsWithRolloutOwner, end))
  1122. }
  1123. // Note: The ResourceQuota metrics are _not_ emitted at the moment. Leaving the query implementations here in case we add metric emission later on.
  1124. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaUptime(start, end time.Time) *source.Future[source.UptimeResult] {
  1125. const queryName = "QueryResourceQuotaUptime"
  1126. const queryFmtResourceQuotaUptime = `avg(resourcequota_info{%s}) by (%s, uid)[%s:%dm]`
  1127. cfg := pds.promConfig
  1128. minsPerResolution := cfg.DataResolutionMinutes
  1129. durStr := pds.durationStringFor(start, end, minsPerResolution, false)
  1130. if durStr == "" {
  1131. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1132. }
  1133. queryResourceQuotaUptime := fmt.Sprintf(queryFmtResourceQuotaUptime, cfg.ClusterFilter, cfg.ClusterLabel, durStr, minsPerResolution)
  1134. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryFmtResourceQuotaUptime)
  1135. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1136. return source.NewFuture(source.DecodeUptimeResult, ctx.QueryAtTime(queryResourceQuotaUptime, end))
  1137. }
  1138. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecCPURequestAverage(start, end time.Time) *source.Future[source.ResourceQuotaSpecCPURequestAvgResult] {
  1139. const queryName = "QueryResourceQuotaSpecCPURequestAverage"
  1140. const queryFmtResourceQuotaSpecCPURequests = `avg(avg_over_time(resourcequota_spec_resource_requests{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1141. cfg := pds.promConfig
  1142. durStr := timeutil.DurationString(end.Sub(start))
  1143. if durStr == "" {
  1144. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1145. }
  1146. queryResourceQuotaSpecCPURequests := fmt.Sprintf(queryFmtResourceQuotaSpecCPURequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1147. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecCPURequests)
  1148. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1149. return source.NewFuture(source.DecodeResourceQuotaSpecCPURequestAvgResult, ctx.QueryAtTime(queryResourceQuotaSpecCPURequests, end))
  1150. }
  1151. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecCPURequestMax(start, end time.Time) *source.Future[source.ResourceQuotaSpecCPURequestMaxResult] {
  1152. const queryName = "QueryResourceQuotaSpecCPURequestMax"
  1153. const queryFmtResourceQuotaSpecCPURequests = `max(max_over_time(resourcequota_spec_resource_requests{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1154. cfg := pds.promConfig
  1155. durStr := timeutil.DurationString(end.Sub(start))
  1156. if durStr == "" {
  1157. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1158. }
  1159. queryResourceQuotaSpecCPURequests := fmt.Sprintf(queryFmtResourceQuotaSpecCPURequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1160. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecCPURequests)
  1161. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1162. return source.NewFuture(source.DecodeResourceQuotaSpecCPURequestMaxResult, ctx.QueryAtTime(queryResourceQuotaSpecCPURequests, end))
  1163. }
  1164. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecRAMRequestAverage(start, end time.Time) *source.Future[source.ResourceQuotaSpecRAMRequestAvgResult] {
  1165. const queryName = "QueryResourceQuotaSpecRAMRequestAverage"
  1166. const queryFmtResourceQuotaSpecRAMRequests = `avg(avg_over_time(resourcequota_spec_resource_requests{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1167. cfg := pds.promConfig
  1168. durStr := timeutil.DurationString(end.Sub(start))
  1169. if durStr == "" {
  1170. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1171. }
  1172. queryResourceQuotaSpecRAMRequests := fmt.Sprintf(queryFmtResourceQuotaSpecRAMRequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1173. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecRAMRequests)
  1174. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1175. return source.NewFuture(source.DecodeResourceQuotaSpecRAMRequestAvgResult, ctx.QueryAtTime(queryResourceQuotaSpecRAMRequests, end))
  1176. }
  1177. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecRAMRequestMax(start, end time.Time) *source.Future[source.ResourceQuotaSpecRAMRequestMaxResult] {
  1178. const queryName = "QueryResourceQuotaSpecRAMRequestMax"
  1179. const queryFmtResourceQuotaSpecRAMRequests = `max(max_over_time(resourcequota_spec_resource_requests{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1180. cfg := pds.promConfig
  1181. durStr := timeutil.DurationString(end.Sub(start))
  1182. if durStr == "" {
  1183. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1184. }
  1185. queryResourceQuotaSpecRAMRequests := fmt.Sprintf(queryFmtResourceQuotaSpecRAMRequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1186. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecRAMRequests)
  1187. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1188. return source.NewFuture(source.DecodeResourceQuotaSpecRAMRequestMaxResult, ctx.QueryAtTime(queryResourceQuotaSpecRAMRequests, end))
  1189. }
  1190. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecCPULimitAverage(start, end time.Time) *source.Future[source.ResourceQuotaSpecCPULimitAvgResult] {
  1191. const queryName = "QueryResourceQuotaSpecCPULimitAverage"
  1192. const queryFmtResourceQuotaSpecCPULimits = `avg(avg_over_time(resourcequota_spec_resource_limits{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1193. cfg := pds.promConfig
  1194. durStr := timeutil.DurationString(end.Sub(start))
  1195. if durStr == "" {
  1196. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1197. }
  1198. queryResourceQuotaSpecCPULimits := fmt.Sprintf(queryFmtResourceQuotaSpecCPULimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1199. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecCPULimits)
  1200. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1201. return source.NewFuture(source.DecodeResourceQuotaSpecCPULimitAvgResult, ctx.QueryAtTime(queryResourceQuotaSpecCPULimits, end))
  1202. }
  1203. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecCPULimitMax(start, end time.Time) *source.Future[source.ResourceQuotaSpecCPULimitMaxResult] {
  1204. const queryName = "QueryResourceQuotaSpecCPULimitMax"
  1205. const queryFmtResourceQuotaSpecCPULimits = `max(max_over_time(resourcequota_spec_resource_limits{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1206. cfg := pds.promConfig
  1207. durStr := timeutil.DurationString(end.Sub(start))
  1208. if durStr == "" {
  1209. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1210. }
  1211. queryResourceQuotaSpecCPULimits := fmt.Sprintf(queryFmtResourceQuotaSpecCPULimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1212. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecCPULimits)
  1213. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1214. return source.NewFuture(source.DecodeResourceQuotaSpecCPULimitMaxResult, ctx.QueryAtTime(queryResourceQuotaSpecCPULimits, end))
  1215. }
  1216. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecRAMLimitAverage(start, end time.Time) *source.Future[source.ResourceQuotaSpecRAMLimitAvgResult] {
  1217. const queryName = "QueryResourceQuotaSpecRAMLimitAverage"
  1218. const queryFmtResourceQuotaSpecRAMLimits = `avg(avg_over_time(resourcequota_spec_resource_limits{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1219. cfg := pds.promConfig
  1220. durStr := timeutil.DurationString(end.Sub(start))
  1221. if durStr == "" {
  1222. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1223. }
  1224. queryResourceQuotaSpecRAMLimits := fmt.Sprintf(queryFmtResourceQuotaSpecRAMLimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1225. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecRAMLimits)
  1226. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1227. return source.NewFuture(source.DecodeResourceQuotaSpecRAMLimitAvgResult, ctx.QueryAtTime(queryResourceQuotaSpecRAMLimits, end))
  1228. }
  1229. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaSpecRAMLimitMax(start, end time.Time) *source.Future[source.ResourceQuotaSpecRAMLimitMaxResult] {
  1230. const queryName = "QueryResourceQuotaSpecRAMLimitMax"
  1231. const queryFmtResourceQuotaSpecRAMLimits = `max(max_over_time(resourcequota_spec_resource_limits{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1232. cfg := pds.promConfig
  1233. durStr := timeutil.DurationString(end.Sub(start))
  1234. if durStr == "" {
  1235. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1236. }
  1237. queryResourceQuotaSpecRAMLimits := fmt.Sprintf(queryFmtResourceQuotaSpecRAMLimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1238. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaSpecRAMLimits)
  1239. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1240. return source.NewFuture(source.DecodeResourceQuotaSpecRAMLimitMaxResult, ctx.QueryAtTime(queryResourceQuotaSpecRAMLimits, end))
  1241. }
  1242. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedCPURequestAverage(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedCPURequestAvgResult] {
  1243. const queryName = "QueryResourceQuotaStatusUsedCPURequestAverage"
  1244. const queryFmtResourceQuotaStatusUsedCPURequests = `avg(avg_over_time(resourcequota_status_used_resource_requests{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1245. cfg := pds.promConfig
  1246. durStr := timeutil.DurationString(end.Sub(start))
  1247. if durStr == "" {
  1248. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1249. }
  1250. queryResourceQuotaStatusUsedCPURequests := fmt.Sprintf(queryFmtResourceQuotaStatusUsedCPURequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1251. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedCPURequests)
  1252. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1253. return source.NewFuture(source.DecodeResourceQuotaStatusUsedCPURequestAvgResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedCPURequests, end))
  1254. }
  1255. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedCPURequestMax(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedCPURequestMaxResult] {
  1256. const queryName = "QueryResourceQuotaStatusUsedCPURequestMax"
  1257. const queryFmtResourceQuotaStatusUsedCPURequests = `max(max_over_time(resourcequota_status_used_resource_requests{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1258. cfg := pds.promConfig
  1259. durStr := timeutil.DurationString(end.Sub(start))
  1260. if durStr == "" {
  1261. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1262. }
  1263. queryResourceQuotaStatusUsedCPURequests := fmt.Sprintf(queryFmtResourceQuotaStatusUsedCPURequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1264. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedCPURequests)
  1265. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1266. return source.NewFuture(source.DecodeResourceQuotaStatusUsedCPURequestMaxResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedCPURequests, end))
  1267. }
  1268. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedRAMRequestAverage(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedRAMRequestAvgResult] {
  1269. const queryName = "QueryResourceQuotaStatusUsedRAMRequestAverage"
  1270. const queryFmtResourceQuotaStatusUsedRAMRequests = `avg(avg_over_time(resourcequota_status_used_resource_requests{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1271. cfg := pds.promConfig
  1272. durStr := timeutil.DurationString(end.Sub(start))
  1273. if durStr == "" {
  1274. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1275. }
  1276. queryResourceQuotaStatusUsedRAMRequests := fmt.Sprintf(queryFmtResourceQuotaStatusUsedRAMRequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1277. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedRAMRequests)
  1278. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1279. return source.NewFuture(source.DecodeResourceQuotaStatusUsedRAMRequestAvgResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedRAMRequests, end))
  1280. }
  1281. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedRAMRequestMax(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedRAMRequestMaxResult] {
  1282. const queryName = "QueryResourceQuotaStatusUsedRAMRequestMax"
  1283. const queryFmtResourceQuotaStatusUsedRAMRequests = `max(max_over_time(resourcequota_status_used_resource_requests{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1284. cfg := pds.promConfig
  1285. durStr := timeutil.DurationString(end.Sub(start))
  1286. if durStr == "" {
  1287. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1288. }
  1289. queryResourceQuotaStatusUsedRAMRequests := fmt.Sprintf(queryFmtResourceQuotaStatusUsedRAMRequests, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1290. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedRAMRequests)
  1291. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1292. return source.NewFuture(source.DecodeResourceQuotaStatusUsedRAMRequestMaxResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedRAMRequests, end))
  1293. }
  1294. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedCPULimitAverage(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedCPULimitAvgResult] {
  1295. const queryName = "QueryResourceQuotaStatusUsedCPULimitAverage"
  1296. const queryFmtResourceQuotaStatusUsedCPULimits = `avg(avg_over_time(resourcequota_status_used_resource_limits{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1297. cfg := pds.promConfig
  1298. durStr := timeutil.DurationString(end.Sub(start))
  1299. if durStr == "" {
  1300. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1301. }
  1302. queryResourceQuotaStatusUsedCPULimits := fmt.Sprintf(queryFmtResourceQuotaStatusUsedCPULimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1303. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedCPULimits)
  1304. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1305. return source.NewFuture(source.DecodeResourceQuotaStatusUsedCPULimitAvgResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedCPULimits, end))
  1306. }
  1307. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedCPULimitMax(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedCPULimitMaxResult] {
  1308. const queryName = "QueryResourceQuotaStatusUsedCPULimitMax"
  1309. const queryFmtResourceQuotaStatusUsedCPULimits = `max(max_over_time(resourcequota_status_used_resource_limits{resource="cpu",unit="core", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1310. cfg := pds.promConfig
  1311. durStr := timeutil.DurationString(end.Sub(start))
  1312. if durStr == "" {
  1313. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1314. }
  1315. queryResourceQuotaStatusUsedCPULimits := fmt.Sprintf(queryFmtResourceQuotaStatusUsedCPULimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1316. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedCPULimits)
  1317. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1318. return source.NewFuture(source.DecodeResourceQuotaStatusUsedCPULimitMaxResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedCPULimits, end))
  1319. }
  1320. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedRAMLimitAverage(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedRAMLimitAvgResult] {
  1321. const queryName = "QueryResourceQuotaStatusUsedRAMLimitAverage"
  1322. const queryFmtResourceQuotaStatusUsedRAMLimits = `avg(avg_over_time(resourcequota_status_used_resource_limits{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1323. cfg := pds.promConfig
  1324. durStr := timeutil.DurationString(end.Sub(start))
  1325. if durStr == "" {
  1326. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1327. }
  1328. queryResourceQuotaStatusUsedRAMLimits := fmt.Sprintf(queryFmtResourceQuotaStatusUsedRAMLimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1329. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedRAMLimits)
  1330. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1331. return source.NewFuture(source.DecodeResourceQuotaStatusUsedRAMLimitAvgResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedRAMLimits, end))
  1332. }
  1333. func (pds *PrometheusMetricsQuerier) QueryResourceQuotaStatusUsedRAMLimitMax(start, end time.Time) *source.Future[source.ResourceQuotaStatusUsedRAMLimitMaxResult] {
  1334. const queryName = "QueryResourceQuotaStatusUsedRAMLimitMax"
  1335. const queryFmtResourceQuotaStatusUsedRAMLimits = `max(max_over_time(resourcequota_status_used_resource_limits{resource="memory",unit="byte", %s}[%s])) by (resourcequota, namespace, uid, %s)`
  1336. cfg := pds.promConfig
  1337. durStr := timeutil.DurationString(end.Sub(start))
  1338. if durStr == "" {
  1339. panic(fmt.Sprintf("failed to parse duration string passed to %s", queryName))
  1340. }
  1341. queryResourceQuotaStatusUsedRAMLimits := fmt.Sprintf(queryFmtResourceQuotaStatusUsedRAMLimits, cfg.ClusterFilter, durStr, cfg.ClusterLabel)
  1342. log.Debugf(PrometheusMetricsQueryLogFormat, queryName, end.Unix(), queryResourceQuotaStatusUsedRAMLimits)
  1343. ctx := pds.promContexts.NewNamedContext(KubeModelContextName)
  1344. return source.NewFuture(source.DecodeResourceQuotaStatusUsedRAMLimitMaxResult, ctx.QueryAtTime(queryResourceQuotaStatusUsedRAMLimits, end))
  1345. }
  1346. func (pds *PrometheusMetricsQuerier) QueryDataCoverage(limitDays int) (time.Time, time.Time, error) {
  1347. const (
  1348. queryName = "QueryDataCoverage"
  1349. queryFmtOldestSample = `min_over_time(timestamp(group(node_cpu_hourly_cost{%s}))[%s:%s])`
  1350. queryFmtNewestSample = `max_over_time(timestamp(group(node_cpu_hourly_cost{%s}))[%s:%s])`
  1351. )
  1352. cfg := pds.promConfig
  1353. minutesPerDuration := 60
  1354. dur := time.Duration(limitDays) * timeutil.Day
  1355. end := time.Now().UTC().Truncate(timeutil.Day).Add(timeutil.Day)
  1356. start := end.Add(-dur)
  1357. durStr := pds.durationStringFor(start, end, minutesPerDuration, false)
  1358. ctx := pds.promContexts.NewNamedContext(AllocationContextName)
  1359. queryOldest := fmt.Sprintf(queryFmtOldestSample, cfg.ClusterFilter, durStr, "1h")
  1360. log.Debugf("[Prometheus][%s[Oldest]][At Time: %d]: %s", queryName, end.Unix(), queryOldest)
  1361. resOldestFut := ctx.QueryAtTime(queryOldest, end)
  1362. resOldest, err := resOldestFut.Await()
  1363. if err != nil {
  1364. return time.Time{}, time.Time{}, fmt.Errorf("querying oldest sample: %w", err)
  1365. }
  1366. if len(resOldest) == 0 || len(resOldest[0].Values) == 0 {
  1367. // If node_cpu_hourly_cost metric is not available, fallback to a reasonable time range
  1368. // This prevents CSV export from failing when the metric doesn't exist yet
  1369. log.Warnf("QueryDataCoverage: node_cpu_hourly_cost metric not available, using fallback time range")
  1370. // Use a reasonable fallback: start from 1 day ago to account for metric collection delay
  1371. fallbackEnd := time.Now().UTC().Truncate(timeutil.Day)
  1372. fallbackStart := fallbackEnd.AddDate(0, 0, -1) // 1 day ago
  1373. return fallbackStart, fallbackEnd, nil
  1374. }
  1375. oldest := time.Unix(int64(resOldest[0].Values[0].Value), 0)
  1376. queryNewest := fmt.Sprintf(queryFmtNewestSample, cfg.ClusterFilter, durStr, "1h")
  1377. log.Debugf("[Prometheus][%s[Newest]][At Time: %d]: %s", queryName, end.Unix(), queryNewest)
  1378. resNewestFut := ctx.QueryAtTime(queryNewest, end)
  1379. resNewest, err := resNewestFut.Await()
  1380. if err != nil {
  1381. return time.Time{}, time.Time{}, fmt.Errorf("querying newest sample: %w", err)
  1382. }
  1383. if len(resNewest) == 0 || len(resNewest[0].Values) == 0 {
  1384. // If newest query fails but oldest succeeded, use oldest as both start and end
  1385. // This allows CSV export to proceed with at least some time range
  1386. log.Warnf("QueryDataCoverage: newest sample query returned no results, using oldest timestamp")
  1387. return oldest, oldest, nil
  1388. }
  1389. newest := time.Unix(int64(resNewest[0].Values[0].Value), 0)
  1390. return oldest, newest, nil
  1391. }
  1392. // durationStringFor simplifies the determination of query duration based on the version of prom and if the function
  1393. // in the query needs all data points in the vector it is provided or if it will extrapolate its own. Functions
  1394. // that extrapolate will add on another resolution if given a duration that is one resolution longer than the intended
  1395. // duration.
  1396. func (pds *PrometheusMetricsQuerier) durationStringFor(start, end time.Time, minsPerResolution int, extrapolated bool) string {
  1397. dur := end.Sub(start)
  1398. // If using a version of Prometheus where the resolution needs duration offset,
  1399. // we need to apply that here.
  1400. //
  1401. // E.g. avg(node_total_hourly_cost{}) by (node, provider_id)[60m:5m] with
  1402. // time=01:00:00 will return, for a node running the entire time, 12
  1403. // timestamps where the first is 00:05:00 and the last is 01:00:00.
  1404. // However, OpenCost expects for there to be 13 timestamps where the first
  1405. // begins at 00:00:00. To achieve this, we must modify our query to
  1406. // avg(node_total_hourly_cost{}) by (node, provider_id)[65m:5m]
  1407. if pds.promConfig.IsOffsetResolution && !extrapolated {
  1408. // increase the query time by the resolution
  1409. dur = dur + (time.Duration(minsPerResolution) * time.Minute)
  1410. }
  1411. return timeutil.DurationString(dur)
  1412. }