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

feat(error): close resources on resource lost (#155)

Browse source
This commit is contained in:
Stephen von Takach 2021-09-10 21:36:01 +10:00 • committed by GitHub
parent 6dc3f2dd6f
commit a25f33611c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 21 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/db/error.cr
View file

@ -18,6 +18,7 @@ module DB
getter resource : T
def initialize(@resource : T)
@resource.close
end
end

8
src/db/pool.cr
View file

@ -179,10 +179,12 @@ module DB
sleep @retry_delay if i >= current_available
return yield
rescue e : PoolResourceLost(T)
# if the connection is lost close it to release resources
# and remove it from the known pool.
# if the connection is lost it will be closed by
# the exception to release resources
# we still need to remove it from the known pool.
# Closed connection will be evicted from statement cache
# in PoolPreparedStatement#clean_connections
sync { delete(e.resource) }
e.resource.close
rescue e : PoolResourceRefused
# a ConnectionRefused means a new connection
# was intended to be created