Contents

module Neo4j
  module Migrations
    class Base < ::Neo4j::Migration
      include Neo4j::Migrations::Helpers
      include Neo4j::Migrations::Helpers::Schema
      include Neo4j::Migrations::Helpers::IdProperty
      include Neo4j::Migrations::Helpers::Relationships

      def initialize(migration_id)
        @migration_id = migration_id
      end

      def migrate(method)
        Benchmark.realtime do
          method == :up ? migrate_up : migrate_down
        end
      end

      def up
        fail NotImplementedError
      end

      def down
        fail NotImplementedError
      end

      private

      def migrate_up
        schema = SchemaMigration.create!(migration_id: @migration_id, incomplete: true)
        begin
          run_migration(:up)
        rescue StandardError => e
          schema.destroy
          handle_migration_error!(e)
        end
        schema.update!(incomplete: nil)
      end

      def migrate_down
        schema = SchemaMigration.find_by!(migration_id: @migration_id)
        schema.update!(incomplete: true)
        begin
          run_migration(:down)
        rescue StandardError => e
          schema.update!(incomplete: nil)
          handle_migration_error!(e)
        end
        schema.destroy
      end

      def run_migration(direction)
        migration_transaction { log_queries { public_send(direction) } }
      end

      def handle_migration_error!(e)
        fail e unless e.message =~ /Cannot perform data updates in a transaction that has performed schema updates./
        fail MigrationError,
             "#{e.message}. Please add `disable_transactions!` in your migration file."
      end

      def migration_transaction(&block)
        ActiveBase.run_transaction(transactions?, &block)
      end

      def log_queries
        subscriber = Neo4j::Core::CypherSession::Adaptors::Base.subscribe_to_query(&method(:output))
        yield
      ensure
        ActiveSupport::Notifications.unsubscribe(subscriber) if subscriber
      end
    end
  end
end

Version data entries

13 entries across 13 versions & 1 rubygems

Version	Path
neo4j-8.0.0.rc.3	lib/neo4j/migrations/base.rb
neo4j-8.0.0.rc.2	lib/neo4j/migrations/base.rb
neo4j-8.0.0.rc.1	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.12	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.11	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.10	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.9	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.8	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.7	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.6	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.5	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.4	lib/neo4j/migrations/base.rb
neo4j-8.0.0.alpha.2	lib/neo4j/migrations/base.rb