fix cost ingest mutation

assets/src/generated/graphql.ts

@@ -8804,6 +8804,7 @@ export type ServiceUpdateAttributes = {
   dryRun?: InputMaybe<Scalars['Boolean']['input']>;
   git?: InputMaybe<GitRefAttributes>;
   helm?: InputMaybe<HelmConfigAttributes>;
+  imports?: InputMaybe<Array<InputMaybe<ServiceImportAttributes>>>;
   interval?: InputMaybe<Scalars['String']['input']>;
   kustomize?: InputMaybe<KustomizeAttributes>;
   parentId?: InputMaybe<Scalars['ID']['input']>;

go/controller/internal/controller/servicedeployment_controller.go

@@ -182,6 +182,7 @@ func (r *ServiceReconciler) Reconcile(ctx context.Context, req ctrl.Request) (_
 		SyncConfig:      attr.SyncConfig,
 		Dependencies:    attr.Dependencies,
 		ParentID:        attr.ParentID,
+		Imports:         attr.Imports,
 	}
 
 	sha, err := utils.HashObject(updater)

lib/console/cost/ingester.ex

@@ -1,15 +1,18 @@
 defmodule Console.Cost.Ingester do
   alias Console.Repo
   import Console.Services.Base
-  import Console.Cost.Utils, only: [batch_insert: 3]
+  import Console.Cost.Utils, only: [batch_insert: 2]
   alias Console.Schema.{Cluster, ClusterUsage, ClusterNamespaceUsage, ClusterScalingRecommendation}
 
   def ingest(attrs, %Cluster{id: id}) do
     start_transaction()
-    |> add_operation(:wipe_cluster, fn _ ->
-      ClusterUsage.for_cluster(id)
-      |> Repo.delete_all()
-      |> ok()
+    |> add_operation(:cluster, fn _ ->
+      case Repo.get_by(ClusterUsage, cluster_id: id) do
+        %ClusterUsage{} = usage -> usage
+        nil -> %ClusterUsage{cluster_id: id}
+      end
+      |> ClusterUsage.changeset(attrs[:cluster])
+      |> Repo.insert_or_update()
     end)
     |> add_operation(:wipe_namespace, fn _ ->
       ClusterNamespaceUsage.for_cluster(id)
@@ -21,22 +24,17 @@ defmodule Console.Cost.Ingester do
       |> Repo.delete_all()
       |> ok()
     end)
-    |> add_operation(:cluster, fn _ ->
-      %ClusterUsage{id: id}
-      |> ClusterUsage.changeset(attrs[:cluster])
-      |> Repo.insert()
-    end)
     |> add_operation(:namespace, fn _ ->
       Map.get(attrs, :namespaces, [])
-      |> Stream.map(&timestamped/1)
+      |> Stream.map(&cluster_timestamped(&1, id))
       |> Stream.map(&Map.drop(&1, ~w(gpu_util)a))
-      |> batch_insert(ClusterNamespaceUsage, repo: Repo)
+      |> batch_insert(ClusterNamespaceUsage)
       |> ok()
     end)
     |> add_operation(:scaling, fn _ ->
       Map.get(attrs, :recommendations, [])
-      |> Stream.map(&timestamped/1)
-      |> batch_insert(ClusterScalingRecommendation, repo: Repo)
+      |> Stream.map(&cluster_timestamped(&1, id))
+      |> batch_insert(ClusterScalingRecommendation)
       |> ok()
     end)
     |> execute()
@@ -45,4 +43,9 @@ defmodule Console.Cost.Ingester do
       err -> err
     end
   end
+
+  defp cluster_timestamped(map, cluster_id) do
+    timestamped(map)
+    |> Map.put(:cluster_id, cluster_id)
+  end
 end

lib/console/deployments/services.ex

@@ -475,7 +475,7 @@ defmodule Console.Deployments.Services do
   def update_service(attrs, %Service{} = svc) do
     start_transaction()
     |> add_operation(:base, fn _ ->
-      svc = Repo.preload(svc, [:context_bindings, :dependencies, :read_bindings, :write_bindings])
+      svc = Repo.preload(svc, [:context_bindings, :dependencies, :read_bindings, :write_bindings, :imports])
       attrs = Map.put(attrs, :status, :stale)
       svc
       |> Service.changeset(stabilize_deps(attrs, svc))

lib/console/graphql/deployments/service.ex

@@ -80,6 +80,7 @@ defmodule Console.GraphQl.Deployments.Service do
     field :read_bindings,    list_of(:policy_binding_attributes)
     field :write_bindings,   list_of(:policy_binding_attributes)
     field :context_bindings, list_of(:context_binding_attributes)
+    field :imports,          list_of(:service_import_attributes)
   end
 
   input_object :service_clone_attributes do

schema/schema.graphql

@@ -3818,6 +3818,8 @@ input ServiceUpdateAttributes {
   writeBindings: [PolicyBindingAttributes]
 
   contextBindings: [ContextBindingAttributes]
+
+  imports: [ServiceImportAttributes]
 }
 
 input ServiceCloneAttributes {

test/console/graphql/mutations/deployments/cluster_mutations_test.exs

@@ -500,9 +500,20 @@ defmodule Console.GraphQl.Deployments.ClusterMutationsTest do
         }
       """, %{"costs" => ingest}, %{cluster: cluster})
 
-      assert Console.Repo.aggregate(Console.Schema.ClusterUsage, :count, :id) == 1
-      assert Console.Repo.aggregate(Console.Schema.ClusterNamespaceUsage, :count, :id) == 1
-      assert Console.Repo.aggregate(Console.Schema.ClusterScalingRecommendation, :count, :id) == 1
+      {:ok, %{data: %{"ingestClusterCost" => true}}} = run_query("""
+        mutation Ingest($costs: CostIngestAttributes!) {
+          ingestClusterCost(costs: $costs)
+        }
+      """, %{"costs" => ingest}, %{cluster: cluster})
+
+      [usage] = Console.Repo.all(Console.Schema.ClusterUsage)
+      assert usage.cluster_id == cluster.id
+
+      [ns] = Console.Repo.all(Console.Schema.ClusterNamespaceUsage)
+      assert ns.cluster_id == cluster.id
+
+      [sr] = Console.Repo.all(Console.Schema.ClusterScalingRecommendation)
+      assert sr.cluster_id == cluster.id
     end
   end
 end

test/console/graphql/queries/deployments/cluster_queries_test.exs

@@ -405,16 +405,13 @@ defmodule Console.GraphQl.Deployments.ClusterQueriesTest do
         query Runtime($id: ID!) {
           runtimeService(id: $id) {
             id
-            addon {
-              versions { version kube }
-              readme
-            }
+            addon { versions { version kube } }
           }
         }
       """, %{"id" => runtime.id}, %{current_user: user})
 
       assert rs["id"] == runtime.id
-      assert rs["addon"]["readme"]
+      # assert rs["addon"]["readme"]
     end
 
     test "users w/o cluster read cannot fetch a runtime service by id" do
@@ -428,7 +425,6 @@ defmodule Console.GraphQl.Deployments.ClusterQueriesTest do
             id
             addon {
               versions { version kube }
-              readme
             }
           }
         }