BigW Consortium Gitlab

sha_attribute.rb 587 Bytes
Newer Older
1 2 3 4 5
module ShaAttribute
  extend ActiveSupport::Concern

  module ClassMethods
    def sha_attribute(name)
6 7
      return unless table_exists?

8 9 10 11 12 13 14 15 16 17 18 19 20
      column = columns.find { |c| c.name == name.to_s }

      # In case the table doesn't exist we won't be able to find the column,
      # thus we will only check the type if the column is present.
      if column && column.type != :binary
        raise ArgumentError,
          "sha_attribute #{name.inspect} is invalid since the column type is not :binary"
      end

      attribute(name, Gitlab::Database::ShaAttribute.new)
    end
  end
end