router.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649
  1. package costmodel
  2. import (
  3. "context"
  4. "encoding/base64"
  5. "fmt"
  6. "net/http"
  7. "os"
  8. "reflect"
  9. "strconv"
  10. "strings"
  11. "sync"
  12. "time"
  13. "github.com/opencost/opencost/core/pkg/kubeconfig"
  14. "github.com/opencost/opencost/core/pkg/nodestats"
  15. "github.com/opencost/opencost/core/pkg/protocol"
  16. "github.com/opencost/opencost/core/pkg/source"
  17. "github.com/opencost/opencost/core/pkg/storage"
  18. "github.com/opencost/opencost/core/pkg/util/retry"
  19. "github.com/opencost/opencost/core/pkg/util/timeutil"
  20. "github.com/opencost/opencost/core/pkg/version"
  21. "github.com/opencost/opencost/pkg/cloud/aws"
  22. cloudconfig "github.com/opencost/opencost/pkg/cloud/config"
  23. "github.com/opencost/opencost/pkg/cloud/gcp"
  24. "github.com/opencost/opencost/pkg/cloud/provider"
  25. "github.com/opencost/opencost/pkg/cloudcost"
  26. "github.com/opencost/opencost/pkg/config"
  27. "github.com/opencost/opencost/pkg/customcost"
  28. "github.com/opencost/opencost/pkg/metrics"
  29. "github.com/opencost/opencost/pkg/util/watcher"
  30. "github.com/julienschmidt/httprouter"
  31. "github.com/opencost/opencost/core/pkg/clustercache"
  32. "github.com/opencost/opencost/core/pkg/clusters"
  33. sysenv "github.com/opencost/opencost/core/pkg/env"
  34. "github.com/opencost/opencost/core/pkg/log"
  35. "github.com/opencost/opencost/core/pkg/util/json"
  36. "github.com/opencost/opencost/modules/collector-source/pkg/collector"
  37. "github.com/opencost/opencost/modules/prometheus-source/pkg/prom"
  38. "github.com/opencost/opencost/pkg/cloud/azure"
  39. "github.com/opencost/opencost/pkg/cloud/models"
  40. clusterc "github.com/opencost/opencost/pkg/clustercache"
  41. "github.com/opencost/opencost/pkg/env"
  42. metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  43. "github.com/patrickmn/go-cache"
  44. "k8s.io/client-go/kubernetes"
  45. )
  46. const (
  47. RFC3339Milli = "2006-01-02T15:04:05.000Z"
  48. maxCacheMinutes1d = 11
  49. maxCacheMinutes2d = 17
  50. maxCacheMinutes7d = 37
  51. maxCacheMinutes30d = 137
  52. CustomPricingSetting = "CustomPricing"
  53. DiscountSetting = "Discount"
  54. )
  55. var (
  56. // gitCommit is set by the build system
  57. gitCommit string
  58. proto = protocol.HTTP()
  59. )
  60. // Accesses defines a singleton application instance, providing access to
  61. // Prometheus, Kubernetes, the cloud provider, and caches.
  62. type Accesses struct {
  63. DataSource source.OpenCostDataSource
  64. KubeClientSet kubernetes.Interface
  65. ClusterCache clustercache.ClusterCache
  66. ClusterMap clusters.ClusterMap
  67. CloudProvider models.Provider
  68. ConfigFileManager *config.ConfigFileManager
  69. ClusterInfoProvider clusters.ClusterInfoProvider
  70. Model *CostModel
  71. MetricsEmitter *CostModelMetricsEmitter
  72. // SettingsCache stores current state of app settings
  73. SettingsCache *cache.Cache
  74. // settingsSubscribers tracks channels through which changes to different
  75. // settings will be published in a pub/sub model
  76. settingsSubscribers map[string][]chan string
  77. settingsMutex sync.Mutex
  78. }
  79. func filterFields(fields string, data map[string]*CostData) map[string]CostData {
  80. fs := strings.Split(fields, ",")
  81. fmap := make(map[string]bool)
  82. for _, f := range fs {
  83. fieldNameLower := strings.ToLower(f) // convert to go struct name by uppercasing first letter
  84. log.Debugf("to delete: %s", fieldNameLower)
  85. fmap[fieldNameLower] = true
  86. }
  87. filteredData := make(map[string]CostData)
  88. for cname, costdata := range data {
  89. s := reflect.TypeOf(*costdata)
  90. val := reflect.ValueOf(*costdata)
  91. costdata2 := CostData{}
  92. cd2 := reflect.New(reflect.Indirect(reflect.ValueOf(costdata2)).Type()).Elem()
  93. n := s.NumField()
  94. for i := 0; i < n; i++ {
  95. field := s.Field(i)
  96. value := val.Field(i)
  97. value2 := cd2.Field(i)
  98. if _, ok := fmap[strings.ToLower(field.Name)]; !ok {
  99. value2.Set(reflect.Value(value))
  100. }
  101. }
  102. filteredData[cname] = cd2.Interface().(CostData)
  103. }
  104. return filteredData
  105. }
  106. // ParsePercentString takes a string of expected format "N%" and returns a floating point 0.0N.
  107. // If the "%" symbol is missing, it just returns 0.0N. Empty string is interpreted as "0%" and
  108. // return 0.0.
  109. func ParsePercentString(percentStr string) (float64, error) {
  110. if len(percentStr) == 0 {
  111. return 0.0, nil
  112. }
  113. if percentStr[len(percentStr)-1:] == "%" {
  114. percentStr = percentStr[:len(percentStr)-1]
  115. }
  116. discount, err := strconv.ParseFloat(percentStr, 64)
  117. if err != nil {
  118. return 0.0, err
  119. }
  120. discount *= 0.01
  121. return discount, nil
  122. }
  123. func WriteData(w http.ResponseWriter, data interface{}, err error) {
  124. if err != nil {
  125. proto.WriteError(w, proto.InternalServerError(err.Error()))
  126. return
  127. }
  128. proto.WriteData(w, data)
  129. }
  130. // 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.
  131. func (a *Accesses) RefreshPricingData(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  132. w.Header().Set("Content-Type", "application/json")
  133. w.Header().Set("Access-Control-Allow-Origin", "*")
  134. err := a.CloudProvider.DownloadPricingData()
  135. if err != nil {
  136. log.Errorf("Error refreshing pricing data: %s", err.Error())
  137. }
  138. WriteData(w, nil, err)
  139. }
  140. func (a *Accesses) CostDataModel(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  141. w.Header().Set("Content-Type", "application/json")
  142. w.Header().Set("Access-Control-Allow-Origin", "*")
  143. window := r.URL.Query().Get("timeWindow")
  144. offset := r.URL.Query().Get("offset")
  145. fields := r.URL.Query().Get("filterFields")
  146. namespace := r.URL.Query().Get("namespace")
  147. duration, err := timeutil.ParseDuration(window)
  148. if err != nil {
  149. WriteData(w, nil, fmt.Errorf("error parsing window (%s): %s", window, err))
  150. return
  151. }
  152. end := time.Now()
  153. if offset != "" {
  154. offsetDur, err := timeutil.ParseDuration(offset)
  155. if err != nil {
  156. WriteData(w, nil, fmt.Errorf("error parsing offset (%s): %s", offset, err))
  157. return
  158. }
  159. end = end.Add(-offsetDur)
  160. }
  161. start := end.Add(-duration)
  162. data, err := a.Model.ComputeCostData(start, end)
  163. // apply filter by removing if != namespace
  164. if namespace != "" {
  165. for key, costData := range data {
  166. if costData.Namespace != namespace {
  167. delete(data, key)
  168. }
  169. }
  170. }
  171. if fields != "" {
  172. filteredData := filterFields(fields, data)
  173. WriteData(w, filteredData, err)
  174. } else {
  175. WriteData(w, data, err)
  176. }
  177. }
  178. func (a *Accesses) GetAllNodePricing(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  179. w.Header().Set("Content-Type", "application/json")
  180. w.Header().Set("Access-Control-Allow-Origin", "*")
  181. data, err := a.CloudProvider.AllNodePricing()
  182. WriteData(w, data, err)
  183. }
  184. func (a *Accesses) GetConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  185. w.Header().Set("Content-Type", "application/json")
  186. w.Header().Set("Access-Control-Allow-Origin", "*")
  187. data, err := a.CloudProvider.GetConfig()
  188. WriteData(w, data, err)
  189. }
  190. func (a *Accesses) UpdateSpotInfoConfigs(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 := a.CloudProvider.UpdateConfig(r.Body, aws.SpotInfoUpdateType)
  194. WriteData(w, data, err)
  195. err = a.CloudProvider.DownloadPricingData()
  196. if err != nil {
  197. log.Errorf("Error redownloading data on config update: %s", err.Error())
  198. }
  199. }
  200. func (a *Accesses) UpdateAthenaInfoConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  201. w.Header().Set("Content-Type", "application/json")
  202. w.Header().Set("Access-Control-Allow-Origin", "*")
  203. data, err := a.CloudProvider.UpdateConfig(r.Body, aws.AthenaInfoUpdateType)
  204. WriteData(w, data, err)
  205. }
  206. func (a *Accesses) UpdateBigQueryInfoConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  207. w.Header().Set("Content-Type", "application/json")
  208. w.Header().Set("Access-Control-Allow-Origin", "*")
  209. data, err := a.CloudProvider.UpdateConfig(r.Body, gcp.BigqueryUpdateType)
  210. WriteData(w, data, err)
  211. }
  212. func (a *Accesses) UpdateAzureStorageConfigs(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  213. w.Header().Set("Content-Type", "application/json")
  214. w.Header().Set("Access-Control-Allow-Origin", "*")
  215. data, err := a.CloudProvider.UpdateConfig(r.Body, azure.AzureStorageUpdateType)
  216. if err != nil {
  217. WriteData(w, nil, err)
  218. return
  219. }
  220. WriteData(w, data, err)
  221. }
  222. func (a *Accesses) UpdateConfigByKey(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  223. w.Header().Set("Content-Type", "application/json")
  224. w.Header().Set("Access-Control-Allow-Origin", "*")
  225. data, err := a.CloudProvider.UpdateConfig(r.Body, "")
  226. WriteData(w, data, err)
  227. }
  228. func (a *Accesses) ManagementPlatform(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 := a.CloudProvider.GetManagementPlatform()
  232. WriteData(w, data, err)
  233. }
  234. func (a *Accesses) ClusterInfo(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  235. w.Header().Set("Content-Type", "application/json")
  236. w.Header().Set("Access-Control-Allow-Origin", "*")
  237. data := a.ClusterInfoProvider.GetClusterInfo()
  238. WriteData(w, data, nil)
  239. }
  240. func (a *Accesses) GetClusterInfoMap(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  241. w.Header().Set("Content-Type", "application/json")
  242. w.Header().Set("Access-Control-Allow-Origin", "*")
  243. data := a.ClusterMap.AsMap()
  244. WriteData(w, data, nil)
  245. }
  246. func (a *Accesses) GetServiceAccountStatus(w http.ResponseWriter, _ *http.Request, _ httprouter.Params) {
  247. w.Header().Set("Content-Type", "application/json")
  248. w.Header().Set("Access-Control-Allow-Origin", "*")
  249. WriteData(w, a.CloudProvider.ServiceAccountStatus(), nil)
  250. }
  251. func (a *Accesses) GetPricingSourceStatus(w http.ResponseWriter, _ *http.Request, _ httprouter.Params) {
  252. w.Header().Set("Content-Type", "application/json")
  253. w.Header().Set("Access-Control-Allow-Origin", "*")
  254. data := a.CloudProvider.PricingSourceStatus()
  255. WriteData(w, data, nil)
  256. }
  257. func (a *Accesses) GetPricingSourceCounts(w http.ResponseWriter, _ *http.Request, _ httprouter.Params) {
  258. w.Header().Set("Content-Type", "application/json")
  259. w.Header().Set("Access-Control-Allow-Origin", "*")
  260. data, err := a.Model.GetPricingSourceCounts()
  261. WriteData(w, data, err)
  262. }
  263. func (a *Accesses) GetPricingSourceSummary(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
  264. w.Header().Set("Content-Type", "application/json")
  265. w.Header().Set("Access-Control-Allow-Origin", "*")
  266. data := a.CloudProvider.PricingSourceSummary()
  267. WriteData(w, data, nil)
  268. }
  269. func (a *Accesses) GetOrphanedPods(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  270. w.Header().Set("Content-Type", "application/json")
  271. w.Header().Set("Access-Control-Allow-Origin", "*")
  272. podlist := a.ClusterCache.GetAllPods()
  273. var lonePods []*clustercache.Pod
  274. for _, pod := range podlist {
  275. if len(pod.OwnerReferences) == 0 {
  276. lonePods = append(lonePods, pod)
  277. }
  278. }
  279. body, err := json.Marshal(lonePods)
  280. if err != nil {
  281. fmt.Fprintf(w, "Error decoding pod: %s", err)
  282. } else {
  283. w.Write(body)
  284. }
  285. }
  286. func (a *Accesses) GetInstallNamespace(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
  287. w.Header().Set("Content-Type", "application/json")
  288. w.Header().Set("Access-Control-Allow-Origin", "*")
  289. ns := env.GetOpencostNamespace()
  290. w.Write([]byte(ns))
  291. }
  292. type InstallInfo struct {
  293. Containers []ContainerInfo `json:"containers"`
  294. ClusterInfo map[string]string `json:"clusterInfo"`
  295. Version string `json:"version"`
  296. }
  297. type ContainerInfo struct {
  298. ContainerName string `json:"containerName"`
  299. Image string `json:"image"`
  300. StartTime string `json:"startTime"`
  301. }
  302. func (a *Accesses) GetInstallInfo(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
  303. w.Header().Set("Content-Type", "application/json")
  304. w.Header().Set("Access-Control-Allow-Origin", "*")
  305. containers, err := GetKubecostContainers(a.KubeClientSet)
  306. if err != nil {
  307. http.Error(w, fmt.Sprintf("Unable to list pods: %s", err.Error()), http.StatusInternalServerError)
  308. return
  309. }
  310. info := InstallInfo{
  311. Containers: containers,
  312. ClusterInfo: make(map[string]string),
  313. Version: version.FriendlyVersion(),
  314. }
  315. nodes := a.ClusterCache.GetAllNodes()
  316. cachePods := a.ClusterCache.GetAllPods()
  317. info.ClusterInfo["nodeCount"] = strconv.Itoa(len(nodes))
  318. info.ClusterInfo["podCount"] = strconv.Itoa(len(cachePods))
  319. body, err := json.Marshal(info)
  320. if err != nil {
  321. http.Error(w, fmt.Sprintf("Error decoding pod: %s", err.Error()), http.StatusInternalServerError)
  322. return
  323. }
  324. w.Write(body)
  325. }
  326. func GetKubecostContainers(kubeClientSet kubernetes.Interface) ([]ContainerInfo, error) {
  327. pods, err := kubeClientSet.CoreV1().Pods(env.GetOpencostNamespace()).List(context.Background(), metav1.ListOptions{
  328. LabelSelector: "app=cost-analyzer",
  329. FieldSelector: "status.phase=Running",
  330. Limit: 1,
  331. })
  332. if err != nil {
  333. return nil, fmt.Errorf("failed to query kubernetes client for kubecost pods: %s", err)
  334. }
  335. // If we have zero pods either something is weird with the install since the app selector is not exposed in the helm
  336. // chart or more likely we are running locally - in either case Images field will return as null
  337. containers := make([]ContainerInfo, 0)
  338. if len(pods.Items) > 0 {
  339. for _, pod := range pods.Items {
  340. for _, container := range pod.Spec.Containers {
  341. c := ContainerInfo{
  342. ContainerName: container.Name,
  343. Image: container.Image,
  344. StartTime: pod.Status.StartTime.String(),
  345. }
  346. containers = append(containers, c)
  347. }
  348. }
  349. }
  350. return containers, nil
  351. }
  352. func (a *Accesses) AddServiceKey(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  353. w.Header().Set("Content-Type", "application/json")
  354. w.Header().Set("Access-Control-Allow-Origin", "*")
  355. r.ParseForm()
  356. key := r.PostForm.Get("key")
  357. k := []byte(key)
  358. err := os.WriteFile(env.GetGCPAuthSecretFilePath(), k, 0644)
  359. if err != nil {
  360. fmt.Fprintf(w, "Error writing service key: %s", err)
  361. }
  362. w.WriteHeader(http.StatusOK)
  363. }
  364. func (a *Accesses) GetHelmValues(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
  365. w.Header().Set("Content-Type", "application/json")
  366. w.Header().Set("Access-Control-Allow-Origin", "*")
  367. encodedValues := sysenv.Get("HELM_VALUES", "")
  368. if encodedValues == "" {
  369. fmt.Fprintf(w, "Values reporting disabled")
  370. return
  371. }
  372. result, err := base64.StdEncoding.DecodeString(encodedValues)
  373. if err != nil {
  374. fmt.Fprintf(w, "Failed to decode encoded values: %s", err)
  375. return
  376. }
  377. w.Write(result)
  378. }
  379. func Initialize(router *httprouter.Router, additionalConfigWatchers ...*watcher.ConfigMapWatcher) *Accesses {
  380. var err error
  381. // Kubernetes API setup
  382. kubeClientset, err := kubeconfig.LoadKubeClient("")
  383. if err != nil {
  384. log.Fatalf("Failed to build Kubernetes client: %s", err.Error())
  385. }
  386. // Create Kubernetes Cluster Cache + Watchers
  387. k8sCache := clusterc.NewKubernetesClusterCache(kubeClientset)
  388. k8sCache.Run()
  389. // Create ConfigFileManager for synchronization of shared configuration
  390. confManager := config.NewConfigFileManager(nil)
  391. cloudProviderKey := env.GetCloudProviderAPIKey()
  392. cloudProvider, err := provider.NewProvider(k8sCache, cloudProviderKey, confManager)
  393. if err != nil {
  394. panic(err.Error())
  395. }
  396. // ClusterInfo Provider to provide the cluster map with local and remote cluster data
  397. var clusterInfoProvider clusters.ClusterInfoProvider
  398. if env.IsClusterInfoFileEnabled() {
  399. clusterInfoFile := confManager.ConfigFileAt(env.GetClusterInfoFilePath())
  400. clusterInfoProvider = NewConfiguredClusterInfoProvider(clusterInfoFile)
  401. } else {
  402. clusterInfoProvider = NewLocalClusterInfoProvider(kubeClientset, cloudProvider)
  403. }
  404. const maxRetries = 10
  405. const retryInterval = 10 * time.Second
  406. var fatalErr error
  407. ctx, cancel := context.WithCancel(context.Background())
  408. fn := func() (source.OpenCostDataSource, error) {
  409. ds, e := prom.NewDefaultPrometheusDataSource(clusterInfoProvider)
  410. if e != nil {
  411. if source.IsRetryable(e) {
  412. return nil, e
  413. }
  414. fatalErr = e
  415. cancel()
  416. }
  417. return ds, e
  418. }
  419. if env.IsCollectorDataSourceEnabled() {
  420. fn = func() (source.OpenCostDataSource, error) {
  421. store := storage.GetDefaultStorage()
  422. nodeStatConf, err := NewNodeClientConfigFromEnv()
  423. if err != nil {
  424. return nil, fmt.Errorf("failed to get node client config: %w", err)
  425. }
  426. clusterConfig, err := kubeconfig.LoadKubeconfig("")
  427. if err != nil {
  428. return nil, fmt.Errorf("failed to load kube config: %w", err)
  429. }
  430. nodeStatClient := nodestats.NewNodeStatsSummaryClient(k8sCache, nodeStatConf, clusterConfig)
  431. ds := collector.NewDefaultCollectorDataSource(
  432. store,
  433. clusterInfoProvider,
  434. k8sCache,
  435. nodeStatClient,
  436. )
  437. return ds, nil
  438. }
  439. }
  440. dataSource, _ := retry.Retry(
  441. ctx,
  442. fn,
  443. maxRetries,
  444. retryInterval,
  445. )
  446. if fatalErr != nil {
  447. log.Fatalf("Failed to create Prometheus data source: %s", fatalErr)
  448. panic(fatalErr)
  449. }
  450. // Append the pricing config watcher
  451. installNamespace := env.GetOpencostNamespace()
  452. configWatchers := watcher.NewConfigMapWatchers(kubeClientset, installNamespace, additionalConfigWatchers...)
  453. configWatchers.AddWatcher(provider.ConfigWatcherFor(cloudProvider))
  454. configWatchers.AddWatcher(metrics.GetMetricsConfigWatcher())
  455. configWatchers.Watch()
  456. clusterMap := dataSource.ClusterMap()
  457. settingsCache := cache.New(cache.NoExpiration, cache.NoExpiration)
  458. costModel := NewCostModel(dataSource, cloudProvider, k8sCache, clusterMap, dataSource.BatchDuration())
  459. metricsEmitter := NewCostModelMetricsEmitter(k8sCache, cloudProvider, clusterInfoProvider, costModel)
  460. a := &Accesses{
  461. DataSource: dataSource,
  462. KubeClientSet: kubeClientset,
  463. ClusterCache: k8sCache,
  464. ClusterMap: clusterMap,
  465. CloudProvider: cloudProvider,
  466. ConfigFileManager: confManager,
  467. ClusterInfoProvider: clusterInfoProvider,
  468. Model: costModel,
  469. MetricsEmitter: metricsEmitter,
  470. SettingsCache: settingsCache,
  471. }
  472. // Initialize mechanism for subscribing to settings changes
  473. a.InitializeSettingsPubSub()
  474. err = a.CloudProvider.DownloadPricingData()
  475. if err != nil {
  476. log.Infof("Failed to download pricing data: %s", err)
  477. }
  478. if !env.IsKubecostMetricsPodEnabled() {
  479. a.MetricsEmitter.Start()
  480. }
  481. a.DataSource.RegisterEndPoints(router)
  482. router.GET("/costDataModel", a.CostDataModel)
  483. router.GET("/allocation/compute", a.ComputeAllocationHandler)
  484. router.GET("/allocation/compute/summary", a.ComputeAllocationHandlerSummary)
  485. router.GET("/allNodePricing", a.GetAllNodePricing)
  486. router.POST("/refreshPricing", a.RefreshPricingData)
  487. router.GET("/managementPlatform", a.ManagementPlatform)
  488. router.GET("/clusterInfo", a.ClusterInfo)
  489. router.GET("/clusterInfoMap", a.GetClusterInfoMap)
  490. router.GET("/serviceAccountStatus", a.GetServiceAccountStatus)
  491. router.GET("/pricingSourceStatus", a.GetPricingSourceStatus)
  492. router.GET("/pricingSourceSummary", a.GetPricingSourceSummary)
  493. router.GET("/pricingSourceCounts", a.GetPricingSourceCounts)
  494. router.GET("/orphanedPods", a.GetOrphanedPods)
  495. router.GET("/installNamespace", a.GetInstallNamespace)
  496. router.GET("/installInfo", a.GetInstallInfo)
  497. router.POST("/serviceKey", a.AddServiceKey)
  498. router.GET("/helmValues", a.GetHelmValues)
  499. return a
  500. }
  501. // InitializeCloudCost Initializes Cloud Cost pipeline and querier and registers endpoints
  502. func InitializeCloudCost(router *httprouter.Router, providerConfig models.ProviderConfig) {
  503. log.Debugf("Cloud Cost config path: %s", env.GetCloudCostConfigPath())
  504. cloudConfigController := cloudconfig.NewMemoryController(providerConfig)
  505. repo := cloudcost.NewMemoryRepository()
  506. cloudCostPipelineService := cloudcost.NewPipelineService(repo, cloudConfigController, cloudcost.DefaultIngestorConfiguration())
  507. repoQuerier := cloudcost.NewRepositoryQuerier(repo)
  508. cloudCostQueryService := cloudcost.NewQueryService(repoQuerier, repoQuerier)
  509. router.GET("/cloud/config/export", cloudConfigController.GetExportConfigHandler())
  510. router.GET("/cloud/config/enable", cloudConfigController.GetEnableConfigHandler())
  511. router.GET("/cloud/config/disable", cloudConfigController.GetDisableConfigHandler())
  512. router.GET("/cloud/config/delete", cloudConfigController.GetDeleteConfigHandler())
  513. router.GET("/cloudCost", cloudCostQueryService.GetCloudCostHandler())
  514. router.GET("/cloudCost/view/graph", cloudCostQueryService.GetCloudCostViewGraphHandler())
  515. router.GET("/cloudCost/view/totals", cloudCostQueryService.GetCloudCostViewTotalsHandler())
  516. router.GET("/cloudCost/view/table", cloudCostQueryService.GetCloudCostViewTableHandler())
  517. router.GET("/cloudCost/status", cloudCostPipelineService.GetCloudCostStatusHandler())
  518. router.GET("/cloudCost/rebuild", cloudCostPipelineService.GetCloudCostRebuildHandler())
  519. router.GET("/cloudCost/repair", cloudCostPipelineService.GetCloudCostRepairHandler())
  520. }
  521. func InitializeCustomCost(router *httprouter.Router) *customcost.PipelineService {
  522. hourlyRepo := customcost.NewMemoryRepository()
  523. dailyRepo := customcost.NewMemoryRepository()
  524. ingConfig := customcost.DefaultIngestorConfiguration()
  525. var err error
  526. customCostPipelineService, err := customcost.NewPipelineService(hourlyRepo, dailyRepo, ingConfig)
  527. if err != nil {
  528. log.Errorf("error instantiating custom cost pipeline service: %v", err)
  529. return nil
  530. }
  531. customCostQuerier := customcost.NewRepositoryQuerier(hourlyRepo, dailyRepo, ingConfig.HourlyDuration, ingConfig.DailyDuration)
  532. customCostQueryService := customcost.NewQueryService(customCostQuerier)
  533. router.GET("/customCost/total", customCostQueryService.GetCustomCostTotalHandler())
  534. router.GET("/customCost/timeseries", customCostQueryService.GetCustomCostTimeseriesHandler())
  535. return customCostPipelineService
  536. }