main.go 21 KB

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