blob: f835a0b8ef0951ed96724c01453a41eb5104f44b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
//
// GTMNSThread+Blocks.m
//
// Copyright 2012 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy
// of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations
// under the License.
//
#import "GTMNSThread+Blocks.h"
#import <pthread.h>
#import <dlfcn.h>
// Only available 10.6 and later.
typedef int (*pthread_setname_np_Ptr)(const char*);
#if NS_BLOCKS_AVAILABLE
@implementation NSThread (GTMBlocksAdditions)
+ (void)gtm_runBlockOnCurrentThread:(void (^)())block {
block();
}
- (void)gtm_performBlock:(void (^)())block {
if ([[NSThread currentThread] isEqual:self]) {
block();
} else {
[self gtm_performWaitingUntilDone:NO block:block];
}
}
- (void)gtm_performWaitingUntilDone:(BOOL)waitDone block:(void (^)())block {
[NSThread performSelector:@selector(gtm_runBlockOnCurrentThread:)
onThread:self
withObject:[[block copy] autorelease]
waitUntilDone:waitDone];
}
+ (void)gtm_performBlockInBackground:(void (^)())block {
[NSThread performSelectorInBackground:@selector(gtm_runBlockOnCurrentThread:)
withObject:[[block copy] autorelease]];
}
@end
#endif // NS_BLOCKS_AVAILABLE
#if GTM_IPHONE_SDK || (MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_5)
enum {
GTMSimpleThreadIsStarting = 1,
GTMSimpleThreadIsFinished
};
@implementation GTMSimpleWorkerThread
- (id)init {
if ((self = [super init])) {
runLock_ =
[[NSConditionLock alloc] initWithCondition:GTMSimpleThreadIsStarting];
}
return self;
}
- (void)dealloc {
[runLock_ release];
[super dealloc];
}
- (void)setThreadDebuggerName:(NSString *)name {
// [NSThread setName:] doesn't actually set the name in such a way that the
// debugger can see it. So we handle it here instead.
// pthread_setname_np only available 10.6 and later, look up dynamically.
pthread_setname_np_Ptr setName = dlsym(RTLD_DEFAULT, "pthread_setname_np");
if (!setName) return;
setName([name UTF8String]);
}
- (void)main {
_GTMDevAssert([runLock_ condition] == GTMSimpleThreadIsStarting,
@"Bad start condition %d", [runLock_ condition]);
[runLock_ lockWhenCondition:GTMSimpleThreadIsStarting];
[self setThreadDebuggerName:[self name]];
// Add a port to the runloop so that it stays alive. Without a port attached
// to it, a runloop will immediately return when you call run on it.
NSPort *tempPort = [NSPort port];
NSRunLoop *loop = [NSRunLoop currentRunLoop];
[loop addPort:tempPort forMode:NSDefaultRunLoopMode];
// Run the loop using CFRunLoopRun because [NSRunLoop run] will sometimes nest
// runloops making it impossible to stop.
runLoop_ = [loop getCFRunLoop];
CFRunLoopRun();
[runLock_ unlockWithCondition:GTMSimpleThreadIsFinished];
}
- (void)stop {
CFRunLoopStop(runLoop_);
if (![[NSThread currentThread] isEqual:self]) {
// If we are calling stop from a separate thread, we block until the
// simple thread actually leaves the runloop so there is no race condition
// between the current thread and the simple thread. In effect it's a
// join operation.
[runLock_ lockWhenCondition:GTMSimpleThreadIsFinished];
[runLock_ unlockWithCondition:GTMSimpleThreadIsFinished];
}
}
- (void)setName:(NSString *)name {
if ([self isExecuting]) {
[self performSelector:@selector(setThreadDebuggerName:)
onThread:self
withObject:name
waitUntilDone:YES];
}
[super setName:name];
}
@end
#endif // GTM_IPHONE_SDK || (MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_5)
|
