|
32 | 32 |
|
33 | 33 | class WorkQ: |
34 | 34 |
|
35 | | - # Invariants: |
36 | | - |
37 | | - # - busy and work are only modified when mutex is locked |
38 | | - # - len(work) is the number of jobs ready to be taken |
39 | | - # - busy is the number of jobs being done |
40 | | - # - todo is locked iff there is no work and somebody is busy |
41 | | - |
42 | | - def __init__(self): |
43 | | - self.mutex = thread.allocate() |
44 | | - self.todo = thread.allocate() |
45 | | - self.todo.acquire() |
46 | | - self.work = [] |
47 | | - self.busy = 0 |
48 | | - |
49 | | - def addwork(self, func, args): |
50 | | - job = (func, args) |
51 | | - self.mutex.acquire() |
52 | | - self.work.append(job) |
53 | | - self.mutex.release() |
54 | | - if len(self.work) == 1: |
55 | | - self.todo.release() |
56 | | - |
57 | | - def _getwork(self): |
58 | | - self.todo.acquire() |
59 | | - self.mutex.acquire() |
60 | | - if self.busy == 0 and len(self.work) == 0: |
61 | | - self.mutex.release() |
62 | | - self.todo.release() |
63 | | - return None |
64 | | - job = self.work[0] |
65 | | - del self.work[0] |
66 | | - self.busy = self.busy + 1 |
67 | | - self.mutex.release() |
68 | | - if len(self.work) > 0: |
69 | | - self.todo.release() |
70 | | - return job |
71 | | - |
72 | | - def _donework(self): |
73 | | - self.mutex.acquire() |
74 | | - self.busy = self.busy - 1 |
75 | | - if self.busy == 0 and len(self.work) == 0: |
76 | | - self.todo.release() |
77 | | - self.mutex.release() |
78 | | - |
79 | | - def _worker(self): |
80 | | - time.sleep(0.00001) # Let other threads run |
81 | | - while 1: |
82 | | - job = self._getwork() |
83 | | - if not job: |
84 | | - break |
85 | | - func, args = job |
86 | | - apply(func, args) |
87 | | - self._donework() |
88 | | - |
89 | | - def run(self, nworkers): |
90 | | - if not self.work: |
91 | | - return # Nothing to do |
92 | | - for i in range(nworkers-1): |
93 | | - thread.start_new(self._worker, ()) |
94 | | - self._worker() |
95 | | - self.todo.acquire() |
| 35 | + # Invariants: |
| 36 | + |
| 37 | + # - busy and work are only modified when mutex is locked |
| 38 | + # - len(work) is the number of jobs ready to be taken |
| 39 | + # - busy is the number of jobs being done |
| 40 | + # - todo is locked iff there is no work and somebody is busy |
| 41 | + |
| 42 | + def __init__(self): |
| 43 | + self.mutex = thread.allocate() |
| 44 | + self.todo = thread.allocate() |
| 45 | + self.todo.acquire() |
| 46 | + self.work = [] |
| 47 | + self.busy = 0 |
| 48 | + |
| 49 | + def addwork(self, func, args): |
| 50 | + job = (func, args) |
| 51 | + self.mutex.acquire() |
| 52 | + self.work.append(job) |
| 53 | + self.mutex.release() |
| 54 | + if len(self.work) == 1: |
| 55 | + self.todo.release() |
| 56 | + |
| 57 | + def _getwork(self): |
| 58 | + self.todo.acquire() |
| 59 | + self.mutex.acquire() |
| 60 | + if self.busy == 0 and len(self.work) == 0: |
| 61 | + self.mutex.release() |
| 62 | + self.todo.release() |
| 63 | + return None |
| 64 | + job = self.work[0] |
| 65 | + del self.work[0] |
| 66 | + self.busy = self.busy + 1 |
| 67 | + self.mutex.release() |
| 68 | + if len(self.work) > 0: |
| 69 | + self.todo.release() |
| 70 | + return job |
| 71 | + |
| 72 | + def _donework(self): |
| 73 | + self.mutex.acquire() |
| 74 | + self.busy = self.busy - 1 |
| 75 | + if self.busy == 0 and len(self.work) == 0: |
| 76 | + self.todo.release() |
| 77 | + self.mutex.release() |
| 78 | + |
| 79 | + def _worker(self): |
| 80 | + time.sleep(0.00001) # Let other threads run |
| 81 | + while 1: |
| 82 | + job = self._getwork() |
| 83 | + if not job: |
| 84 | + break |
| 85 | + func, args = job |
| 86 | + apply(func, args) |
| 87 | + self._donework() |
| 88 | + |
| 89 | + def run(self, nworkers): |
| 90 | + if not self.work: |
| 91 | + return # Nothing to do |
| 92 | + for i in range(nworkers-1): |
| 93 | + thread.start_new(self._worker, ()) |
| 94 | + self._worker() |
| 95 | + self.todo.acquire() |
96 | 96 |
|
97 | 97 |
|
98 | 98 | # Main program |
99 | 99 |
|
100 | 100 | def main(): |
101 | | - sys.argv.append("/tmp") |
102 | | - nworkers = 4 |
103 | | - opts, args = getopt.getopt(sys.argv[1:], '-w:') |
104 | | - for opt, arg in opts: |
105 | | - if opt == '-w': |
106 | | - nworkers = string.atoi(arg) |
107 | | - if not args: |
108 | | - args = [os.curdir] |
| 101 | + sys.argv.append("/tmp") |
| 102 | + nworkers = 4 |
| 103 | + opts, args = getopt.getopt(sys.argv[1:], '-w:') |
| 104 | + for opt, arg in opts: |
| 105 | + if opt == '-w': |
| 106 | + nworkers = string.atoi(arg) |
| 107 | + if not args: |
| 108 | + args = [os.curdir] |
109 | 109 |
|
110 | | - wq = WorkQ() |
111 | | - for dir in args: |
112 | | - wq.addwork(find, (dir, selector, wq)) |
| 110 | + wq = WorkQ() |
| 111 | + for dir in args: |
| 112 | + wq.addwork(find, (dir, selector, wq)) |
113 | 113 |
|
114 | | - t1 = time.time() |
115 | | - wq.run(nworkers) |
116 | | - t2 = time.time() |
| 114 | + t1 = time.time() |
| 115 | + wq.run(nworkers) |
| 116 | + t2 = time.time() |
117 | 117 |
|
118 | | - sys.stderr.write('Total time ' + `t2-t1` + ' sec.\n') |
| 118 | + sys.stderr.write('Total time ' + `t2-t1` + ' sec.\n') |
119 | 119 |
|
120 | 120 |
|
121 | 121 | # The predicate -- defines what files we look for. |
122 | 122 | # Feel free to change this to suit your purpose |
123 | 123 |
|
124 | 124 | def selector(dir, name, fullname, stat): |
125 | | - # Look for group or world writable files |
126 | | - return (stat[ST_MODE] & 0022) != 0 |
| 125 | + # Look for group or world writable files |
| 126 | + return (stat[ST_MODE] & 0022) != 0 |
127 | 127 |
|
128 | 128 |
|
129 | 129 | # The find procedure -- calls wq.addwork() for subdirectories |
130 | 130 |
|
131 | 131 | def find(dir, pred, wq): |
132 | | - try: |
133 | | - names = os.listdir(dir) |
134 | | - except os.error, msg: |
135 | | - print `dir`, ':', msg |
136 | | - return |
137 | | - for name in names: |
138 | | - if name not in (os.curdir, os.pardir): |
139 | | - fullname = os.path.join(dir, name) |
140 | | - try: |
141 | | - stat = os.lstat(fullname) |
142 | | - except os.error, msg: |
143 | | - print `fullname`, ':', msg |
144 | | - continue |
145 | | - if pred(dir, name, fullname, stat): |
146 | | - print fullname |
147 | | - if S_ISDIR(stat[ST_MODE]): |
148 | | - if not os.path.ismount(fullname): |
149 | | - wq.addwork(find, (fullname, pred, wq)) |
| 132 | + try: |
| 133 | + names = os.listdir(dir) |
| 134 | + except os.error, msg: |
| 135 | + print `dir`, ':', msg |
| 136 | + return |
| 137 | + for name in names: |
| 138 | + if name not in (os.curdir, os.pardir): |
| 139 | + fullname = os.path.join(dir, name) |
| 140 | + try: |
| 141 | + stat = os.lstat(fullname) |
| 142 | + except os.error, msg: |
| 143 | + print `fullname`, ':', msg |
| 144 | + continue |
| 145 | + if pred(dir, name, fullname, stat): |
| 146 | + print fullname |
| 147 | + if S_ISDIR(stat[ST_MODE]): |
| 148 | + if not os.path.ismount(fullname): |
| 149 | + wq.addwork(find, (fullname, pred, wq)) |
150 | 150 |
|
151 | 151 |
|
152 | 152 | # Call the main program |
|
0 commit comments