Timescale extensions for ActiveRecord

active_record_timescale_extensions.rb

# Put this in config/initializers/

module ActiveRecord::Tasks::TimescaleExtensions
  def structure_load(filename, extra_flags)
    # If this structure is using timescaledb then we need to take some extra steps
    # https://docs.timescale.com/timescaledb/latest/how-to-guides/backup-and-restore/pg-dump-and-restore#entire-database
    if system "grep", "-Eq", "CREATE EXTENSION( IF NOT EXISTS)? timescaledb", filename
      puts "Enabling timescale extension"
      connection.enable_extension("timescaledb")

      puts "Starting timescale restore"
      connection.execute("SELECT timescaledb_pre_restore()")

      super

      puts "Finishing timescale restore"
      connection.execute("SELECT timescaledb_post_restore()")
    else
      super
    end
  end
end

ActiveSupport.on_load(:active_record) do
  ActiveRecord::Tasks::PostgreSQLDatabaseTasks.prepend ActiveRecord::Tasks::TimescaleExtensions
end