services/api/db/migrate/20230503224107_priority_update_functions.rb

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: AGPL-3.0
   4
   5 class PriorityUpdateFunctions < ActiveRecord::Migration[5.2]
   6   def up
   7     ActiveRecord::Base.connection.execute %{
   8 CREATE OR REPLACE FUNCTION container_priority(for_container_uuid character varying, inherited bigint, inherited_from character varying) returns bigint
   9     LANGUAGE sql
  10     AS $$
  11 /* Determine the priority of an individual container.
  12    The "inherited" priority comes from the path we followed from the root, the parent container
  13    priority hasn't been updated in the table yet but we need to behave it like it has been.
  14 */
  15 select coalesce(max(case when containers.uuid = inherited_from then inherited
  16                          when containers.priority is not NULL then containers.priority
  17                          else container_requests.priority * 1125899906842624::bigint - (extract(epoch from container_requests.created_at)*1000)::bigint
  18                     end), 0) from
  19     container_requests left outer join containers on container_requests.requesting_container_uuid = containers.uuid
  20     where container_requests.container_uuid = for_container_uuid and container_requests.state = 'Committed' and container_requests.priority > 0;
  21 $$;
  22 }
  23
  24     ActiveRecord::Base.connection.execute %{
  25 CREATE OR REPLACE FUNCTION container_tree_priorities(for_container_uuid character varying) returns table (pri_container_uuid character varying, upd_priority bigint)
  26     LANGUAGE sql
  27     AS $$
  28 /* Calculate the priorities of all containers starting from for_container_uuid.
  29    This traverses the process tree downward and calls container_priority for each container
  30    and returns a table of container uuids and their new priorities.
  31 */
  32 with recursive tab(upd_container_uuid, upd_priority) as (
  33   select for_container_uuid, container_priority(for_container_uuid, 0, '')
  34 union
  35   select containers.uuid, container_priority(containers.uuid, child_requests.upd_priority, child_requests.upd_container_uuid)
  36   from (tab join container_requests on tab.upd_container_uuid = container_requests.requesting_container_uuid) as child_requests
  37   join containers on child_requests.container_uuid = containers.uuid
  38   where containers.state in ('Queued', 'Locked', 'Running')
  39 )
  40 select upd_container_uuid, upd_priority from tab;
  41 $$;
  42 }
  43
  44     ActiveRecord::Base.connection.execute %{
  45 CREATE OR REPLACE FUNCTION container_tree(for_container_uuid character varying) returns table (pri_container_uuid character varying)
  46     LANGUAGE sql
  47     AS $$
  48 /* A lighter weight version of the update_priorities query that only returns the containers in a tree,
  49    used by SELECT FOR UPDATE.
  50 */
  51 with recursive tab(upd_container_uuid) as (
  52   select for_container_uuid
  53 union
  54   select containers.uuid
  55   from (tab join container_requests on tab.upd_container_uuid = container_requests.requesting_container_uuid) as child_requests
  56   join containers on child_requests.container_uuid = containers.uuid
  57   where containers.state in ('Queued', 'Locked', 'Running')
  58 )
  59 select upd_container_uuid from tab;
  60 $$;
  61 }
  62   end
  63
  64   def down
  65     ActiveRecord::Base.connection.execute "DROP FUNCTION container_priority"
  66     ActiveRecord::Base.connection.execute "DROP FUNCTION container_tree_priorities"
  67     ActiveRecord::Base.connection.execute "DROP FUNCTION container_tree"
  68   end
  69 end