|
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 |
| package org.objectledge.scheduler; |
|
30 |
| |
|
31 |
| import java.text.DateFormat; |
|
32 |
| import java.text.SimpleDateFormat; |
|
33 |
| import java.util.Date; |
|
34 |
| import java.util.HashMap; |
|
35 |
| import java.util.HashSet; |
|
36 |
| import java.util.Iterator; |
|
37 |
| import java.util.Map; |
|
38 |
| import java.util.Set; |
|
39 |
| import java.util.SortedMap; |
|
40 |
| import java.util.TreeMap; |
|
41 |
| |
|
42 |
| import org.jcontainer.dna.Configuration; |
|
43 |
| import org.jcontainer.dna.Logger; |
|
44 |
| import org.objectledge.context.Context; |
|
45 |
| import org.objectledge.threads.Task; |
|
46 |
| import org.objectledge.threads.ThreadPool; |
|
47 |
| import org.picocontainer.MutablePicoContainer; |
|
48 |
| import org.picocontainer.Startable; |
|
49 |
| |
|
50 |
| |
|
51 |
| |
|
52 |
| |
|
53 |
| |
|
54 |
| |
|
55 |
| |
|
56 |
| public abstract class AbstractScheduler |
|
57 |
| implements Startable |
|
58 |
| { |
|
59 |
| |
|
60 |
| public static final String DATE_FORMAT_DEFAULT = "yyyy-MM-dd HH:mm"; |
|
61 |
| |
|
62 |
| |
|
63 |
| |
|
64 |
| |
|
65 |
| private MutablePicoContainer container; |
|
66 |
| |
|
67 |
| |
|
68 |
| protected Configuration config; |
|
69 |
| |
|
70 |
| |
|
71 |
| protected Logger logger; |
|
72 |
| |
|
73 |
| |
|
74 |
| protected ThreadPool threadPool; |
|
75 |
| |
|
76 |
| |
|
77 |
| private Map scheduleFactory = new HashMap(); |
|
78 |
| |
|
79 |
| |
|
80 |
| protected Map jobs = new HashMap(); |
|
81 |
| |
|
82 |
| |
|
83 |
| private SortedMap queue = new TreeMap(); |
|
84 |
| |
|
85 |
| |
|
86 |
| private Map runners = new HashMap(); |
|
87 |
| |
|
88 |
| |
|
89 |
| protected DateFormat format; |
|
90 |
| |
|
91 |
| |
|
92 |
| |
|
93 |
| |
|
94 |
| |
|
95 |
| |
|
96 |
| |
|
97 |
| |
|
98 |
| |
|
99 |
| |
|
100 |
920
| public AbstractScheduler(MutablePicoContainer container, Configuration config, Logger logger,
|
|
101 |
| ThreadPool threadPool, ScheduleFactory[] scheduleFactories) |
|
102 |
| { |
|
103 |
920
| this.container = container;
|
|
104 |
920
| this.config = config;
|
|
105 |
920
| this.logger = logger;
|
|
106 |
920
| this.threadPool = threadPool;
|
|
107 |
920
| for(int i = 0; i < scheduleFactories.length; i++)
|
|
108 |
| { |
|
109 |
1380
| scheduleFactory.put(scheduleFactories[i].getName(), scheduleFactories[i]);
|
|
110 |
| } |
|
111 |
920
| String formatString = config.getChild("date_format").getValue(DATE_FORMAT_DEFAULT);
|
|
112 |
920
| format = new SimpleDateFormat(formatString);
|
|
113 |
| } |
|
114 |
| |
|
115 |
| |
|
116 |
| |
|
117 |
| |
|
118 |
920
| public void start()
|
|
119 |
| { |
|
120 |
920
| loadJobs();
|
|
121 |
920
| Iterator i = jobs.values().iterator();
|
|
122 |
920
| while(i.hasNext())
|
|
123 |
| { |
|
124 |
1748
| AbstractJobDescriptor job = (AbstractJobDescriptor)i.next();
|
|
125 |
1748
| if(job.getSchedule().atStartup())
|
|
126 |
| { |
|
127 |
0
| run(job);
|
|
128 |
| } |
|
129 |
1748
| schedule(job);
|
|
130 |
| } |
|
131 |
920
| threadPool.runDaemon(new SchedulerTask());
|
|
132 |
| } |
|
133 |
| |
|
134 |
| |
|
135 |
| |
|
136 |
| |
|
137 |
0
| public void stop()
|
|
138 |
| { |
|
139 |
| |
|
140 |
| } |
|
141 |
| |
|
142 |
| |
|
143 |
| |
|
144 |
| |
|
145 |
| |
|
146 |
| |
|
147 |
| |
|
148 |
| |
|
149 |
| |
|
150 |
| |
|
151 |
| |
|
152 |
| |
|
153 |
| |
|
154 |
| |
|
155 |
| |
|
156 |
| |
|
157 |
| |
|
158 |
| |
|
159 |
| |
|
160 |
| |
|
161 |
| |
|
162 |
| public abstract AbstractJobDescriptor createJobDescriptor(String name, Schedule schedule, |
|
163 |
| String jobSpec) |
|
164 |
| throws JobModificationException; |
|
165 |
| |
|
166 |
| |
|
167 |
| |
|
168 |
| |
|
169 |
| |
|
170 |
| |
|
171 |
| |
|
172 |
| |
|
173 |
| |
|
174 |
| |
|
175 |
| public abstract void deleteJobDescriptor(AbstractJobDescriptor job) |
|
176 |
| throws JobModificationException; |
|
177 |
| |
|
178 |
| |
|
179 |
| |
|
180 |
| |
|
181 |
| |
|
182 |
| |
|
183 |
| |
|
184 |
92
| public void enable(AbstractJobDescriptor job)
|
|
185 |
| throws JobModificationException |
|
186 |
| { |
|
187 |
92
| JobModificationException e = null;
|
|
188 |
92
| synchronized(job)
|
|
189 |
| { |
|
190 |
92
| try
|
|
191 |
| { |
|
192 |
92
| job.setEnabled(true);
|
|
193 |
| } |
|
194 |
| catch(JobModificationException ee) |
|
195 |
| { |
|
196 |
0
| e = ee;
|
|
197 |
| } |
|
198 |
| } |
|
199 |
92
| schedule(job);
|
|
200 |
92
| if(e != null)
|
|
201 |
| { |
|
202 |
0
| throw e;
|
|
203 |
| } |
|
204 |
| } |
|
205 |
| |
|
206 |
| |
|
207 |
| |
|
208 |
| |
|
209 |
| |
|
210 |
| |
|
211 |
| |
|
212 |
| |
|
213 |
| |
|
214 |
| |
|
215 |
| |
|
216 |
| |
|
217 |
46
| public void disable(AbstractJobDescriptor job)
|
|
218 |
| throws JobModificationException |
|
219 |
| { |
|
220 |
46
| JobModificationException e = null;
|
|
221 |
46
| synchronized(job)
|
|
222 |
| { |
|
223 |
46
| try
|
|
224 |
| { |
|
225 |
46
| job.setEnabled(false);
|
|
226 |
| } |
|
227 |
| catch(JobModificationException ee) |
|
228 |
| { |
|
229 |
0
| e = ee;
|
|
230 |
| } |
|
231 |
| } |
|
232 |
46
| terminate(job);
|
|
233 |
46
| if(e != null)
|
|
234 |
| { |
|
235 |
0
| throw e;
|
|
236 |
| } |
|
237 |
| } |
|
238 |
| |
|
239 |
| |
|
240 |
| |
|
241 |
| |
|
242 |
| |
|
243 |
| |
|
244 |
92
| public synchronized AbstractJobDescriptor[] getJobDescriptors()
|
|
245 |
| { |
|
246 |
92
| AbstractJobDescriptor[] result = new AbstractJobDescriptor[jobs.size()];
|
|
247 |
92
| jobs.values().toArray(result);
|
|
248 |
92
| return result;
|
|
249 |
| } |
|
250 |
| |
|
251 |
| |
|
252 |
| |
|
253 |
| |
|
254 |
| |
|
255 |
| |
|
256 |
| |
|
257 |
230
| public AbstractJobDescriptor getJobDescriptor(String name)
|
|
258 |
| { |
|
259 |
230
| return (AbstractJobDescriptor)jobs.get(name);
|
|
260 |
| } |
|
261 |
| |
|
262 |
| |
|
263 |
| |
|
264 |
| |
|
265 |
| |
|
266 |
| |
|
267 |
92
| public String[] getScheduleTypes()
|
|
268 |
| { |
|
269 |
92
| String[] result = new String[scheduleFactory.size()];
|
|
270 |
92
| scheduleFactory.keySet().toArray(result);
|
|
271 |
92
| return result;
|
|
272 |
| } |
|
273 |
| |
|
274 |
| |
|
275 |
| |
|
276 |
| |
|
277 |
| |
|
278 |
| |
|
279 |
| |
|
280 |
| |
|
281 |
| |
|
282 |
2392
| public Schedule createSchedule(String type, String config)
|
|
283 |
| throws InvalidScheduleException |
|
284 |
| { |
|
285 |
2392
| ScheduleFactory factory = (ScheduleFactory)scheduleFactory.get(type);
|
|
286 |
2392
| if(factory == null)
|
|
287 |
| { |
|
288 |
92
| throw new InvalidScheduleException("Schedule factory for type '" + type
|
|
289 |
| + "' not registered"); |
|
290 |
| } |
|
291 |
2300
| Schedule schedule = factory.getInstance();
|
|
292 |
2300
| schedule.init(this, config);
|
|
293 |
2208
| return schedule;
|
|
294 |
| } |
|
295 |
| |
|
296 |
| |
|
297 |
| |
|
298 |
| |
|
299 |
| |
|
300 |
| |
|
301 |
| |
|
302 |
| |
|
303 |
| |
|
304 |
| |
|
305 |
| |
|
306 |
| |
|
307 |
| |
|
308 |
| |
|
309 |
| |
|
310 |
| |
|
311 |
| |
|
312 |
| |
|
313 |
| |
|
314 |
| public abstract boolean allowsModifications(); |
|
315 |
| |
|
316 |
| |
|
317 |
| |
|
318 |
| |
|
319 |
| |
|
320 |
| |
|
321 |
1104
| public DateFormat getDateFormat()
|
|
322 |
| { |
|
323 |
1104
| return format;
|
|
324 |
| } |
|
325 |
| |
|
326 |
| |
|
327 |
| |
|
328 |
| |
|
329 |
| |
|
330 |
| |
|
331 |
| protected abstract void loadJobs(); |
|
332 |
| |
|
333 |
| |
|
334 |
| |
|
335 |
| |
|
336 |
| |
|
337 |
| |
|
338 |
1851
| void schedule(AbstractJobDescriptor job)
|
|
339 |
| { |
|
340 |
1851
| if(job.isEnabled())
|
|
341 |
| { |
|
342 |
1391
| Date nextRun = job.getSchedule().getNextRunTime(new Date(), job.getLastRunTime());
|
|
343 |
1391
| logger.debug("scheduling " + job.getName() + " at " + nextRun + " (last run " + job.getLastRunTime() + ")");
|
|
344 |
1391
| if(nextRun != null)
|
|
345 |
| { |
|
346 |
471
| Date start = job.getTimeLimitStart();
|
|
347 |
471
| Date end = job.getTimeLimitEnd();
|
|
348 |
471
| if((start == null || nextRun.compareTo(start) > 0)
|
|
349 |
| && (end == null || nextRun.compareTo(end) < 0)) |
|
350 |
| { |
|
351 |
471
| int countLimit = job.getRunCountLimit();
|
|
352 |
471
| if(countLimit < 0 || job.getRunCount() < countLimit)
|
|
353 |
| { |
|
354 |
460
| Long target = new Long(nextRun.getTime());
|
|
355 |
460
| Set set = (Set)queue.get(target);
|
|
356 |
460
| if(set == null)
|
|
357 |
| { |
|
358 |
460
| set = new HashSet();
|
|
359 |
460
| queue.put(target, set);
|
|
360 |
| } |
|
361 |
460
| set.add(job);
|
|
362 |
460
| synchronized(queue)
|
|
363 |
| { |
|
364 |
460
| queue.notify();
|
|
365 |
| } |
|
366 |
460
| return;
|
|
367 |
| } |
|
368 |
| } |
|
369 |
| } |
|
370 |
931
| if(job.getAutoClean())
|
|
371 |
| { |
|
372 |
0
| try
|
|
373 |
| { |
|
374 |
0
| deleteJobDescriptor(job);
|
|
375 |
| } |
|
376 |
| catch(JobModificationException e) |
|
377 |
| { |
|
378 |
0
| logger.error("failed to auto-clean expired job " + job.getName(), e);
|
|
379 |
| } |
|
380 |
| } |
|
381 |
| } |
|
382 |
| } |
|
383 |
| |
|
384 |
| |
|
385 |
| |
|
386 |
| |
|
387 |
46
| private void terminate(AbstractJobDescriptor job)
|
|
388 |
| { |
|
389 |
46
| synchronized(runners)
|
|
390 |
| { |
|
391 |
46
| Set set = (Set)runners.get(job);
|
|
392 |
46
| if(set != null)
|
|
393 |
| { |
|
394 |
0
| Iterator i = set.iterator();
|
|
395 |
0
| while(i.hasNext())
|
|
396 |
| { |
|
397 |
0
| ((RunnerTask)i.next()).terminate();
|
|
398 |
| } |
|
399 |
| } |
|
400 |
| } |
|
401 |
| } |
|
402 |
| |
|
403 |
| |
|
404 |
| |
|
405 |
| |
|
406 |
| |
|
407 |
| |
|
408 |
11
| void run(AbstractJobDescriptor job)
|
|
409 |
| { |
|
410 |
11
| synchronized(job)
|
|
411 |
| { |
|
412 |
11
| if(!job.isRunning() || job.isReentrant())
|
|
413 |
| { |
|
414 |
11
| logger.debug("starting " + job.getName() + " at " + new Date());
|
|
415 |
11
| job.setRunning(true);
|
|
416 |
11
| RunnerTask rt = new RunnerTask(job);
|
|
417 |
11
| threadPool.runWorker(rt);
|
|
418 |
| } |
|
419 |
| } |
|
420 |
| } |
|
421 |
| |
|
422 |
| |
|
423 |
| |
|
424 |
| |
|
425 |
| |
|
426 |
| |
|
427 |
| |
|
428 |
| |
|
429 |
11
| Job getJobObject(AbstractJobDescriptor job)
|
|
430 |
| throws JobNotFoundException |
|
431 |
| { |
|
432 |
11
| String className = job.getJobClassName();
|
|
433 |
11
| try
|
|
434 |
| { |
|
435 |
11
| Class clazz = Class.forName(className);
|
|
436 |
0
| if(container.getComponentInstance(clazz) == null)
|
|
437 |
| { |
|
438 |
0
| container.registerComponentImplementation(clazz);
|
|
439 |
| } |
|
440 |
0
| return (Job)container.getComponentInstance(clazz);
|
|
441 |
| } |
|
442 |
| catch(ClassNotFoundException e) |
|
443 |
| { |
|
444 |
11
| throw new JobNotFoundException("Couldn't find job class '" + className + "' ", e);
|
|
445 |
| } |
|
446 |
| } |
|
447 |
| |
|
448 |
| |
|
449 |
| |
|
450 |
| |
|
451 |
| private class RunnerTask |
|
452 |
| extends Task |
|
453 |
| { |
|
454 |
| private AbstractJobDescriptor job; |
|
455 |
| |
|
456 |
| private Thread thread; |
|
457 |
| |
|
458 |
11
| public RunnerTask(AbstractJobDescriptor job)
|
|
459 |
| { |
|
460 |
11
| this.job = job;
|
|
461 |
11
| this.thread = Thread.currentThread();
|
|
462 |
| } |
|
463 |
| |
|
464 |
33
| public String getName()
|
|
465 |
| { |
|
466 |
33
| return "scheduler: " + job.getName();
|
|
467 |
| } |
|
468 |
| |
|
469 |
11
| public void process(Context context)
|
|
470 |
| { |
|
471 |
11
| synchronized(runners)
|
|
472 |
| { |
|
473 |
11
| Set set = (Set)runners.get(job);
|
|
474 |
11
| if(set == null)
|
|
475 |
| { |
|
476 |
11
| set = new HashSet();
|
|
477 |
11
| runners.put(job, set);
|
|
478 |
| } |
|
479 |
11
| set.add(this);
|
|
480 |
| } |
|
481 |
| |
|
482 |
11
| synchronized(job)
|
|
483 |
| { |
|
484 |
11
| try
|
|
485 |
| { |
|
486 |
11
| job.setLastRunTime(new Date());
|
|
487 |
11
| job.setRunCount(job.getRunCount() + 1);
|
|
488 |
| } |
|
489 |
| catch(JobModificationException e) |
|