<commit_before>module Slowpoke
  module Migration
    def connection
      connection = super
      if Slowpoke.migration_statement_timeout && !@migration_statement_timeout_set
        connection.execute("SET statement_timeout = #{Slowpoke.migration_statement_timeout.to_i}")
        @migration_statement_timeout_set = true
      end
      connection
    end
  end
end

ActiveSupport.on_load(:active_record) do
  ActiveRecord::Migration.prepend(Slowpoke::Migration)<commit_msg>Use local timeout for Postgres<commit_after>module Slowpoke
  module Migration
    def connection
      connection = super
      if Slowpoke.migration_statement_timeout && !@migration_statement_timeout_set
        connection.execute("SET LOCAL statement_timeout = #{Slowpoke.migration_statement_timeout.to_i}")
        @migration_statement_timeout_set = true
      end
      connection
    end
  end
end

ActiveSupport.on_load(:active_record) do
  ActiveRecord::Migration.prepend(Slowpoke::Migration)
{'fraction_non_alphanumeric': 0.06755373592630501, 'fraction_numerical': 0.0, 'mean_word_length': 4.373626373626373, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '3711474', 'n_tokens_mistral': 280, 'n_tokens_neox': 278, 'n_words': 52}