80 lines
2.1 KiB
Ruby
80 lines
2.1 KiB
Ruby
require "influxdb"
|
|
require "colorize"
|
|
require "benchmark"
|
|
|
|
require_relative "log"
|
|
require_relative "random_gaussian"
|
|
|
|
BATCH_SIZE = 10_000
|
|
|
|
Log.info "Starting burn-in suite"
|
|
master = InfluxDB::Client.new
|
|
master.delete_database("burn-in") rescue nil
|
|
master.create_database("burn-in")
|
|
master.create_database_user("burn-in", "user", "pass")
|
|
|
|
master.database = "burn-in"
|
|
# master.query "select * from test1 into test2;"
|
|
# master.query "select count(value) from test1 group by time(1m) into test2;"
|
|
|
|
influxdb = InfluxDB::Client.new "burn-in", username: "user", password: "pass"
|
|
|
|
Log.success "Connected to server #{influxdb.host}:#{influxdb.port}"
|
|
|
|
Log.log "Creating RandomGaussian(500, 25)"
|
|
gaussian = RandomGaussian.new(500, 25)
|
|
point_count = 0
|
|
|
|
while true
|
|
Log.log "Generating 10,000 points.."
|
|
points = []
|
|
BATCH_SIZE.times do |n|
|
|
points << {value: gaussian.rand.to_i.abs}
|
|
end
|
|
point_count += points.length
|
|
|
|
Log.info "Sending points to server.."
|
|
begin
|
|
st = Time.now
|
|
foo = influxdb.write_point("test1", points)
|
|
et = Time.now
|
|
Log.log foo.inspect
|
|
Log.log "#{et-st} seconds elapsed"
|
|
Log.success "Write successful."
|
|
rescue => e
|
|
Log.failure "Write failed:"
|
|
Log.log e
|
|
end
|
|
sleep 0.5
|
|
|
|
Log.info "Checking regular points"
|
|
st = Time.now
|
|
response = influxdb.query("select count(value) from test1;")
|
|
et = Time.now
|
|
|
|
Log.log "#{et-st} seconds elapsed"
|
|
|
|
response_count = response["test1"].first["count"]
|
|
if point_count == response_count
|
|
Log.success "Point counts match: #{point_count} == #{response_count}"
|
|
else
|
|
Log.failure "Point counts don't match: #{point_count} != #{response_count}"
|
|
end
|
|
|
|
# Log.info "Checking continuous query points for test2"
|
|
# st = Time.now
|
|
# response = influxdb.query("select count(value) from test2;")
|
|
# et = Time.now
|
|
|
|
# Log.log "#{et-st} seconds elapsed"
|
|
|
|
# response_count = response["test2"].first["count"]
|
|
# if point_count == response_count
|
|
# Log.success "Point counts match: #{point_count} == #{response_count}"
|
|
# else
|
|
# Log.failure "Point counts don't match: #{point_count} != #{response_count}"
|
|
# end
|
|
end
|
|
|
|
|