|
7 | 7 | // |
8 | 8 |
|
9 | 9 | import Foundation |
| 10 | +import os.log |
10 | 11 |
|
11 | 12 |
|
12 | 13 | final class DiagnosticLogger { |
13 | | - private lazy var isSimulator: Bool = TARGET_OS_SIMULATOR != 0 |
| 14 | + private let isSimulator: Bool = TARGET_OS_SIMULATOR != 0 |
| 15 | + let subsystem: String |
| 16 | + let version: String |
14 | 17 |
|
15 | 18 | var mLabService: MLabService { |
16 | 19 | didSet { |
17 | 20 | try! KeychainManager().setMLabDatabaseName(mLabService.databaseName, APIKey: mLabService.APIKey) |
18 | 21 | } |
19 | 22 | } |
20 | 23 |
|
21 | | - init() { |
| 24 | + var logglyService: LogglyService { |
| 25 | + didSet { |
| 26 | + try! KeychainManager().setLogglyCustomerToken(logglyService.customerToken) |
| 27 | + } |
| 28 | + } |
| 29 | + |
| 30 | + let remoteLogLevel: OSLogType |
| 31 | + |
| 32 | + static var shared: DiagnosticLogger? |
| 33 | + |
| 34 | + init(subsystem: String, version: String) { |
| 35 | + self.subsystem = subsystem |
| 36 | + self.version = version |
| 37 | + remoteLogLevel = isSimulator ? .fault : .info |
| 38 | + |
22 | 39 | if let (databaseName, APIKey) = KeychainManager().getMLabCredentials() { |
23 | 40 | mLabService = MLabService(databaseName: databaseName, APIKey: APIKey) |
24 | 41 | } else { |
25 | 42 | mLabService = MLabService(databaseName: nil, APIKey: nil) |
26 | 43 | } |
| 44 | + |
| 45 | + let customerToken = KeychainManager().getLogglyCustomerToken() |
| 46 | + logglyService = LogglyService(customerToken: customerToken) |
27 | 47 | } |
28 | 48 |
|
29 | | - func addMessage(_ message: [String: Any], toCollection collection: String) { |
30 | | - if !isSimulator, |
31 | | - let messageData = try? JSONSerialization.data(withJSONObject: message, options: []), |
32 | | - let task = mLabService.uploadTaskWithData(messageData, inCollection: collection) |
33 | | - { |
34 | | - task.resume() |
35 | | - } else { |
36 | | - NSLog("%@: %@", collection, message) |
| 49 | + func forCategory(_ category: String) -> CategoryLogger { |
| 50 | + return CategoryLogger(logger: self, category: category) |
| 51 | + } |
| 52 | +} |
| 53 | + |
| 54 | + |
| 55 | +extension OSLogType { |
| 56 | + fileprivate var tagName: String { |
| 57 | + switch self { |
| 58 | + case let t where t == .info: |
| 59 | + return "info" |
| 60 | + case let t where t == .debug: |
| 61 | + return "debug" |
| 62 | + case let t where t == .error: |
| 63 | + return "error" |
| 64 | + case let t where t == .fault: |
| 65 | + return "fault" |
| 66 | + default: |
| 67 | + return "default" |
| 68 | + } |
| 69 | + } |
| 70 | +} |
| 71 | + |
| 72 | + |
| 73 | +final class CategoryLogger { |
| 74 | + private let logger: DiagnosticLogger |
| 75 | + let category: String |
| 76 | + |
| 77 | + private let systemLog: OSLog |
| 78 | + |
| 79 | + fileprivate init(logger: DiagnosticLogger, category: String) { |
| 80 | + self.logger = logger |
| 81 | + self.category = category |
| 82 | + |
| 83 | + systemLog = OSLog(subsystem: logger.subsystem, category: category) |
| 84 | + } |
| 85 | + |
| 86 | + private func remoteLog(_ type: OSLogType, message: String) { |
| 87 | + guard logger.remoteLogLevel.rawValue <= type.rawValue else { |
| 88 | + return |
| 89 | + } |
| 90 | + |
| 91 | + logger.logglyService.client?.send(message, tags: [type.tagName, category]) |
| 92 | + } |
| 93 | + |
| 94 | + private func remoteLog(_ type: OSLogType, message: [String: Any]) { |
| 95 | + guard logger.remoteLogLevel.rawValue <= type.rawValue else { |
| 96 | + return |
37 | 97 | } |
| 98 | + |
| 99 | + logger.logglyService.client?.send(message, tags: [type.tagName, category]) |
| 100 | + |
| 101 | + // Legacy mLab logging. To be removed. |
| 102 | + if let messageData = try? JSONSerialization.data(withJSONObject: message, options: []) { |
| 103 | + logger.mLabService.uploadTaskWithData(messageData, inCollection: category)?.resume() |
| 104 | + } |
| 105 | + } |
| 106 | + |
| 107 | + func debug(_ message: [String: Any]) { |
| 108 | + systemLog.debug("%{public}@", String(describing: message)) |
| 109 | + remoteLog(.debug, message: message) |
| 110 | + } |
| 111 | + |
| 112 | + func debug(_ message: String) { |
| 113 | + systemLog.error("%{public}@", message) |
| 114 | + remoteLog(.debug, message: message) |
| 115 | + } |
| 116 | + |
| 117 | + func info(_ message: [String: Any]) { |
| 118 | + systemLog.info("%{public}@", String(describing: message)) |
| 119 | + remoteLog(.info, message: message) |
| 120 | + } |
| 121 | + |
| 122 | + func info(_ message: String) { |
| 123 | + systemLog.error("%{public}@", message) |
| 124 | + remoteLog(.info, message: message) |
| 125 | + } |
| 126 | + |
| 127 | + func error(_ message: [String: Any]) { |
| 128 | + systemLog.error("%{public}@", String(reflecting: message)) |
| 129 | + remoteLog(.error, message: message) |
| 130 | + } |
| 131 | + |
| 132 | + func error(_ message: String) { |
| 133 | + systemLog.error("%{public}@", message) |
| 134 | + remoteLog(.error, message: message) |
| 135 | + } |
| 136 | + |
| 137 | + func error(_ error: Error) { |
| 138 | + self.error(String(reflecting: error)) |
38 | 139 | } |
39 | 140 | } |
40 | 141 |
|
0 commit comments