aboutsummaryrefslogtreecommitdiffstats
path: root/lib/action_cable/server/worker/clear_database_connections.rb
blob: 722d363a41ffffaff57d21e53ca1215f2f176ad7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
module ActionCable
  module Server
    class Worker
      # Clear active connections between units of work so the long-running channel or connection processes do not hoard connections.
      module ClearDatabaseConnections
        extend ActiveSupport::Concern

        included do
          if defined?(ActiveRecord::Base)
            set_callback :work, :around, :with_database_connections
          end
        end

        def with_database_connections
          yield
        ensure
          ActiveRecord::Base.clear_active_connections!
        end
      end
    end
  end
end