Sha256: 04f8c298fe5219660a0fa55199d3e7169d7c9f326f969be89f92a772eae1e98a
Contents?: true
Size: 1.46 KB
Versions: 48
Compression:
Stored size: 1.46 KB
Contents
# frozen_string_literal: true
require "weakref"
# based on Rails's active_support/fork_tracker.rb
module SQLite3
module ForkSafety
module CoreExt
def _fork
pid = super
if pid == 0
ForkSafety.discard
end
pid
end
end
@databases = []
@mutex = Mutex.new
@suppress = false
class << self
def hook!
::Process.singleton_class.prepend(CoreExt)
end
def track(database)
@mutex.synchronize do
@databases << WeakRef.new(database)
end
end
def discard
warned = @suppress
@databases.each do |db|
next unless db.weakref_alive?
unless db.closed? || db.readonly?
unless warned
# If you are here, you may want to read
# https://github.com/sparklemotion/sqlite3-ruby/pull/558
warn("Writable sqlite database connection(s) were inherited from a forked process. " \
"This is unsafe and the connections are being closed to prevent possible data " \
"corruption. Please close writable sqlite database connections before forking.",
uplevel: 0)
warned = true
end
db.close
end
end
@databases.clear
end
# Call to suppress the fork-related warnings.
def suppress_warnings!
@suppress = true
end
end
end
end
SQLite3::ForkSafety.hook!
Version data entries
48 entries across 48 versions & 1 rubygems