1 | /* |
2 | * Copyright 2010-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved. |
3 | * |
4 | * Licensed under the Apache License, Version 2.0 (the "License"). |
5 | * You may not use this file except in compliance with the License. |
6 | * A copy of the License is located at |
7 | * |
8 | * http://aws.amazon.com/apache2.0 |
9 | * |
10 | * or in the "license" file accompanying this file. This file is distributed |
11 | * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either |
12 | * express or implied. See the License for the specific language governing |
13 | * permissions and limitations under the License. |
14 | */ |
15 | |
16 | |
17 | #include <aws/core/utils/logging/FormattedLogSystem.h> |
18 | |
19 | #include <aws/core/utils/DateTime.h> |
20 | #include <aws/core/utils/Array.h> |
21 | |
22 | #include <fstream> |
23 | #include <cstdarg> |
24 | #include <stdio.h> |
25 | #include <thread> |
26 | |
27 | using namespace Aws::Utils; |
28 | using namespace Aws::Utils::Logging; |
29 | |
30 | static Aws::String CreateLogPrefixLine(LogLevel logLevel, const char* tag) |
31 | { |
32 | Aws::StringStream ss; |
33 | |
34 | switch(logLevel) |
35 | { |
36 | case LogLevel::Error: |
37 | ss << "[ERROR] " ; |
38 | break; |
39 | |
40 | case LogLevel::Fatal: |
41 | ss << "[FATAL] " ; |
42 | break; |
43 | |
44 | case LogLevel::Warn: |
45 | ss << "[WARN] " ; |
46 | break; |
47 | |
48 | case LogLevel::Info: |
49 | ss << "[INFO] " ; |
50 | break; |
51 | |
52 | case LogLevel::Debug: |
53 | ss << "[DEBUG] " ; |
54 | break; |
55 | |
56 | case LogLevel::Trace: |
57 | ss << "[TRACE] " ; |
58 | break; |
59 | |
60 | default: |
61 | ss << "[UNKOWN] " ; |
62 | break; |
63 | } |
64 | |
65 | ss << DateTime::Now().CalculateGmtTimeWithMsPrecision() << " " << tag << " [" << std::this_thread::get_id() << "] " ; |
66 | |
67 | return ss.str(); |
68 | } |
69 | |
70 | FormattedLogSystem::FormattedLogSystem(LogLevel logLevel) : |
71 | m_logLevel(logLevel) |
72 | { |
73 | } |
74 | |
75 | void FormattedLogSystem::Log(LogLevel logLevel, const char* tag, const char* formatStr, ...) |
76 | { |
77 | Aws::StringStream ss; |
78 | ss << CreateLogPrefixLine(logLevel, tag); |
79 | |
80 | std::va_list args; |
81 | va_start(args, formatStr); |
82 | |
83 | va_list tmp_args; //unfortunately you cannot consume a va_list twice |
84 | va_copy(tmp_args, args); //so we have to copy it |
85 | #ifdef WIN32 |
86 | const int requiredLength = _vscprintf(formatStr, tmp_args) + 1; |
87 | #else |
88 | const int requiredLength = vsnprintf(nullptr, 0, formatStr, tmp_args) + 1; |
89 | #endif |
90 | va_end(tmp_args); |
91 | |
92 | Array<char> outputBuff(requiredLength); |
93 | #ifdef WIN32 |
94 | vsnprintf_s(outputBuff.GetUnderlyingData(), requiredLength, _TRUNCATE, formatStr, args); |
95 | #else |
96 | vsnprintf(outputBuff.GetUnderlyingData(), requiredLength, formatStr, args); |
97 | #endif // WIN32 |
98 | |
99 | ss << outputBuff.GetUnderlyingData() << std::endl; |
100 | |
101 | ProcessFormattedStatement(ss.str()); |
102 | |
103 | va_end(args); |
104 | } |
105 | |
106 | void FormattedLogSystem::LogStream(LogLevel logLevel, const char* tag, const Aws::OStringStream &message_stream) |
107 | { |
108 | ProcessFormattedStatement(CreateLogPrefixLine(logLevel, tag) + message_stream.str() + "\n" ); |
109 | } |
110 | |