added initializer which returns definitly count of domains

This commit is contained in:
Oleg Hasjanov 2021-05-07 17:14:14 +03:00
parent dffa6470d8
commit 54f7e87370

View file

@ -6,6 +6,8 @@ module Api
class DomainsController < ::Api::V1::Registrant::BaseController class DomainsController < ::Api::V1::Registrant::BaseController
before_action :set_tech_flag, only: [:show] before_action :set_tech_flag, only: [:show]
LIMIT_DOMAIN_TOTAL = 3000.freeze
def index def index
limit = params[:limit] || 200 limit = params[:limit] || 200
offset = params[:offset] || 0 offset = params[:offset] || 0
@ -57,10 +59,24 @@ module Api
end end
def current_user_domains def current_user_domains
return initialization_count_of__domains if params[:tech] == 'init'
current_registrant_user.domains(admin: params[:tech] != 'true') current_registrant_user.domains(admin: params[:tech] != 'true')
rescue CompanyRegister::NotAvailableError rescue CompanyRegister::NotAvailableError
return initialization_count_of__direct_domains if params[:tech] == 'init'
current_registrant_user.direct_domains(admin: params[:tech] != 'true') current_registrant_user.direct_domains(admin: params[:tech] != 'true')
end end
def initialization_count_of__direct_domains
return current_registrant_user.direct_domains(admin: false) if current_user_domains_total_count < LIMIT_DOMAIN_TOTAL
current_registrant_user.direct_domains(admin: true)
end
def initialization_count_of__domains
return current_registrant_user.domains(admin: false) if current_user_domains_total_count < LIMIT_DOMAIN_TOTAL
current_registrant_user.domains(admin: true)
end
end end
end end
end end