main.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760
  1. package main
  2. import (
  3. "context"
  4. "encoding/json"
  5. "flag"
  6. "fmt"
  7. "net"
  8. "net/http"
  9. "os"
  10. "reflect"
  11. "strconv"
  12. "strings"
  13. "time"
  14. "k8s.io/klog"
  15. "github.com/julienschmidt/httprouter"
  16. costAnalyzerCloud "github.com/kubecost/cost-model/cloud"
  17. costModel "github.com/kubecost/cost-model/costmodel"
  18. prometheusClient "github.com/prometheus/client_golang/api"
  19. prometheusAPI "github.com/prometheus/client_golang/api/prometheus/v1"
  20. v1 "k8s.io/api/core/v1"
  21. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  22. "github.com/prometheus/client_golang/prometheus"
  23. "github.com/prometheus/client_golang/prometheus/promhttp"
  24. "k8s.io/apimachinery/pkg/fields"
  25. "k8s.io/client-go/kubernetes"
  26. "k8s.io/client-go/rest"
  27. "k8s.io/client-go/tools/cache"
  28. )
  29. const (
  30. prometheusServerEndpointEnvVar = "PROMETHEUS_SERVER_ENDPOINT"
  31. prometheusTroubleshootingEp = "http://docs.kubecost.com/custom-prom#troubleshoot"
  32. remoteEnabled = "REMOTE_WRITE_ENABLED"
  33. )
  34. var (
  35. // gitCommit is set by the build system
  36. gitCommit string
  37. )
  38. type Accesses struct {
  39. PrometheusClient prometheusClient.Client
  40. KubeClientSet kubernetes.Interface
  41. Cloud costAnalyzerCloud.Provider
  42. CPUPriceRecorder *prometheus.GaugeVec
  43. RAMPriceRecorder *prometheus.GaugeVec
  44. PersistentVolumePriceRecorder *prometheus.GaugeVec
  45. GPUPriceRecorder *prometheus.GaugeVec
  46. NodeTotalPriceRecorder *prometheus.GaugeVec
  47. RAMAllocationRecorder *prometheus.GaugeVec
  48. CPUAllocationRecorder *prometheus.GaugeVec
  49. GPUAllocationRecorder *prometheus.GaugeVec
  50. ContainerUptimeRecorder *prometheus.GaugeVec
  51. ServiceSelectorRecorder *prometheus.GaugeVec
  52. DeploymentSelectorRecorder *prometheus.GaugeVec
  53. Model *costModel.CostModel
  54. }
  55. type DataEnvelope struct {
  56. Code int `json:"code"`
  57. Status string `json:"status"`
  58. Data interface{} `json:"data"`
  59. Message string `json:"message,omitempty"`
  60. }
  61. func wrapData(data interface{}, err error) []byte {
  62. var resp []byte
  63. if err != nil {
  64. klog.V(1).Infof("Error returned to client: %s", err.Error())
  65. resp, _ = json.Marshal(&DataEnvelope{
  66. Code: 500,
  67. Status: "error",
  68. Message: err.Error(),
  69. Data: data,
  70. })
  71. } else {
  72. resp, _ = json.Marshal(&DataEnvelope{
  73. Code: 200,
  74. Status: "success",
  75. Data: data,
  76. })
  77. }
  78. return resp
  79. }
  80. // RefreshPricingData needs to be called when a new node joins the fleet, since we cache the relevant subsets of pricing data to avoid storing the whole thing.
  81. func (a *Accesses) RefreshPricingData(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  82. w.Header().Set("Content-Type", "application/json")
  83. w.Header().Set("Access-Control-Allow-Origin", "*")
  84. err := a.Cloud.DownloadPricingData()
  85. w.Write(wrapData(nil, err))
  86. }
  87. func filterFields(fields string, data map[string]*costModel.CostData) map[string]costModel.CostData {
  88. fs := strings.Split(fields, ",")
  89. fmap := make(map[string]bool)
  90. for _, f := range fs {
  91. fieldNameLower := strings.ToLower(f) // convert to go struct name by uppercasing first letter
  92. klog.V(1).Infof("to delete: %s", fieldNameLower)
  93. fmap[fieldNameLower] = true
  94. }
  95. filteredData := make(map[string]costModel.CostData)
  96. for cname, costdata := range data {
  97. s := reflect.TypeOf(*costdata)
  98. val := reflect.ValueOf(*costdata)
  99. costdata2 := costModel.CostData{}
  100. cd2 := reflect.New(reflect.Indirect(reflect.ValueOf(costdata2)).Type()).Elem()
  101. n := s.NumField()
  102. for i := 0; i < n; i++ {
  103. field := s.Field(i)
  104. value := val.Field(i)
  105. value2 := cd2.Field(i)
  106. if _, ok := fmap[strings.ToLower(field.Name)]; !ok {
  107. value2.Set(reflect.Value(value))
  108. }
  109. }
  110. filteredData[cname] = cd2.Interface().(costModel.CostData)
  111. }
  112. return filteredData
  113. }
  114. func (a *Accesses) CostDataModel(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  115. w.Header().Set("Content-Type", "application/json")
  116. w.Header().Set("Access-Control-Allow-Origin", "*")
  117. window := r.URL.Query().Get("timeWindow")
  118. offset := r.URL.Query().Get("offset")
  119. fields := r.URL.Query().Get("filterFields")
  120. namespace := r.URL.Query().Get("namespace")
  121. aggregation := r.URL.Query().Get("aggregation")
  122. aggregationSubField := r.URL.Query().Get("aggregationSubfield")
  123. if offset != "" {
  124. offset = "offset " + offset
  125. }
  126. data, err := a.Model.ComputeCostData(a.PrometheusClient, a.KubeClientSet, a.Cloud, window, offset, namespace)
  127. if aggregation != "" {
  128. agg := costModel.AggregateCostModel(data, aggregation, aggregationSubField)
  129. w.Write(wrapData(agg, nil))
  130. } else {
  131. if fields != "" {
  132. filteredData := filterFields(fields, data)
  133. w.Write(wrapData(filteredData, err))
  134. } else {
  135. w.Write(wrapData(data, err))
  136. }
  137. }
  138. }
  139. func (a *Accesses) ClusterCosts(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  140. w.Header().Set("Content-Type", "application/json")
  141. w.Header().Set("Access-Control-Allow-Origin", "*")
  142. window := r.URL.Query().Get("window")
  143. offset := r.URL.Query().Get("offset")
  144. if offset != "" {
  145. offset = "offset " + offset
  146. }
  147. data, err := costModel.ClusterCosts(a.PrometheusClient, a.Cloud, window, offset)
  148. w.Write(wrapData(data, err))
  149. }
  150. func (a *Accesses) ClusterCostsOverTime(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  151. w.Header().Set("Content-Type", "application/json")
  152. w.Header().Set("Access-Control-Allow-Origin", "*")
  153. start := r.URL.Query().Get("start")
  154. end := r.URL.Query().Get("end")
  155. window := r.URL.Query().Get("window")
  156. offset := r.URL.Query().Get("offset")
  157. if offset != "" {
  158. offset = "offset " + offset
  159. }
  160. data, err := costModel.ClusterCostsOverTime(a.PrometheusClient, a.Cloud, start, end, window, offset)
  161. w.Write(wrapData(data, err))
  162. }
  163. func (a *Accesses) AggregateCostModel(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  164. w.Header().Set("Content-Type", "application/json")
  165. w.Header().Set("Access-Control-Allow-Origin", "*")
  166. window := r.URL.Query().Get("window")
  167. offset := r.URL.Query().Get("offset")
  168. aggregation := r.URL.Query().Get("aggregation")
  169. namespace := r.URL.Query().Get("namespace")
  170. aggregationSubField := r.URL.Query().Get("aggregationSubfield")
  171. endTime := time.Now()
  172. if offset != "" {
  173. o, err := time.ParseDuration(offset)
  174. if err != nil {
  175. w.Write(wrapData(nil, err))
  176. return
  177. }
  178. endTime = endTime.Add(-1 * o)
  179. }
  180. d, err := time.ParseDuration(window)
  181. if err != nil {
  182. w.Write(wrapData(nil, err))
  183. return
  184. }
  185. startTime := endTime.Add(-1 * d)
  186. layout := "2006-01-02T15:04:05.000Z"
  187. start := startTime.Format(layout)
  188. end := startTime.Format(layout)
  189. data, err := a.Model.ComputeCostDataRange(a.PrometheusClient, a.KubeClientSet, a.Cloud, start, end, "1h", namespace)
  190. if err != nil {
  191. w.Write(wrapData(nil, err))
  192. return
  193. }
  194. if aggregation != "" {
  195. agg := costModel.AggregateCostModel(data, aggregation, aggregationSubField)
  196. w.Write(wrapData(agg, nil))
  197. }
  198. }
  199. func (a *Accesses) CostDataModelRange(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  200. w.Header().Set("Content-Type", "application/json")
  201. w.Header().Set("Access-Control-Allow-Origin", "*")
  202. start := r.URL.Query().Get("start")
  203. end := r.URL.Query().Get("end")
  204. window := r.URL.Query().Get("window")
  205. fields := r.URL.Query().Get("filterFields")
  206. namespace := r.URL.Query().Get("namespace")
  207. aggregation := r.URL.Query().Get("aggregation")
  208. aggregationSubField := r.URL.Query().Get("aggregationSubfield")
  209. data, err := a.Model.ComputeCostDataRange(a.PrometheusClient, a.KubeClientSet, a.Cloud, start, end, window, namespace)
  210. if err != nil {
  211. w.Write(wrapData(nil, err))
  212. }
  213. if aggregation != "" {
  214. agg := costModel.AggregateCostModel(data, aggregation, aggregationSubField)
  215. w.Write(wrapData(agg, nil))
  216. } else {
  217. if fields != "" {
  218. filteredData := filterFields(fields, data)
  219. w.Write(wrapData(filteredData, err))
  220. } else {
  221. w.Write(wrapData(data, err))
  222. }
  223. }
  224. }
  225. // CostDataModelRangeLarge is experimental multi-cluster and long-term data storage in SQL support.
  226. func (a *Accesses) CostDataModelRangeLarge(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  227. w.Header().Set("Content-Type", "application/json")
  228. w.Header().Set("Access-Control-Allow-Origin", "*")
  229. startString := r.URL.Query().Get("start")
  230. endString := r.URL.Query().Get("end")
  231. windowString := r.URL.Query().Get("window")
  232. layout := "2006-01-02T15:04:05.000Z"
  233. var start time.Time
  234. var end time.Time
  235. var err error
  236. if windowString == "" {
  237. windowString = "1h"
  238. }
  239. if startString != "" {
  240. start, err = time.Parse(layout, startString)
  241. if err != nil {
  242. klog.V(1).Infof("Error parsing time " + startString + ". Error: " + err.Error())
  243. w.Write(wrapData(nil, err))
  244. }
  245. } else {
  246. window, err := time.ParseDuration(windowString)
  247. if err != nil {
  248. w.Write(wrapData(nil, fmt.Errorf("Invalid duration '%s'", windowString)))
  249. }
  250. start = time.Now().Add(-2 * window)
  251. }
  252. if endString != "" {
  253. end, err = time.Parse(layout, endString)
  254. if err != nil {
  255. klog.V(1).Infof("Error parsing time " + endString + ". Error: " + err.Error())
  256. w.Write(wrapData(nil, err))
  257. }
  258. } else {
  259. end = time.Now()
  260. }
  261. remoteLayout := "2006-01-02T15:04:05Z"
  262. remoteStartStr := start.Format(remoteLayout)
  263. remoteEndStr := end.Format(remoteLayout)
  264. klog.V(1).Infof("Using remote database for query from %s to %s with window %s", startString, endString, windowString)
  265. data, err := costModel.CostDataRangeFromSQL("", "", windowString, remoteStartStr, remoteEndStr)
  266. w.Write(wrapData(data, err))
  267. }
  268. func (a *Accesses) OutofClusterCosts(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  269. w.Header().Set("Content-Type", "application/json")
  270. w.Header().Set("Access-Control-Allow-Origin", "*")
  271. start := r.URL.Query().Get("start")
  272. end := r.URL.Query().Get("end")
  273. aggregator := r.URL.Query().Get("aggregator")
  274. data, err := a.Cloud.ExternalAllocations(start, end, aggregator)
  275. w.Write(wrapData(data, err))
  276. }
  277. func (p *Accesses) GetAllNodePricing(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  278. w.Header().Set("Content-Type", "application/json")
  279. w.Header().Set("Access-Control-Allow-Origin", "*")
  280. data, err := p.Cloud.AllNodePricing()
  281. w.Write(wrapData(data, err))
  282. }
  283. func (p *Accesses) GetConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  284. w.Header().Set("Content-Type", "application/json")
  285. w.Header().Set("Access-Control-Allow-Origin", "*")
  286. data, err := p.Cloud.GetConfig()
  287. w.Write(wrapData(data, err))
  288. }
  289. func (p *Accesses) UpdateSpotInfoConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  290. w.Header().Set("Content-Type", "application/json")
  291. w.Header().Set("Access-Control-Allow-Origin", "*")
  292. data, err := p.Cloud.UpdateConfig(r.Body, costAnalyzerCloud.SpotInfoUpdateType)
  293. if err != nil {
  294. w.Write(wrapData(data, err))
  295. return
  296. }
  297. w.Write(wrapData(data, err))
  298. err = p.Cloud.DownloadPricingData()
  299. if err != nil {
  300. klog.V(1).Infof("Error redownloading data on config update: %s", err.Error())
  301. }
  302. return
  303. }
  304. func (p *Accesses) UpdateAthenaInfoConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  305. w.Header().Set("Content-Type", "application/json")
  306. w.Header().Set("Access-Control-Allow-Origin", "*")
  307. data, err := p.Cloud.UpdateConfig(r.Body, costAnalyzerCloud.AthenaInfoUpdateType)
  308. if err != nil {
  309. w.Write(wrapData(data, err))
  310. return
  311. }
  312. w.Write(wrapData(data, err))
  313. return
  314. }
  315. func (p *Accesses) UpdateBigQueryInfoConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  316. w.Header().Set("Content-Type", "application/json")
  317. w.Header().Set("Access-Control-Allow-Origin", "*")
  318. data, err := p.Cloud.UpdateConfig(r.Body, costAnalyzerCloud.BigqueryUpdateType)
  319. if err != nil {
  320. w.Write(wrapData(data, err))
  321. return
  322. }
  323. w.Write(wrapData(data, err))
  324. return
  325. }
  326. func (p *Accesses) UpdateConfigByKey(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  327. w.Header().Set("Content-Type", "application/json")
  328. w.Header().Set("Access-Control-Allow-Origin", "*")
  329. data, err := p.Cloud.UpdateConfig(r.Body, "")
  330. if err != nil {
  331. w.Write(wrapData(data, err))
  332. return
  333. }
  334. w.Write(wrapData(data, err))
  335. return
  336. }
  337. func (p *Accesses) ManagementPlatform(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  338. w.Header().Set("Content-Type", "application/json")
  339. w.Header().Set("Access-Control-Allow-Origin", "*")
  340. data, err := p.Cloud.GetManagementPlatform()
  341. if err != nil {
  342. w.Write(wrapData(data, err))
  343. return
  344. }
  345. w.Write(wrapData(data, err))
  346. return
  347. }
  348. func (p *Accesses) ClusterInfo(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  349. w.Header().Set("Content-Type", "application/json")
  350. w.Header().Set("Access-Control-Allow-Origin", "*")
  351. data, err := p.Cloud.ClusterInfo()
  352. w.Write(wrapData(data, err))
  353. }
  354. func Healthz(w http.ResponseWriter, _ *http.Request, _ httprouter.Params) {
  355. w.WriteHeader(200)
  356. w.Header().Set("Content-Length", "0")
  357. w.Header().Set("Content-Type", "text/plain")
  358. }
  359. func (p *Accesses) GetPrometheusMetadata(w http.ResponseWriter, _ *http.Request, _ httprouter.Params) {
  360. w.Header().Set("Content-Type", "application/json")
  361. w.Header().Set("Access-Control-Allow-Origin", "*")
  362. w.Write(wrapData(costModel.ValidatePrometheus(p.PrometheusClient)))
  363. }
  364. func (p *Accesses) ContainerUptimes(w http.ResponseWriter, _ *http.Request, _ httprouter.Params) {
  365. w.Header().Set("Content-Type", "application/json")
  366. w.Header().Set("Access-Control-Allow-Origin", "*")
  367. res, err := costModel.ComputeUptimes(p.PrometheusClient)
  368. w.Write(wrapData(res, err))
  369. }
  370. func (a *Accesses) recordPrices() {
  371. go func() {
  372. containerSeen := make(map[string]bool)
  373. nodeSeen := make(map[string]bool)
  374. pvSeen := make(map[string]bool)
  375. getKeyFromLabelStrings := func(labels ...string) string {
  376. return strings.Join(labels, ",")
  377. }
  378. getLabelStringsFromKey := func(key string) []string {
  379. return strings.Split(key, ",")
  380. }
  381. for {
  382. klog.V(4).Info("Recording prices...")
  383. podlist := a.Model.Controller.GetAll()
  384. podStatus := make(map[string]v1.PodPhase)
  385. for _, pod := range podlist {
  386. podStatus[pod.Name] = pod.Status.Phase
  387. }
  388. data, err := a.Model.ComputeCostData(a.PrometheusClient, a.KubeClientSet, a.Cloud, "2m", "", "")
  389. if err != nil {
  390. klog.V(1).Info("Error in price recording: " + err.Error())
  391. // zero the for loop so the time.Sleep will still work
  392. data = map[string]*costModel.CostData{}
  393. }
  394. for _, costs := range data {
  395. nodeName := costs.NodeName
  396. node := costs.NodeData
  397. if node == nil {
  398. klog.V(4).Infof("Skipping Node \"%s\" due to missing Node Data costs", nodeName)
  399. continue
  400. }
  401. cpuCost, _ := strconv.ParseFloat(node.VCPUCost, 64)
  402. cpu, _ := strconv.ParseFloat(node.VCPU, 64)
  403. ramCost, _ := strconv.ParseFloat(node.RAMCost, 64)
  404. ram, _ := strconv.ParseFloat(node.RAMBytes, 64)
  405. gpu, _ := strconv.ParseFloat(node.GPU, 64)
  406. gpuCost, _ := strconv.ParseFloat(node.GPUCost, 64)
  407. totalCost := cpu*cpuCost + ramCost*(ram/1024/1024/1024) + gpu*gpuCost
  408. if costs.PVCData != nil {
  409. for _, pvc := range costs.PVCData {
  410. if pvc.Volume != nil {
  411. pvCost, _ := strconv.ParseFloat(pvc.Volume.Cost, 64)
  412. a.PersistentVolumePriceRecorder.WithLabelValues(pvc.VolumeName, pvc.VolumeName).Set(pvCost)
  413. }
  414. }
  415. }
  416. a.CPUPriceRecorder.WithLabelValues(nodeName, nodeName).Set(cpuCost)
  417. a.RAMPriceRecorder.WithLabelValues(nodeName, nodeName).Set(ramCost)
  418. a.GPUPriceRecorder.WithLabelValues(nodeName, nodeName).Set(gpuCost)
  419. a.NodeTotalPriceRecorder.WithLabelValues(nodeName, nodeName).Set(totalCost)
  420. labelKey := getKeyFromLabelStrings(nodeName, nodeName)
  421. nodeSeen[labelKey] = true
  422. namespace := costs.Namespace
  423. podName := costs.PodName
  424. containerName := costs.Name
  425. if len(costs.RAMAllocation) > 0 {
  426. a.RAMAllocationRecorder.WithLabelValues(namespace, podName, containerName, nodeName, nodeName).Set(costs.RAMAllocation[0].Value)
  427. }
  428. if len(costs.CPUAllocation) > 0 {
  429. a.CPUAllocationRecorder.WithLabelValues(namespace, podName, containerName, nodeName, nodeName).Set(costs.CPUAllocation[0].Value)
  430. }
  431. if len(costs.GPUReq) > 0 {
  432. // allocation here is set to the request because shared GPU usage not yet supported.
  433. a.GPUAllocationRecorder.WithLabelValues(namespace, podName, containerName, nodeName, nodeName).Set(costs.GPUReq[0].Value)
  434. }
  435. labelKey = getKeyFromLabelStrings(namespace, podName, containerName, nodeName, nodeName)
  436. if podStatus[podName] == v1.PodRunning { // Only report data for current pods
  437. containerSeen[labelKey] = true
  438. } else {
  439. klog.Infof("Container %s not running", labelKey)
  440. containerSeen[labelKey] = false
  441. }
  442. storageClasses, _ := a.KubeClientSet.StorageV1().StorageClasses().List(metav1.ListOptions{})
  443. storageClassMap := make(map[string]map[string]string)
  444. for _, storageClass := range storageClasses.Items {
  445. params := storageClass.Parameters
  446. storageClassMap[storageClass.ObjectMeta.Name] = params
  447. if storageClass.GetAnnotations()["storageclass.kubernetes.io/is-default-class"] == "true" || storageClass.GetAnnotations()["storageclass.beta.kubernetes.io/is-default-class"] == "true" {
  448. storageClassMap["default"] = params
  449. storageClassMap[""] = params
  450. }
  451. }
  452. pvs, _ := a.KubeClientSet.CoreV1().PersistentVolumes().List(metav1.ListOptions{})
  453. for _, pv := range pvs.Items {
  454. parameters, ok := storageClassMap[pv.Spec.StorageClassName]
  455. if !ok {
  456. klog.V(4).Infof("Unable to find parameters for storage class \"%s\". Does pv \"%s\" have a storageClassName?", pv.Spec.StorageClassName, pv.Name)
  457. }
  458. cacPv := &costAnalyzerCloud.PV{
  459. Class: pv.Spec.StorageClassName,
  460. Region: pv.Labels[v1.LabelZoneRegion],
  461. Parameters: parameters,
  462. }
  463. costModel.GetPVCost(cacPv, &pv, a.Cloud)
  464. c, _ := strconv.ParseFloat(cacPv.Cost, 64)
  465. a.PersistentVolumePriceRecorder.WithLabelValues(pv.Name, pv.Name).Set(c)
  466. labelKey := getKeyFromLabelStrings(pv.Name, pv.Name)
  467. pvSeen[labelKey] = true
  468. }
  469. containerUptime, _ := costModel.ComputeUptimes(a.PrometheusClient)
  470. for key, uptime := range containerUptime {
  471. container, _ := costModel.NewContainerMetricFromKey(key)
  472. a.ContainerUptimeRecorder.WithLabelValues(container.Namespace, container.PodName, container.ContainerName).Set(uptime)
  473. }
  474. }
  475. for labelString, seen := range nodeSeen {
  476. if !seen {
  477. labels := getLabelStringsFromKey(labelString)
  478. a.NodeTotalPriceRecorder.DeleteLabelValues(labels...)
  479. a.CPUPriceRecorder.DeleteLabelValues(labels...)
  480. a.GPUPriceRecorder.DeleteLabelValues(labels...)
  481. a.RAMPriceRecorder.DeleteLabelValues(labels...)
  482. delete(nodeSeen, labelString)
  483. }
  484. nodeSeen[labelString] = false
  485. }
  486. for labelString, seen := range containerSeen {
  487. if !seen {
  488. labels := getLabelStringsFromKey(labelString)
  489. a.RAMAllocationRecorder.DeleteLabelValues(labels...)
  490. a.CPUAllocationRecorder.DeleteLabelValues(labels...)
  491. a.GPUAllocationRecorder.DeleteLabelValues(labels...)
  492. a.ContainerUptimeRecorder.DeleteLabelValues(labels...)
  493. delete(containerSeen, labelString)
  494. }
  495. containerSeen[labelString] = false
  496. }
  497. for labelString, seen := range pvSeen {
  498. if !seen {
  499. labels := getLabelStringsFromKey(labelString)
  500. a.PersistentVolumePriceRecorder.DeleteLabelValues(labels...)
  501. delete(pvSeen, labelString)
  502. }
  503. pvSeen[labelString] = false
  504. }
  505. time.Sleep(time.Minute)
  506. }
  507. }()
  508. }
  509. func main() {
  510. klog.InitFlags(nil)
  511. flag.Set("v", "3")
  512. flag.Parse()
  513. klog.V(1).Infof("Starting cost-model (git commit \"%s\")", gitCommit)
  514. address := os.Getenv(prometheusServerEndpointEnvVar)
  515. if address == "" {
  516. klog.Fatalf("No address for prometheus set in $%s. Aborting.", prometheusServerEndpointEnvVar)
  517. }
  518. var LongTimeoutRoundTripper http.RoundTripper = &http.Transport{ // may be necessary for long prometheus queries. TODO: make this configurable
  519. Proxy: http.ProxyFromEnvironment,
  520. DialContext: (&net.Dialer{
  521. Timeout: 120 * time.Second,
  522. KeepAlive: 120 * time.Second,
  523. }).DialContext,
  524. TLSHandshakeTimeout: 10 * time.Second,
  525. }
  526. pc := prometheusClient.Config{
  527. Address: address,
  528. RoundTripper: LongTimeoutRoundTripper,
  529. }
  530. promCli, _ := prometheusClient.NewClient(pc)
  531. api := prometheusAPI.NewAPI(promCli)
  532. _, err := api.Config(context.Background())
  533. if err != nil {
  534. klog.Fatalf("No valid prometheus config file at %s. Error: %s . Troubleshooting help available at: %s", address, err.Error(), prometheusTroubleshootingEp)
  535. }
  536. klog.V(1).Info("Success: retrieved a prometheus config file from: " + address)
  537. _, err = costModel.ValidatePrometheus(promCli)
  538. if err != nil {
  539. klog.Fatalf("Failed to query prometheus at %s. Error: %s . Troubleshooting help available at: %s", address, err.Error(), prometheusTroubleshootingEp)
  540. }
  541. klog.V(1).Info("Success: retrieved the 'up' query against prometheus at: " + address)
  542. // Kubernetes API setup
  543. kc, err := rest.InClusterConfig()
  544. if err != nil {
  545. panic(err.Error())
  546. }
  547. kubeClientset, err := kubernetes.NewForConfig(kc)
  548. if err != nil {
  549. panic(err.Error())
  550. }
  551. cloudProviderKey := os.Getenv("CLOUD_PROVIDER_API_KEY")
  552. cloudProvider, err := costAnalyzerCloud.NewProvider(kubeClientset, cloudProviderKey)
  553. if err != nil {
  554. panic(err.Error())
  555. }
  556. cpuGv := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  557. Name: "node_cpu_hourly_cost",
  558. Help: "node_cpu_hourly_cost hourly cost for each cpu on this node",
  559. }, []string{"instance", "node"})
  560. ramGv := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  561. Name: "node_ram_hourly_cost",
  562. Help: "node_ram_hourly_cost hourly cost for each gb of ram on this node",
  563. }, []string{"instance", "node"})
  564. gpuGv := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  565. Name: "node_gpu_hourly_cost",
  566. Help: "node_gpu_hourly_cost hourly cost for each gpu on this node",
  567. }, []string{"instance", "node"})
  568. totalGv := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  569. Name: "node_total_hourly_cost",
  570. Help: "node_total_hourly_cost Total node cost per hour",
  571. }, []string{"instance", "node"})
  572. pvGv := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  573. Name: "pv_hourly_cost",
  574. Help: "pv_hourly_cost Cost per GB per hour on a persistent disk",
  575. }, []string{"volumename", "persistentvolume"})
  576. RAMAllocation := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  577. Name: "container_memory_allocation_bytes",
  578. Help: "container_memory_allocation_bytes Bytes of RAM used",
  579. }, []string{"namespace", "pod", "container", "instance", "node"})
  580. CPUAllocation := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  581. Name: "container_cpu_allocation",
  582. Help: "container_cpu_allocation Percent of a single CPU used in a minute",
  583. }, []string{"namespace", "pod", "container", "instance", "node"})
  584. GPUAllocation := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  585. Name: "container_gpu_allocation",
  586. Help: "container_gpu_allocation GPU used",
  587. }, []string{"namespace", "pod", "container", "instance", "node"})
  588. ContainerUptimeRecorder := prometheus.NewGaugeVec(prometheus.GaugeOpts{
  589. Name: "container_uptime_seconds",
  590. Help: "container_uptime_seconds Seconds a container has been running",
  591. }, []string{"namespace", "pod", "container"})
  592. prometheus.MustRegister(cpuGv)
  593. prometheus.MustRegister(ramGv)
  594. prometheus.MustRegister(gpuGv)
  595. prometheus.MustRegister(totalGv)
  596. prometheus.MustRegister(pvGv)
  597. prometheus.MustRegister(RAMAllocation)
  598. prometheus.MustRegister(CPUAllocation)
  599. prometheus.MustRegister(ContainerUptimeRecorder)
  600. prometheus.MustRegister(costModel.ServiceCollector{
  601. KubeClientSet: kubeClientset,
  602. })
  603. prometheus.MustRegister(costModel.DeploymentCollector{
  604. KubeClientSet: kubeClientset,
  605. })
  606. podCache := cache.NewListWatchFromClient(kubeClientset.CoreV1().RESTClient(), "pods", "", fields.Everything())
  607. a := Accesses{
  608. PrometheusClient: promCli,
  609. KubeClientSet: kubeClientset,
  610. Cloud: cloudProvider,
  611. CPUPriceRecorder: cpuGv,
  612. RAMPriceRecorder: ramGv,
  613. GPUPriceRecorder: gpuGv,
  614. NodeTotalPriceRecorder: totalGv,
  615. RAMAllocationRecorder: RAMAllocation,
  616. CPUAllocationRecorder: CPUAllocation,
  617. GPUAllocationRecorder: GPUAllocation,
  618. ContainerUptimeRecorder: ContainerUptimeRecorder,
  619. PersistentVolumePriceRecorder: pvGv,
  620. Model: costModel.NewCostModel(podCache),
  621. }
  622. remoteEnabled := os.Getenv(remoteEnabled)
  623. if remoteEnabled == "true" {
  624. info, err := cloudProvider.ClusterInfo()
  625. klog.Infof("Saving cluster with id:'%s', and name:'%s' to durable storage", info["id"], info["name"])
  626. if err != nil {
  627. klog.Infof("Error saving cluster id %s", err.Error())
  628. }
  629. _, _, err = costAnalyzerCloud.GetOrCreateClusterMeta(info["id"], info["name"])
  630. if err != nil {
  631. klog.Infof("Unable to set cluster id '%s' for cluster '%s', %s", info["id"], info["name"], err.Error())
  632. }
  633. }
  634. err = a.Cloud.DownloadPricingData()
  635. if err != nil {
  636. klog.V(1).Info("Failed to download pricing data: " + err.Error())
  637. }
  638. a.recordPrices()
  639. router := httprouter.New()
  640. router.GET("/costDataModel", a.CostDataModel)
  641. router.GET("/costDataModelRange", a.CostDataModelRange)
  642. router.GET("/costDataModelRangeLarge", a.CostDataModelRangeLarge)
  643. router.GET("/outOfClusterCosts", a.OutofClusterCosts)
  644. router.GET("/allNodePricing", a.GetAllNodePricing)
  645. router.GET("/healthz", Healthz)
  646. router.GET("/getConfigs", a.GetConfigs)
  647. router.POST("/refreshPricing", a.RefreshPricingData)
  648. router.POST("/updateSpotInfoConfigs", a.UpdateSpotInfoConfigs)
  649. router.POST("/updateAthenaInfoConfigs", a.UpdateAthenaInfoConfigs)
  650. router.POST("/updateBigQueryInfoConfigs", a.UpdateBigQueryInfoConfigs)
  651. router.POST("/updateConfigByKey", a.UpdateConfigByKey)
  652. router.GET("/clusterCostsOverTime", a.ClusterCostsOverTime)
  653. router.GET("/clusterCosts", a.ClusterCosts)
  654. router.GET("/validatePrometheus", a.GetPrometheusMetadata)
  655. router.GET("/managementPlatform", a.ManagementPlatform)
  656. router.GET("/clusterInfo", a.ClusterInfo)
  657. router.GET("/containerUptimes", a.ContainerUptimes)
  658. router.GET("/aggregatedCostModel", a.AggregateCostModel)
  659. rootMux := http.NewServeMux()
  660. rootMux.Handle("/", router)
  661. rootMux.Handle("/metrics", promhttp.Handler())
  662. klog.Fatal(http.ListenAndServe(":9003", rootMux))
  663. }