Sha256: c92b7fa03cc6cf2bf48a54cc6a316a580b0dbf966231479174bea8f8ee7ab478
Contents?: true
Size: 1.51 KB
Versions: 17
Compression:
Stored size: 1.51 KB
Contents
module TreasureData
module Command
def query(op)
db_name = nil
wait = false
output = nil
format = 'tsv'
result = nil
op.on('-d', '--database DB_NAME', 'use the database (required)') {|s|
db_name = s
}
op.on('-w', '--wait', 'wait for finishing the job', TrueClass) {|b|
wait = b
}
op.on('-r', '--result RESULT_TABLE', 'write result to the result table (use result:create command)') {|s|
result = s
}
op.on('-o', '--output PATH', 'write result to the file') {|s|
output = s
}
op.on('-f', '--format FORMAT', 'format of the result to write to the file (tsv, csv, json or msgpack)') {|s|
unless ['tsv', 'csv', 'json', 'msgpack'].include?(s)
raise "Unknown format #{s.dump}. Supported format: tsv, csv, json, msgpack"
end
format = s
}
sql = op.cmd_parse
unless db_name
$stderr.puts "-d, --database DB_NAME option is required."
exit 1
end
client = get_client
# local existance check
get_database(client, db_name)
job = client.query(db_name, sql, result)
$stderr.puts "Job #{job.job_id} is queued."
$stderr.puts "Use '#{$prog} job:show #{job.job_id}' to show the status."
$stderr.puts "See #{job.url} to see the progress."
if wait && !job.finished?
wait_job(job)
puts "Status : #{job.status}"
if job.success?
puts "Result :"
show_result(job, output, format)
end
end
end
require 'td/command/job' # wait_job
end
end
Version data entries
17 entries across 17 versions & 1 rubygems