Class: Tapsoob::Push

Inherits:
Operation show all
Defined in:
lib/tapsoob/operation.rb

Instance Attribute Summary

Attributes inherited from Operation

#database_url, #dump_path, #opts

Instance Method Summary collapse

Methods inherited from Operation

#apply_table_filter, #catch_errors, #completed_tables, #db, #default_chunksize, #exclude_tables, #exiting?, factory, #format_number, #indexes_first?, #initialize, #log, #resuming?, #setup_signal_trap, #skip_schema?, #store_session, #stream_state, #stream_state=, #table_filter

Constructor Details

This class inherits a constructor from Tapsoob::Operation

Instance Method Details

#fetch_local_tables_info ⇒ Object 



482
483
484
485
486
487
488
489
490
491
492
493
494
 
# File 'lib/tapsoob/operation.rb', line 482

def fetch_local_tables_info
  tables_with_counts = {}
  tbls = Dir.glob(File.join(dump_path, "schemas", "*")).map { |path| File.basename(path, ".rb") }
  tbls.each do |table|
    if File.exists?(File.join(dump_path, "data", "#{table}.json"))
      data = JSON.parse(File.read(File.join(dump_path, "data", "#{table}.json")))
      tables_with_counts[table] = data["data"].size
    else
      tables_with_counts[table] = 0
    end
  end
  apply_table_filter(tables_with_counts)
end

#file_prefix ⇒ Object 



313
314
315
 
# File 'lib/tapsoob/operation.rb', line 313

def file_prefix
  "push"
end

#local_tables_info ⇒ Object 



465
466
467
 
# File 'lib/tapsoob/operation.rb', line 465

def local_tables_info
  opts[:local_tables_info] ||= fetch_local_tables_info
end

#push_data ⇒ Object 



386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
 
# File 'lib/tapsoob/operation.rb', line 386

def push_data
  log.info "Sending data"

  log.info "#{tables.size} tables, #{format_number(record_count)} records"

  tables.each do |table_name, count|
    next unless File.exists?(File.join(dump_path, "data", "#{table_name}.json")) || File.exists?(File.join(dump_path, "data", "#{table_name}.json")) && JSON.parse(File.read(File.join(dump_path, "data", "#{table_name}.json")))["data"].size == 0
    db[table_name.to_sym].truncate if @opts[:purge]
    stream = Tapsoob::DataStream.factory(db, {
      :table_name => table_name,
      :chunksize => default_chunksize
    }, {
      :"skip-duplicates" => opts[:"skip-duplicates"] || false,
      :"discard-identity" => opts[:"discard-identity"] || false,
      :purge => opts[:purge] || false,
      :debug => opts[:debug]
    })
    progress = ProgressBar.new(table_name.to_s, count)
    push_data_from_file(stream, progress)
  end
end

#push_data_from_file(stream, progress) ⇒ Object 



408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
 
# File 'lib/tapsoob/operation.rb', line 408

def push_data_from_file(stream, progress)
  loop do
    if exiting?
      store_session
      exit 0
    end

    row_size = 0
    chunksize = stream.state[:chunksize]

    begin
      chunksize = Tapsoob::Utils.calculate_chunksize(chunksize) do |c|
        stream.state[:chunksize] = c.to_i
        encoded_data, row_size, elapsed_time = nil
        d1 = c.time_delta do
          encoded_data, row_size, elapsed_time = stream.fetch({ :type => "file", :source => dump_path })
        end
        break if stream.complete?

        data = nil
        d2 = c.time_delta do
          data = {
            :state    => stream.to_hash,
            :checksum => Tapsoob::Utils.checksum(encoded_data).to_s
          }
        end

        row_size = stream.fetch_data_in_database({ :encoded_data => encoded_data, :checksum => data[:checksum] })
        log.debug "row size: #{row_size}"
        self.stream_state = stream.to_hash

        c.idle_secs = (d1 + d2)

        elapsed_time
      end
    rescue Tapsoob::CorruptedData => e
      # retry the same data, it got corrupted somehow.
      next
    rescue Tapsoob::DuplicatePrimaryKeyError => e
      # verify the stream and retry it
      stream.verify_stream
      stream = JSON.generate({ :state => stream.to_hash })
      next
    end
    stream.state[:chunksize] = chunksize

    progress.inc(row_size)

    stream.increment(row_size)
    break if stream.complete?
  end

  progress.finish
  completed_tables << stream.table_name.to_s
  self.stream_state = {}
end

#push_indexes ⇒ Object 



335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
 
# File 'lib/tapsoob/operation.rb', line 335

def push_indexes
  idxs = {}
  table_idxs = Dir.glob(File.join(dump_path, "indexes", "*.json")).map { |path| File.basename(path, '.json') }
  table_idxs.each do |table_idx|
    idxs[table_idx] = JSON.parse(File.read(File.join(dump_path, "indexes", "#{table_idx}.json")))
  end

  return unless idxs.size > 0

  log.info "Sending indexes"

  apply_table_filter(idxs).each do |table, indexes|
    next unless indexes.size > 0
    progress = ProgressBar.new(table, indexes.size)
    indexes.each do |idx|
      Tapsoob::Utils.load_indexes(database_url, idx)
      progress.inc(1)
    end
    progress.finish
  end
end

#push_partial_data ⇒ Object 



375
376
377
378
379
380
381
382
383
384
 
# File 'lib/tapsoob/operation.rb', line 375

def push_partial_data
  return if stream_state == {}

  table_name = stream_state[:table_name]
  record_count = tables[table_name.to_s]
  log.info "Resuming #{table_name}, #{format_number(record_count)} records"
  progress = ProgressBar.new(table_name.to_s, record_count)
  stream = Tapsoob::DataStream.factory(db, stream_state)
  push_data_from_file(stream, progress)
end

#push_reset_sequences ⇒ Object 



369
370
371
372
373
 
# File 'lib/tapsoob/operation.rb', line 369

def push_reset_sequences
  log.info "Resetting sequences"

  Tapsoob::Utils.schema_bin(:reset_db_sequences, database_url)
end

#push_schema ⇒ Object 



357
358
359
360
361
362
363
364
365
366
367
 
# File 'lib/tapsoob/operation.rb', line 357

def push_schema
  log.info "Sending schema"

  progress = ProgressBar.new('Schema', tables.size)
  tables.each do |table, count|
    log.debug "Loading '#{table}' schema\n"
    Tapsoob::Utils.load_schema(dump_path, database_url, table)
    progress.inc(1)
  end
  progress.finish
end

#record_count ⇒ Object 



478
479
480
 
# File 'lib/tapsoob/operation.rb', line 478

def record_count
  @record_count ||= local_tables_info.values.inject(0) { |a,c| a += c }
end

#run ⇒ Object 



321
322
323
324
325
326
327
328
329
330
331
332
333
 
# File 'lib/tapsoob/operation.rb', line 321

def run
  catch_errors do
    unless resuming?
      push_schema if !skip_schema?
      push_indexes if indexes_first? && !skip_schema?
    end
    setup_signal_trap
    push_partial_data if resuming?
    push_data
    push_indexes if !indexes_first? && !skip_schema?
    push_reset_sequences
  end
end

#tables ⇒ Object 



469
470
471
472
473
474
475
476
 
# File 'lib/tapsoob/operation.rb', line 469

def tables
  h = {}
  local_tables_info.each do |table_name, count|
    next if completed_tables.include?(table_name.to_s)
    h[table_name.to_s] = count
  end
  h
end

#to_hash ⇒ Object 



317
318
319
 
# File 'lib/tapsoob/operation.rb', line 317

def to_hash
  super.merge(:local_tables_info => local_tables_info)
end