def create_kinesis_stream(stream_name, shards=1, env=None, delete=False): env = get_environment(env) # stream stream = KinesisStream(id=stream_name, num_shards=shards) conn = connect_to_service('kinesis', env=env) stream.connect(conn) if delete: run_safe(lambda: stream.destroy(), print_error=False) stream.create() stream.wait_for() return stream
def create_kinesis_stream(stream_name, shards=1, env=None, delete=False): env = get_environment(env) # stream stream = KinesisStream(id=stream_name, num_shards=shards) conn = connect_to_service('kinesis', env=env) stream.connect(conn) if delete: run_safe(lambda: stream.destroy(), print_error=False) stream.create() stream.wait_for() return stream
def create_kinesis_stream(stream_name, shards=1, env=None, delete=False): env = get_environment(env) # stream stream = KinesisStream(id=stream_name, num_shards=shards) conn = connect_to_service('kinesis', env=env) stream.connect(conn) if delete: run_safe(lambda: stream.destroy(), print_error=False) stream.create() # Note: Returning the stream without awaiting its creation (via wait_for()) to avoid API call timeouts/retries. return stream