Contents

module CassandraObject
  class Schema
    module Tasks
      def dump(io)
        column_families.each do |column_family|
          io.puts run_command("DESCRIBE COLUMNFAMILY #{column_family}")
          io.puts
        end
      end

      def load(io)
        current_cql = ''

        io.each_line do |line|
          next if line.blank?

          current_cql << line.rstrip

          if current_cql =~ /;$/
            CassandraObject::Base.execute_cql current_cql
            current_cql = ''
          end
        end
      end

      def column_families
        run_command('DESCRIBE COLUMNFAMILIES').split.sort
      end

      private
        def run_command(command)
          `echo "#{command};" | #{cqlsh} -2 -k #{keyspace} #{server}`.sub(/^(.*)$/, '').strip
        end

        def cqlsh
          ENV['CQLSH'] || 'cqlsh'
        end

        def keyspace
          CassandraObject::Base.config.keyspace
        end

        def server
          CassandraObject::Base.config.servers.first.gsub(/:.*/, '')
        end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
gotime-cassandra_object-4.11.6	lib/cassandra_object/schema/tasks.rb
gotime-cassandra_object-4.11.5	lib/cassandra_object/schema/tasks.rb