Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
67 changes: 65 additions & 2 deletions lib/cadet/logger/cloudwatch_logger.ex
Original file line number Diff line number Diff line change
Expand Up @@ -27,13 +27,17 @@ defmodule Cadet.Logger.CloudWatchLogger do
@impl true
def init({__MODULE__, opts}) when is_list(opts) do
config = configure_merge(read_env(), opts)
{:ok, init(config, %__MODULE__{})}
state = init(config, %__MODULE__{})
ensure_log_stream_exists(state.log_group, state.log_stream)
{:ok, state}
end

@impl true
def init({__MODULE__, name}) when is_atom(name) do
config = read_env()
{:ok, init(config, %__MODULE__{})}
state = init(config, %__MODULE__{})
ensure_log_stream_exists(state.log_group, state.log_stream)
{:ok, state}
end

@impl true
Expand Down Expand Up @@ -159,6 +163,65 @@ defmodule Cadet.Logger.CloudWatchLogger do
end
end

# Ensures the log stream exists, creates it if not
# Returns :ok or :error
# Uses ExAws.Logs.describe_log_streams and ExAws.Logs.create_log_stream
# Assumes ExAws.Logs is available

defp ensure_log_stream_exists(log_group, log_stream) do
describe_op = %ExAws.Operation.JSON{
http_method: :post,
service: :logs,
headers: [
{"x-amz-target", "Logs_20140328.DescribeLogStreams"},
{"content-type", "application/x-amz-json-1.1"}
],
data: %{
"logGroupName" => log_group,
"logStreamNamePrefix" => log_stream
}
}

client = Application.get_env(:ex_aws, :ex_aws_mock, ExAws)

case client.request(describe_op) do
{:ok, %{"logStreams" => streams}} ->
if Enum.any?(streams, fn s -> s["logStreamName"] == log_stream end) do
:ok
else
create_log_stream(log_group, log_stream, client)
end

{:error, reason} ->
Logger.error("Failed to describe log streams: #{inspect(reason)}")
:error
end
end

defp create_log_stream(log_group, log_stream, client) do
create_op = %ExAws.Operation.JSON{
http_method: :post,
service: :logs,
headers: [
{"x-amz-target", "Logs_20140328.CreateLogStream"},
{"content-type", "application/x-amz-json-1.1"}
],
data: %{
"logGroupName" => log_group,
"logStreamName" => log_stream
}
}

case client.request(create_op) do
{:ok, _} ->
:ok

{:error, reason} ->
Logger.error("Failed to create log stream: #{inspect(reason)}")
:error
end
end

defp build_log_operation(log_stream, log_group, buffer) do
# The headers and body structure can be found in the AWS API documentation:
# https://docs.aws.amazon.com/AmazonCloudWatchLogs/latest/APIReference/API_PutLogEvents.html
Expand Down
20 changes: 20 additions & 0 deletions test/cadet/logger/cloudwatch_logger_test.exs
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,26 @@ defmodule Cadet.Logger.CloudWatchLoggerTest do
metadata: [:request_id]
)

# Mock the DescribeLogStreams request during initialization
expect(ExAwsMock, :request, fn %ExAws.Operation.JSON{} = op
when op.headers == [
{"x-amz-target", "Logs_20140328.DescribeLogStreams"},
{"content-type", "application/x-amz-json-1.1"}
] ->
# Simulate no existing log streams
{:ok, %{"logStreams" => []}}
end)

# Mock the CreateLogStream request during initialization
expect(ExAwsMock, :request, fn %ExAws.Operation.JSON{} = op
when op.headers == [
{"x-amz-target", "Logs_20140328.CreateLogStream"},
{"content-type", "application/x-amz-json-1.1"}
] ->
# Simulate successful log stream creation
{:ok, %{}}
end)

LoggerBackends.add({CloudWatchLogger, :cloudwatch_logger})

Logger.configure_backend(:console, level: :error, format: "$metadata[$level] $message\n")
Expand Down
Loading