iv-org-mirror/shard-crystal-db
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/shard-crystal-db.git synced 2024-08-15 00:53:32 +00:00
Code Issues Projects Releases Packages Wiki Activity

Use select / timeout in Crystal 0.34

Browse source
This commit is contained in:
Brian J. Cardiff 2020-03-25 10:56:03 -03:00
parent af200eac54
commit f4b298d3a5
1 changed files with 30 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
src/db/pool.cr
View file

@ -206,25 +206,38 @@ module DB
@idle.first? @idle.first?
end end
private def wait_for_available {% if compare_versions(Crystal::VERSION, "0.34.0-0") > 0 %}
timeout = TimeoutHelper.new(@checkout_timeout.to_f64) private def wait_for_available
sync_inc_waiting_resource sync_inc_waiting_resource
timeout.start select
when @availability_channel.receive
# TODO update to select keyword for crystal 0.19 sync_dec_waiting_resource
index, _ = Channel.select(@availability_channel.receive_select_action, timeout.receive_select_action) when timeout(@checkout_timeout.seconds)
case index sync_dec_waiting_resource
when 0 raise DB::PoolTimeout.new
timeout.cancel end
sync_dec_waiting_resource
when 1
sync_dec_waiting_resource
raise DB::PoolTimeout.new
else
raise DB::Error.new
end end
end {% else %}
private def wait_for_available
timeout = TimeoutHelper.new(@checkout_timeout.to_f64)
sync_inc_waiting_resource
timeout.start
index, _ = Channel.select(@availability_channel.receive_select_action, timeout.receive_select_action)
case index
when 0
timeout.cancel
sync_dec_waiting_resource
when 1
sync_dec_waiting_resource
raise DB::PoolTimeout.new
else
raise DB::Error.new
end
end
{% end %}
private def sync_inc_waiting_resource private def sync_inc_waiting_resource
sync { @waiting_resource += 1 } sync { @waiting_resource += 1 }