123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284 |
- // Software License Agreement (BSD License)
- //
- // Copyright (c) 2010-2018, Deusty, LLC
- // All rights reserved.
- //
- // Redistribution and use of this software in source and binary forms,
- // with or without modification, are permitted provided that the following conditions are met:
- //
- // * Redistributions of source code must retain the above copyright notice,
- // this list of conditions and the following disclaimer.
- //
- // * Neither the name of Deusty nor the names of its contributors may be used
- // to endorse or promote products derived from this software without specific
- // prior written permission of Deusty, LLC.
- // Disable legacy macros
- #ifndef DD_LEGACY_MACROS
- #define DD_LEGACY_MACROS 0
- #endif
- #import "DDLog.h"
- #import <pthread.h>
- #import <objc/runtime.h>
- #if TARGET_OS_IOS
- #import <UIKit/UIDevice.h>
- #import <UIKit/UIApplication.h>
- #elif !defined(DD_CLI) && __has_include(<AppKit/NSApplication.h>)
- #import <AppKit/NSApplication.h>
- #endif
- #if !__has_feature(objc_arc)
- #error This file must be compiled with ARC. Use -fobjc-arc flag (or convert project to ARC).
- #endif
- // We probably shouldn't be using DDLog() statements within the DDLog implementation.
- // But we still want to leave our log statements for any future debugging,
- // and to allow other developers to trace the implementation (which is a great learning tool).
- //
- // So we use a primitive logging macro around NSLog.
- // We maintain the NS prefix on the macros to be explicit about the fact that we're using NSLog.
- #ifndef DD_DEBUG
- #define DD_DEBUG NO
- #endif
- #define NSLogDebug(frmt, ...) do{ if(DD_DEBUG) NSLog((frmt), ##__VA_ARGS__); } while(0)
- // Specifies the maximum queue size of the logging thread.
- //
- // Since most logging is asynchronous, its possible for rogue threads to flood the logging queue.
- // That is, to issue an abundance of log statements faster than the logging thread can keep up.
- // Typically such a scenario occurs when log statements are added haphazardly within large loops,
- // but may also be possible if relatively slow loggers are being used.
- //
- // This property caps the queue size at a given number of outstanding log statements.
- // If a thread attempts to issue a log statement when the queue is already maxed out,
- // the issuing thread will block until the queue size drops below the max again.
- #ifndef DDLOG_MAX_QUEUE_SIZE
- #define DDLOG_MAX_QUEUE_SIZE 1000 // Should not exceed INT32_MAX
- #endif
- // The "global logging queue" refers to [DDLog loggingQueue].
- // It is the queue that all log statements go through.
- //
- // The logging queue sets a flag via dispatch_queue_set_specific using this key.
- // We can check for this key via dispatch_get_specific() to see if we're on the "global logging queue".
- static void *const GlobalLoggingQueueIdentityKey = (void *)&GlobalLoggingQueueIdentityKey;
- @interface DDLoggerNode : NSObject
- {
- // Direct accessors to be used only for performance
- @public
- id <DDLogger> _logger;
- DDLogLevel _level;
- dispatch_queue_t _loggerQueue;
- }
- @property (nonatomic, readonly) id <DDLogger> logger;
- @property (nonatomic, readonly) DDLogLevel level;
- @property (nonatomic, readonly) dispatch_queue_t loggerQueue;
- + (DDLoggerNode *)nodeWithLogger:(id <DDLogger>)logger
- loggerQueue:(dispatch_queue_t)loggerQueue
- level:(DDLogLevel)level;
- @end
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark -
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- @interface DDLog ()
- // An array used to manage all the individual loggers.
- // The array is only modified on the loggingQueue/loggingThread.
- @property (nonatomic, strong) NSMutableArray *_loggers;
- @end
- @implementation DDLog
- // All logging statements are added to the same queue to ensure FIFO operation.
- static dispatch_queue_t _loggingQueue;
- // Individual loggers are executed concurrently per log statement.
- // Each logger has it's own associated queue, and a dispatch group is used for synchronization.
- static dispatch_group_t _loggingGroup;
- // In order to prevent to queue from growing infinitely large,
- // a maximum size is enforced (DDLOG_MAX_QUEUE_SIZE).
- static dispatch_semaphore_t _queueSemaphore;
- // Minor optimization for uniprocessor machines
- static NSUInteger _numProcessors;
- /**
- * Returns the singleton `DDLog`.
- * The instance is used by `DDLog` class methods.
- *
- * @return The singleton `DDLog`.
- */
- + (instancetype)sharedInstance {
- static id sharedInstance = nil;
-
- static dispatch_once_t onceToken;
- dispatch_once(&onceToken, ^{
- sharedInstance = [[self alloc] init];
- });
-
- return sharedInstance;
- }
- /**
- * The runtime sends initialize to each class in a program exactly one time just before the class,
- * or any class that inherits from it, is sent its first message from within the program. (Thus the
- * method may never be invoked if the class is not used.) The runtime sends the initialize message to
- * classes in a thread-safe manner. Superclasses receive this message before their subclasses.
- *
- * This method may also be called directly, hence the safety mechanism.
- **/
- + (void)initialize {
- static dispatch_once_t DDLogOnceToken;
-
- dispatch_once(&DDLogOnceToken, ^{
- NSLogDebug(@"DDLog: Using grand central dispatch");
-
- _loggingQueue = dispatch_queue_create("cocoa.lumberjack", NULL);
- _loggingGroup = dispatch_group_create();
-
- void *nonNullValue = GlobalLoggingQueueIdentityKey; // Whatever, just not null
- dispatch_queue_set_specific(_loggingQueue, GlobalLoggingQueueIdentityKey, nonNullValue, NULL);
-
- _queueSemaphore = dispatch_semaphore_create(DDLOG_MAX_QUEUE_SIZE);
-
- // Figure out how many processors are available.
- // This may be used later for an optimization on uniprocessor machines.
-
- _numProcessors = MAX([NSProcessInfo processInfo].processorCount, (NSUInteger) 1);
-
- NSLogDebug(@"DDLog: numProcessors = %@", @(_numProcessors));
- });
- }
- /**
- * The `DDLog` initializer.
- * Static variables are set only once.
- *
- * @return An initialized `DDLog` instance.
- */
- - (id)init {
- self = [super init];
-
- if (self) {
- self._loggers = [[NSMutableArray alloc] initWithCapacity:4];
-
- #if TARGET_OS_IOS
- NSString *notificationName = UIApplicationWillTerminateNotification;
- #else
- NSString *notificationName = nil;
- // On Command Line Tool apps AppKit may not be available
- #if !defined(DD_CLI) && __has_include(<AppKit/NSApplication.h>)
- if (NSApp) {
- notificationName = NSApplicationWillTerminateNotification;
- }
- #endif
- if (!notificationName) {
- // If there is no NSApp -> we are running Command Line Tool app.
- // In this case terminate notification wouldn't be fired, so we use workaround.
- __weak __auto_type weakSelf = self;
- atexit_b (^{
- [weakSelf applicationWillTerminate:nil];
- });
- }
-
- #endif /* if TARGET_OS_IOS */
-
- if (notificationName) {
- [[NSNotificationCenter defaultCenter] addObserver:self
- selector:@selector(applicationWillTerminate:)
- name:notificationName
- object:nil];
- }
- }
-
- return self;
- }
- /**
- * Provides access to the logging queue.
- **/
- + (dispatch_queue_t)loggingQueue {
- return _loggingQueue;
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark Notifications
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- - (void)applicationWillTerminate:(NSNotification * __attribute__((unused)))notification {
- [self flushLog];
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark Logger Management
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- + (void)addLogger:(id <DDLogger>)logger {
- [self.sharedInstance addLogger:logger];
- }
- - (void)addLogger:(id <DDLogger>)logger {
- [self addLogger:logger withLevel:DDLogLevelAll]; // DDLogLevelAll has all bits set
- }
- + (void)addLogger:(id <DDLogger>)logger withLevel:(DDLogLevel)level {
- [self.sharedInstance addLogger:logger withLevel:level];
- }
- - (void)addLogger:(id <DDLogger>)logger withLevel:(DDLogLevel)level {
- if (!logger) {
- return;
- }
-
- dispatch_async(_loggingQueue, ^{ @autoreleasepool {
- [self lt_addLogger:logger level:level];
- } });
- }
- + (void)removeLogger:(id <DDLogger>)logger {
- [self.sharedInstance removeLogger:logger];
- }
- - (void)removeLogger:(id <DDLogger>)logger {
- if (!logger) {
- return;
- }
-
- dispatch_async(_loggingQueue, ^{ @autoreleasepool {
- [self lt_removeLogger:logger];
- } });
- }
- + (void)removeAllLoggers {
- [self.sharedInstance removeAllLoggers];
- }
- - (void)removeAllLoggers {
- dispatch_async(_loggingQueue, ^{ @autoreleasepool {
- [self lt_removeAllLoggers];
- } });
- }
- + (NSArray<id<DDLogger>> *)allLoggers {
- return [self.sharedInstance allLoggers];
- }
- - (NSArray<id<DDLogger>> *)allLoggers {
- __block NSArray *theLoggers;
-
- dispatch_sync(_loggingQueue, ^{ @autoreleasepool {
- theLoggers = [self lt_allLoggers];
- } });
-
- return theLoggers;
- }
- + (NSArray<DDLoggerInformation *> *)allLoggersWithLevel {
- return [self.sharedInstance allLoggersWithLevel];
- }
- - (NSArray<DDLoggerInformation *> *)allLoggersWithLevel {
- __block NSArray *theLoggersWithLevel;
-
- dispatch_sync(_loggingQueue, ^{ @autoreleasepool {
- theLoggersWithLevel = [self lt_allLoggersWithLevel];
- } });
-
- return theLoggersWithLevel;
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark - Master Logging
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- - (void)queueLogMessage:(DDLogMessage *)logMessage asynchronously:(BOOL)asyncFlag {
- // We have a tricky situation here...
- //
- // In the common case, when the queueSize is below the maximumQueueSize,
- // we want to simply enqueue the logMessage. And we want to do this as fast as possible,
- // which means we don't want to block and we don't want to use any locks.
- //
- // However, if the queueSize gets too big, we want to block.
- // But we have very strict requirements as to when we block, and how long we block.
- //
- // The following example should help illustrate our requirements:
- //
- // Imagine that the maximum queue size is configured to be 5,
- // and that there are already 5 log messages queued.
- // Let us call these 5 queued log messages A, B, C, D, and E. (A is next to be executed)
- //
- // Now if our thread issues a log statement (let us call the log message F),
- // it should block before the message is added to the queue.
- // Furthermore, it should be unblocked immediately after A has been unqueued.
- //
- // The requirements are strict in this manner so that we block only as long as necessary,
- // and so that blocked threads are unblocked in the order in which they were blocked.
- //
- // Returning to our previous example, let us assume that log messages A through E are still queued.
- // Our aforementioned thread is blocked attempting to queue log message F.
- // Now assume we have another separate thread that attempts to issue log message G.
- // It should block until log messages A and B have been unqueued.
- // We are using a counting semaphore provided by GCD.
- // The semaphore is initialized with our DDLOG_MAX_QUEUE_SIZE value.
- // Every time we want to queue a log message we decrement this value.
- // If the resulting value is less than zero,
- // the semaphore function waits in FIFO order for a signal to occur before returning.
- //
- // A dispatch semaphore is an efficient implementation of a traditional counting semaphore.
- // Dispatch semaphores call down to the kernel only when the calling thread needs to be blocked.
- // If the calling semaphore does not need to block, no kernel call is made.
- dispatch_block_t logBlock = ^{
- dispatch_semaphore_wait(_queueSemaphore, DISPATCH_TIME_FOREVER);
- // We're now sure we won't overflow the queue.
- // It is time to queue our log message.
- @autoreleasepool {
- [self lt_log:logMessage];
- }
- };
- if (asyncFlag) {
- dispatch_async(_loggingQueue, logBlock);
- } else if (dispatch_get_specific(GlobalLoggingQueueIdentityKey)) {
- // We've logged an error message while on the logging queue...
- logBlock();
- } else {
- dispatch_sync(_loggingQueue, logBlock);
- }
- }
- + (void)log:(BOOL)asynchronous
- level:(DDLogLevel)level
- flag:(DDLogFlag)flag
- context:(NSInteger)context
- file:(const char *)file
- function:(const char *)function
- line:(NSUInteger)line
- tag:(id)tag
- format:(NSString *)format, ... {
- va_list args;
-
- if (format) {
- va_start(args, format);
-
- NSString *message = [[NSString alloc] initWithFormat:format arguments:args];
-
- va_end(args);
-
- va_start(args, format);
-
- [self log:asynchronous
- message:message
- level:level
- flag:flag
- context:context
- file:file
- function:function
- line:line
- tag:tag];
-
- va_end(args);
- }
- }
- - (void)log:(BOOL)asynchronous
- level:(DDLogLevel)level
- flag:(DDLogFlag)flag
- context:(NSInteger)context
- file:(const char *)file
- function:(const char *)function
- line:(NSUInteger)line
- tag:(id)tag
- format:(NSString *)format, ... {
- va_list args;
-
- if (format) {
- va_start(args, format);
-
- NSString *message = [[NSString alloc] initWithFormat:format arguments:args];
-
- va_end(args);
-
- va_start(args, format);
-
- [self log:asynchronous
- message:message
- level:level
- flag:flag
- context:context
- file:file
- function:function
- line:line
- tag:tag];
-
- va_end(args);
- }
- }
- + (void)log:(BOOL)asynchronous
- level:(DDLogLevel)level
- flag:(DDLogFlag)flag
- context:(NSInteger)context
- file:(const char *)file
- function:(const char *)function
- line:(NSUInteger)line
- tag:(id)tag
- format:(NSString *)format
- args:(va_list)args {
- [self.sharedInstance log:asynchronous level:level flag:flag context:context file:file function:function line:line tag:tag format:format args:args];
- }
- - (void)log:(BOOL)asynchronous
- level:(DDLogLevel)level
- flag:(DDLogFlag)flag
- context:(NSInteger)context
- file:(const char *)file
- function:(const char *)function
- line:(NSUInteger)line
- tag:(id)tag
- format:(NSString *)format
- args:(va_list)args {
- if (format) {
- NSString *message = [[NSString alloc] initWithFormat:format arguments:args];
- [self log:asynchronous
- message:message
- level:level
- flag:flag
- context:context
- file:file
- function:function
- line:line
- tag:tag];
- }
- }
- + (void)log:(BOOL)asynchronous
- message:(NSString *)message
- level:(DDLogLevel)level
- flag:(DDLogFlag)flag
- context:(NSInteger)context
- file:(const char *)file
- function:(const char *)function
- line:(NSUInteger)line
- tag:(id)tag {
- [self.sharedInstance log:asynchronous message:message level:level flag:flag context:context file:file function:function line:line tag:tag];
- }
- - (void)log:(BOOL)asynchronous
- message:(NSString *)message
- level:(DDLogLevel)level
- flag:(DDLogFlag)flag
- context:(NSInteger)context
- file:(const char *)file
- function:(const char *)function
- line:(NSUInteger)line
- tag:(id)tag {
- DDLogMessage *logMessage = [[DDLogMessage alloc] initWithMessage:message
- level:level
- flag:flag
- context:context
- file:[NSString stringWithFormat:@"%s", file]
- function:[NSString stringWithFormat:@"%s", function]
- line:line
- tag:tag
- options:(DDLogMessageOptions)0
- timestamp:nil];
-
- [self queueLogMessage:logMessage asynchronously:asynchronous];
- }
- + (void)log:(BOOL)asynchronous
- message:(DDLogMessage *)logMessage {
- [self.sharedInstance log:asynchronous message:logMessage];
- }
- - (void)log:(BOOL)asynchronous
- message:(DDLogMessage *)logMessage {
- [self queueLogMessage:logMessage asynchronously:asynchronous];
- }
- + (void)flushLog {
- [self.sharedInstance flushLog];
- }
- - (void)flushLog {
- dispatch_sync(_loggingQueue, ^{ @autoreleasepool {
- [self lt_flush];
- } });
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark Registered Dynamic Logging
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- + (BOOL)isRegisteredClass:(Class)class {
- SEL getterSel = @selector(ddLogLevel);
- SEL setterSel = @selector(ddSetLogLevel:);
- #if TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR
- // Issue #6 (GoogleCode) - Crashes on iOS 4.2.1 and iPhone 4
- //
- // Crash caused by class_getClassMethod(2).
- //
- // "It's a bug with UIAccessibilitySafeCategory__NSObject so it didn't pop up until
- // users had VoiceOver enabled [...]. I was able to work around it by searching the
- // result of class_copyMethodList() instead of calling class_getClassMethod()"
- BOOL result = NO;
- unsigned int methodCount, i;
- Method *methodList = class_copyMethodList(object_getClass(class), &methodCount);
- if (methodList != NULL) {
- BOOL getterFound = NO;
- BOOL setterFound = NO;
- for (i = 0; i < methodCount; ++i) {
- SEL currentSel = method_getName(methodList[i]);
- if (currentSel == getterSel) {
- getterFound = YES;
- } else if (currentSel == setterSel) {
- setterFound = YES;
- }
- if (getterFound && setterFound) {
- result = YES;
- break;
- }
- }
- free(methodList);
- }
- return result;
- #else /* if TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR */
- // Issue #24 (GitHub) - Crashing in in ARC+Simulator
- //
- // The method +[DDLog isRegisteredClass] will crash a project when using it with ARC + Simulator.
- // For running in the Simulator, it needs to execute the non-iOS code.
- Method getter = class_getClassMethod(class, getterSel);
- Method setter = class_getClassMethod(class, setterSel);
- if ((getter != NULL) && (setter != NULL)) {
- return YES;
- }
- return NO;
- #endif /* if TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR */
- }
- + (NSArray *)registeredClasses {
- // We're going to get the list of all registered classes.
- // The Objective-C runtime library automatically registers all the classes defined in your source code.
- //
- // To do this we use the following method (documented in the Objective-C Runtime Reference):
- //
- // int objc_getClassList(Class *buffer, int bufferLen)
- //
- // We can pass (NULL, 0) to obtain the total number of
- // registered class definitions without actually retrieving any class definitions.
- // This allows us to allocate the minimum amount of memory needed for the application.
- NSUInteger numClasses = 0;
- Class *classes = NULL;
- while (numClasses == 0) {
- numClasses = (NSUInteger)MAX(objc_getClassList(NULL, 0), 0);
- // numClasses now tells us how many classes we have (but it might change)
- // So we can allocate our buffer, and get pointers to all the class definitions.
- NSUInteger bufferSize = numClasses;
- classes = numClasses ? (Class *)malloc(sizeof(Class) * bufferSize) : NULL;
- if (classes == NULL) {
- return @[]; //no memory or classes?
- }
- numClasses = (NSUInteger)MAX(objc_getClassList(classes, (int)bufferSize),0);
- if (numClasses > bufferSize || numClasses == 0) {
- //apparently more classes added between calls (or a problem); try again
- free(classes);
- classes = NULL;
- numClasses = 0;
- }
- }
- // We can now loop through the classes, and test each one to see if it is a DDLogging class.
- NSMutableArray *result = [NSMutableArray arrayWithCapacity:numClasses];
- for (NSUInteger i = 0; i < numClasses; i++) {
- Class class = classes[i];
- if ([self isRegisteredClass:class]) {
- [result addObject:class];
- }
- }
- free(classes);
- return result;
- }
- + (NSArray *)registeredClassNames {
- NSArray *registeredClasses = [self registeredClasses];
- NSMutableArray *result = [NSMutableArray arrayWithCapacity:[registeredClasses count]];
- for (Class class in registeredClasses) {
- [result addObject:NSStringFromClass(class)];
- }
- return result;
- }
- + (DDLogLevel)levelForClass:(Class)aClass {
- if ([self isRegisteredClass:aClass]) {
- return [aClass ddLogLevel];
- }
- return (DDLogLevel)-1;
- }
- + (DDLogLevel)levelForClassWithName:(NSString *)aClassName {
- Class aClass = NSClassFromString(aClassName);
- return [self levelForClass:aClass];
- }
- + (void)setLevel:(DDLogLevel)level forClass:(Class)aClass {
- if ([self isRegisteredClass:aClass]) {
- [aClass ddSetLogLevel:level];
- }
- }
- + (void)setLevel:(DDLogLevel)level forClassWithName:(NSString *)aClassName {
- Class aClass = NSClassFromString(aClassName);
- [self setLevel:level forClass:aClass];
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark Logging Thread
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- - (void)lt_addLogger:(id <DDLogger>)logger level:(DDLogLevel)level {
- // Add to loggers array.
- // Need to create loggerQueue if loggerNode doesn't provide one.
- for (DDLoggerNode* node in self._loggers) {
- if (node->_logger == logger
- && node->_level == level) {
- // Exactly same logger already added, exit
- return;
- }
- }
- NSAssert(dispatch_get_specific(GlobalLoggingQueueIdentityKey),
- @"This method should only be run on the logging thread/queue");
- dispatch_queue_t loggerQueue = NULL;
- if ([logger respondsToSelector:@selector(loggerQueue)]) {
- // Logger may be providing its own queue
- loggerQueue = [logger loggerQueue];
- }
- if (loggerQueue == nil) {
- // Automatically create queue for the logger.
- // Use the logger name as the queue name if possible.
- const char *loggerQueueName = NULL;
- if ([logger respondsToSelector:@selector(loggerName)]) {
- loggerQueueName = [[logger loggerName] UTF8String];
- }
- loggerQueue = dispatch_queue_create(loggerQueueName, NULL);
- }
- DDLoggerNode *loggerNode = [DDLoggerNode nodeWithLogger:logger loggerQueue:loggerQueue level:level];
- [self._loggers addObject:loggerNode];
- if ([logger respondsToSelector:@selector(didAddLoggerInQueue:)]) {
- dispatch_async(loggerNode->_loggerQueue, ^{ @autoreleasepool {
- [logger didAddLoggerInQueue:loggerNode->_loggerQueue];
- } });
- } else if ([logger respondsToSelector:@selector(didAddLogger)]) {
- dispatch_async(loggerNode->_loggerQueue, ^{ @autoreleasepool {
- [logger didAddLogger];
- } });
- }
- }
- - (void)lt_removeLogger:(id <DDLogger>)logger {
- // Find associated loggerNode in list of added loggers
- NSAssert(dispatch_get_specific(GlobalLoggingQueueIdentityKey),
- @"This method should only be run on the logging thread/queue");
- DDLoggerNode *loggerNode = nil;
- for (DDLoggerNode *node in self._loggers) {
- if (node->_logger == logger) {
- loggerNode = node;
- break;
- }
- }
-
- if (loggerNode == nil) {
- NSLogDebug(@"DDLog: Request to remove logger which wasn't added");
- return;
- }
-
- // Notify logger
- if ([logger respondsToSelector:@selector(willRemoveLogger)]) {
- dispatch_async(loggerNode->_loggerQueue, ^{ @autoreleasepool {
- [logger willRemoveLogger];
- } });
- }
-
- // Remove from loggers array
- [self._loggers removeObject:loggerNode];
- }
- - (void)lt_removeAllLoggers {
- NSAssert(dispatch_get_specific(GlobalLoggingQueueIdentityKey),
- @"This method should only be run on the logging thread/queue");
-
- // Notify all loggers
- for (DDLoggerNode *loggerNode in self._loggers) {
- if ([loggerNode->_logger respondsToSelector:@selector(willRemoveLogger)]) {
- dispatch_async(loggerNode->_loggerQueue, ^{ @autoreleasepool {
- [loggerNode->_logger willRemoveLogger];
- } });
- }
- }
-
- // Remove all loggers from array
- [self._loggers removeAllObjects];
- }
- - (NSArray *)lt_allLoggers {
- NSAssert(dispatch_get_specific(GlobalLoggingQueueIdentityKey),
- @"This method should only be run on the logging thread/queue");
- NSMutableArray *theLoggers = [NSMutableArray new];
- for (DDLoggerNode *loggerNode in self._loggers) {
- [theLoggers addObject:loggerNode->_logger];
- }
- return [theLoggers copy];
- }
- - (NSArray *)lt_allLoggersWithLevel {
- NSAssert(dispatch_get_specific(GlobalLoggingQueueIdentityKey),
- @"This method should only be run on the logging thread/queue");
-
- NSMutableArray *theLoggersWithLevel = [NSMutableArray new];
-
- for (DDLoggerNode *loggerNode in self._loggers) {
- [theLoggersWithLevel addObject:[DDLoggerInformation informationWithLogger:loggerNode->_logger
- andLevel:loggerNode->_level]];
- }
-
- return [theLoggersWithLevel copy];
- }
- - (void)lt_log:(DDLogMessage *)logMessage {
- // Execute the given log message on each of our loggers.
- NSAssert(dispatch_get_specific(GlobalLoggingQueueIdentityKey),
- @"This method should only be run on the logging thread/queue");
- if (_numProcessors > 1) {
- // Execute each logger concurrently, each within its own queue.
- // All blocks are added to same group.
- // After each block has been queued, wait on group.
- //
- // The waiting ensures that a slow logger doesn't end up with a large queue of pending log messages.
- // This would defeat the purpose of the efforts we made earlier to restrict the max queue size.
- for (DDLoggerNode *loggerNode in self._loggers) {
- // skip the loggers that shouldn't write this message based on the log level
- if (!(logMessage->_flag & loggerNode->_level)) {
- continue;
- }
-
- dispatch_group_async(_loggingGroup, loggerNode->_loggerQueue, ^{ @autoreleasepool {
- [loggerNode->_logger logMessage:logMessage];
- } });
- }
-
- dispatch_group_wait(_loggingGroup, DISPATCH_TIME_FOREVER);
- } else {
- // Execute each logger serially, each within its own queue.
-
- for (DDLoggerNode *loggerNode in self._loggers) {
- // skip the loggers that shouldn't write this message based on the log level
- if (!(logMessage->_flag & loggerNode->_level)) {
- continue;
- }
-
- dispatch_sync(loggerNode->_loggerQueue, ^{ @autoreleasepool {
- [loggerNode->_logger logMessage:logMessage];
- } });
- }
- }
- // If our queue got too big, there may be blocked threads waiting to add log messages to the queue.
- // Since we've now dequeued an item from the log, we may need to unblock the next thread.
- // We are using a counting semaphore provided by GCD.
- // The semaphore is initialized with our DDLOG_MAX_QUEUE_SIZE value.
- // When a log message is queued this value is decremented.
- // When a log message is dequeued this value is incremented.
- // If the value ever drops below zero,
- // the queueing thread blocks and waits in FIFO order for us to signal it.
- //
- // A dispatch semaphore is an efficient implementation of a traditional counting semaphore.
- // Dispatch semaphores call down to the kernel only when the calling thread needs to be blocked.
- // If the calling semaphore does not need to block, no kernel call is made.
- dispatch_semaphore_signal(_queueSemaphore);
- }
- - (void)lt_flush {
- // All log statements issued before the flush method was invoked have now been executed.
- //
- // Now we need to propagate the flush request to any loggers that implement the flush method.
- // This is designed for loggers that buffer IO.
-
- NSAssert(dispatch_get_specific(GlobalLoggingQueueIdentityKey),
- @"This method should only be run on the logging thread/queue");
-
- for (DDLoggerNode *loggerNode in self._loggers) {
- if ([loggerNode->_logger respondsToSelector:@selector(flush)]) {
- dispatch_group_async(_loggingGroup, loggerNode->_loggerQueue, ^{ @autoreleasepool {
- [loggerNode->_logger flush];
- } });
- }
- }
-
- dispatch_group_wait(_loggingGroup, DISPATCH_TIME_FOREVER);
- }
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark Utilities
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- NSString * __nullable DDExtractFileNameWithoutExtension(const char *filePath, BOOL copy) {
- if (filePath == NULL) {
- return nil;
- }
- char *lastSlash = NULL;
- char *lastDot = NULL;
- char *p = (char *)filePath;
- while (*p != '\0') {
- if (*p == '/') {
- lastSlash = p;
- } else if (*p == '.') {
- lastDot = p;
- }
- p++;
- }
- char *subStr;
- NSUInteger subLen;
- if (lastSlash) {
- if (lastDot) {
- // lastSlash -> lastDot
- subStr = lastSlash + 1;
- subLen = (NSUInteger)(lastDot - subStr);
- } else {
- // lastSlash -> endOfString
- subStr = lastSlash + 1;
- subLen = (NSUInteger)(p - subStr);
- }
- } else {
- if (lastDot) {
- // startOfString -> lastDot
- subStr = (char *)filePath;
- subLen = (NSUInteger)(lastDot - subStr);
- } else {
- // startOfString -> endOfString
- subStr = (char *)filePath;
- subLen = (NSUInteger)(p - subStr);
- }
- }
- if (copy) {
- return [[NSString alloc] initWithBytes:subStr
- length:subLen
- encoding:NSUTF8StringEncoding];
- } else {
- // We can take advantage of the fact that __FILE__ is a string literal.
- // Specifically, we don't need to waste time copying the string.
- // We can just tell NSString to point to a range within the string literal.
- return [[NSString alloc] initWithBytesNoCopy:subStr
- length:subLen
- encoding:NSUTF8StringEncoding
- freeWhenDone:NO];
- }
- }
- @end
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark -
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- @implementation DDLoggerNode
- - (instancetype)initWithLogger:(id <DDLogger>)logger loggerQueue:(dispatch_queue_t)loggerQueue level:(DDLogLevel)level {
- if ((self = [super init])) {
- _logger = logger;
- if (loggerQueue) {
- _loggerQueue = loggerQueue;
- #if !OS_OBJECT_USE_OBJC
- dispatch_retain(loggerQueue);
- #endif
- }
- _level = level;
- }
- return self;
- }
- + (DDLoggerNode *)nodeWithLogger:(id <DDLogger>)logger loggerQueue:(dispatch_queue_t)loggerQueue level:(DDLogLevel)level {
- return [[DDLoggerNode alloc] initWithLogger:logger loggerQueue:loggerQueue level:level];
- }
- - (void)dealloc {
- #if !OS_OBJECT_USE_OBJC
- if (_loggerQueue) {
- dispatch_release(_loggerQueue);
- }
- #endif
- }
- @end
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark -
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- @implementation DDLogMessage
- - (instancetype)init {
- self = [super init];
- return self;
- }
- - (instancetype)initWithMessage:(NSString *)message
- level:(DDLogLevel)level
- flag:(DDLogFlag)flag
- context:(NSInteger)context
- file:(NSString *)file
- function:(NSString *)function
- line:(NSUInteger)line
- tag:(id)tag
- options:(DDLogMessageOptions)options
- timestamp:(NSDate *)timestamp {
- if ((self = [super init])) {
- BOOL copyMessage = (options & DDLogMessageDontCopyMessage) == 0;
- _message = copyMessage ? [message copy] : message;
- _level = level;
- _flag = flag;
- _context = context;
- BOOL copyFile = (options & DDLogMessageCopyFile) != 0;
- _file = copyFile ? [file copy] : file;
- BOOL copyFunction = (options & DDLogMessageCopyFunction) != 0;
- _function = copyFunction ? [function copy] : function;
- _line = line;
- _tag = tag;
- _options = options;
- _timestamp = timestamp ?: [NSDate new];
- __uint64_t tid;
- if (pthread_threadid_np(NULL, &tid) == 0) {
- _threadID = [[NSString alloc] initWithFormat:@"%llu", tid];
- } else {
- _threadID = @"missing threadId";
- }
- _threadName = NSThread.currentThread.name;
- // Get the file name without extension
- _fileName = [_file lastPathComponent];
- NSUInteger dotLocation = [_fileName rangeOfString:@"." options:NSBackwardsSearch].location;
- if (dotLocation != NSNotFound)
- {
- _fileName = [_fileName substringToIndex:dotLocation];
- }
-
- // Try to get the current queue's label
- _queueLabel = [[NSString alloc] initWithFormat:@"%s", dispatch_queue_get_label(DISPATCH_CURRENT_QUEUE_LABEL)];
- }
- return self;
- }
- - (id)copyWithZone:(NSZone * __attribute__((unused)))zone {
- DDLogMessage *newMessage = [DDLogMessage new];
-
- newMessage->_message = _message;
- newMessage->_level = _level;
- newMessage->_flag = _flag;
- newMessage->_context = _context;
- newMessage->_file = _file;
- newMessage->_fileName = _fileName;
- newMessage->_function = _function;
- newMessage->_line = _line;
- newMessage->_tag = _tag;
- newMessage->_options = _options;
- newMessage->_timestamp = _timestamp;
- newMessage->_threadID = _threadID;
- newMessage->_threadName = _threadName;
- newMessage->_queueLabel = _queueLabel;
- return newMessage;
- }
- @end
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark -
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- @implementation DDAbstractLogger
- - (instancetype)init {
- if ((self = [super init])) {
- const char *loggerQueueName = NULL;
- if ([self respondsToSelector:@selector(loggerName)]) {
- loggerQueueName = [[self loggerName] UTF8String];
- }
- _loggerQueue = dispatch_queue_create(loggerQueueName, NULL);
- // We're going to use dispatch_queue_set_specific() to "mark" our loggerQueue.
- // Later we can use dispatch_get_specific() to determine if we're executing on our loggerQueue.
- // The documentation states:
- //
- // > Keys are only compared as pointers and are never dereferenced.
- // > Thus, you can use a pointer to a static variable for a specific subsystem or
- // > any other value that allows you to identify the value uniquely.
- // > Specifying a pointer to a string constant is not recommended.
- //
- // So we're going to use the very convenient key of "self",
- // which also works when multiple logger classes extend this class, as each will have a different "self" key.
- //
- // This is used primarily for thread-safety assertions (via the isOnInternalLoggerQueue method below).
- void *key = (__bridge void *)self;
- void *nonNullValue = (__bridge void *)self;
- dispatch_queue_set_specific(_loggerQueue, key, nonNullValue, NULL);
- }
- return self;
- }
- - (void)dealloc {
- #if !OS_OBJECT_USE_OBJC
- if (_loggerQueue) {
- dispatch_release(_loggerQueue);
- }
- #endif
- }
- - (void)logMessage:(DDLogMessage * __attribute__((unused)))logMessage {
- // Override me
- }
- - (id <DDLogFormatter>)logFormatter {
- // This method must be thread safe and intuitive.
- // Therefore if somebody executes the following code:
- //
- // [logger setLogFormatter:myFormatter];
- // formatter = [logger logFormatter];
- //
- // They would expect formatter to equal myFormatter.
- // This functionality must be ensured by the getter and setter method.
- //
- // The thread safety must not come at a cost to the performance of the logMessage method.
- // This method is likely called sporadically, while the logMessage method is called repeatedly.
- // This means, the implementation of this method:
- // - Must NOT require the logMessage method to acquire a lock.
- // - Must NOT require the logMessage method to access an atomic property (also a lock of sorts).
- //
- // Thread safety is ensured by executing access to the formatter variable on the loggerQueue.
- // This is the same queue that the logMessage method operates on.
- //
- // Note: The last time I benchmarked the performance of direct access vs atomic property access,
- // direct access was over twice as fast on the desktop and over 6 times as fast on the iPhone.
- //
- // Furthermore, consider the following code:
- //
- // DDLogVerbose(@"log msg 1");
- // DDLogVerbose(@"log msg 2");
- // [logger setFormatter:myFormatter];
- // DDLogVerbose(@"log msg 3");
- //
- // Our intuitive requirement means that the new formatter will only apply to the 3rd log message.
- // This must remain true even when using asynchronous logging.
- // We must keep in mind the various queue's that are in play here:
- //
- // loggerQueue : Our own private internal queue that the logMessage method runs on.
- // Operations are added to this queue from the global loggingQueue.
- //
- // globalLoggingQueue : The queue that all log messages go through before they arrive in our loggerQueue.
- //
- // All log statements go through the serial globalLoggingQueue before they arrive at our loggerQueue.
- // Thus this method also goes through the serial globalLoggingQueue to ensure intuitive operation.
- // IMPORTANT NOTE:
- //
- // Methods within the DDLogger implementation MUST access the formatter ivar directly.
- // This method is designed explicitly for external access.
- //
- // Using "self." syntax to go through this method will cause immediate deadlock.
- // This is the intended result. Fix it by accessing the ivar directly.
- // Great strides have been take to ensure this is safe to do. Plus it's MUCH faster.
- NSAssert(![self isOnGlobalLoggingQueue], @"Core architecture requirement failure");
- NSAssert(![self isOnInternalLoggerQueue], @"MUST access ivar directly, NOT via self.* syntax.");
- dispatch_queue_t globalLoggingQueue = [DDLog loggingQueue];
- __block id <DDLogFormatter> result;
- dispatch_sync(globalLoggingQueue, ^{
- dispatch_sync(self->_loggerQueue, ^{
- result = self->_logFormatter;
- });
- });
- return result;
- }
- - (void)setLogFormatter:(id <DDLogFormatter>)logFormatter {
- // The design of this method is documented extensively in the logFormatter message (above in code).
- NSAssert(![self isOnGlobalLoggingQueue], @"Core architecture requirement failure");
- NSAssert(![self isOnInternalLoggerQueue], @"MUST access ivar directly, NOT via self.* syntax.");
- dispatch_block_t block = ^{
- @autoreleasepool {
- if (self->_logFormatter != logFormatter) {
- if ([self->_logFormatter respondsToSelector:@selector(willRemoveFromLogger:)]) {
- [self->_logFormatter willRemoveFromLogger:self];
- }
- self->_logFormatter = logFormatter;
-
- if ([self->_logFormatter respondsToSelector:@selector(didAddToLogger:inQueue:)]) {
- [self->_logFormatter didAddToLogger:self inQueue:self->_loggerQueue];
- } else if ([self->_logFormatter respondsToSelector:@selector(didAddToLogger:)]) {
- [self->_logFormatter didAddToLogger:self];
- }
- }
- }
- };
- dispatch_queue_t globalLoggingQueue = [DDLog loggingQueue];
- dispatch_async(globalLoggingQueue, ^{
- dispatch_async(self->_loggerQueue, block);
- });
- }
- - (dispatch_queue_t)loggerQueue {
- return _loggerQueue;
- }
- - (NSString *)loggerName {
- return NSStringFromClass([self class]);
- }
- - (BOOL)isOnGlobalLoggingQueue {
- return (dispatch_get_specific(GlobalLoggingQueueIdentityKey) != NULL);
- }
- - (BOOL)isOnInternalLoggerQueue {
- void *key = (__bridge void *)self;
- return (dispatch_get_specific(key) != NULL);
- }
- @end
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- #pragma mark -
- ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
- @interface DDLoggerInformation()
- {
- // Direct accessors to be used only for performance
- @public
- id <DDLogger> _logger;
- DDLogLevel _level;
- }
- @end
- @implementation DDLoggerInformation
- - (instancetype)initWithLogger:(id <DDLogger>)logger andLevel:(DDLogLevel)level {
- if ((self = [super init])) {
- _logger = logger;
- _level = level;
- }
- return self;
- }
- + (DDLoggerInformation *)informationWithLogger:(id <DDLogger>)logger andLevel:(DDLogLevel)level {
- return [[DDLoggerInformation alloc] initWithLogger:logger andLevel:level];
- }
- @end
|