File: id_in_ordered.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (20 lines) | stat: -rw-r--r-- 600 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# frozen_string_literal: true

module IdInOrdered
  extend ActiveSupport::Concern

  included do
    scope :id_in_ordered, ->(ids) do
      raise ArgumentError, "ids must be an array of integers" unless ids.is_a?(Enumerable) && ids.all?(Integer)

      # No need to sort if no more than 1 and the sorting code doesn't work
      # with an empty array
      return id_in(ids) unless ids.count > 1

      id_attribute = arel_table[:id]
      id_in(ids)
        .order(
          Arel.sql("array_position(ARRAY[#{ids.join(',')}], #{id_attribute.relation.name}.#{id_attribute.name})"))
    end
  end
end