Skip to content

Commit

Permalink
Removed extraneous line break
Browse files Browse the repository at this point in the history
  • Loading branch information
wmtaff committed Jan 14, 2021
1 parent a721e23 commit cfe2aa8
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions tools/kinesis_ingestor/timestream_kinesis_data_gen.py
Original file line number Diff line number Diff line change
Expand Up @@ -257,11 +257,10 @@ def send_records_to_kinesis(all_dimensions, kinesis_client, stream_name, sleep_t
kinesis_client.put_records(StreamName=stream_name, Records=records)

print("Wrote {} records to Kinesis Stream '{}'".format(len(metrics), stream_name))

if sleep_time > 0:
time.sleep(float(sleep_time))


def main(args):
global utilizationRand
global lowUtilizationHosts
Expand Down Expand Up @@ -299,7 +298,8 @@ def signal_handler(sig, frame):
print("Unable to describe Kinesis Stream '{}' in region {}".format(stream_name, region_name))
sys.exit(0)

send_records_to_kinesis(dimension_measures + dimensions_events, kinesis_client, stream_name, sleep_time, percent_late, late_time)
send_records_to_kinesis(dimension_measures + dimensions_events,
kinesis_client, stream_name, sleep_time, percent_late, late_time)


if __name__ == "__main__":
Expand All @@ -315,13 +315,13 @@ def signal_handler(sig, frame):
parser.add_argument('--profile', action="store", type=str, default=None, help="The AWS Config profile to use.")

# Optional sleep timer to slow down data
parser.add_argument('--sleep-time', action="store", type=int, default=0,
parser.add_argument('--sleep-time', action="store", type=int, default=0,
help="The amount of time in seconds to sleep between sending batches.")

# Optional "Late" arriving data parameters
parser.add_argument('--percent-late', action="store", type=float, default=0,
parser.add_argument('--percent-late', action="store", type=float, default=0,
help="The percentage of data written that is late arriving ")
parser.add_argument("--late-time", action="store", type=int, default=0,
parser.add_argument("--late-time", action="store", type=int, default=0,
help="The amount of time in seconds late that the data arrives")

main(parser.parse_args())

0 comments on commit cfe2aa8

Please sign in to comment.