can_lookup.rb 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. # Copyright (C) 2012-2016 Zammad Foundation, http://zammad-foundation.org/
  2. module ApplicationModel::CanLookup
  3. extend ActiveSupport::Concern
  4. class_methods do
  5. =begin
  6. lookup model from cache (if exists) or retrieve it from db, id, name, login or email possible
  7. result = Model.lookup(id: 123)
  8. result = Model.lookup(name: 'some name')
  9. result = Model.lookup(login: 'some login')
  10. result = Model.lookup(email: 'some login')
  11. returns
  12. result = model # with all attributes
  13. =end
  14. def lookup(**attr)
  15. raise ArgumentError, "Multiple lookup attributes given (#{attr.keys.join(', ')}), only support (#{lookup_keys.join(', ')})" if attr.many?
  16. attr.transform_keys!(&:to_sym).slice!(*lookup_keys)
  17. raise ArgumentError, "Valid lookup attribute required (#{lookup_keys.join(', ')})" if attr.empty?
  18. record = cache_get(attr.values.first)
  19. record ||= find_and_save_to_cache_by(attr)
  20. end
  21. =begin
  22. return possible lookup keys for model
  23. result = Model.lookup_keys
  24. returns
  25. [:id, :name] # or fror users [:id, :login, :email]
  26. =end
  27. def lookup_keys
  28. @lookup_keys ||= %i[id name login email number] & attribute_names.map(&:to_sym)
  29. end
  30. private
  31. def find_and_save_to_cache_by(attr)
  32. if !Rails.application.config.db_case_sensitive && string_key?(attr.keys.first)
  33. where(attr).find { |r| r[attr.keys.first] == attr.values.first }
  34. else
  35. find_by(attr)
  36. end.tap { |r| cache_set(attr.values.first, r) }
  37. end
  38. def string_key?(key)
  39. type_for_attribute(key.to_s).type == :string
  40. end
  41. end
  42. end