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
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2005, JBoss Inc., and individual contributors as indicated
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jbpm.pvm.internal.jobexecutor;
 
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.List;
 
import org.jbpm.api.cmd.Command;
import org.jbpm.api.cmd.Environment;
import org.jbpm.internal.log.Log;
import org.jbpm.pvm.internal.job.JobImpl;
import org.jbpm.pvm.internal.session.DbSession;
 
 
/**
 * @author Tom Baeyens
 */
public class AcquireJobsCmd implements Command<Collection<Long>> {
 
  private static final long serialVersionUID = 1L;
 
  private static final Log log = Log.getLog(AcquireJobsCmd.class.getName());
  private static final DateFormat timeFormat = new SimpleDateFormat("HH:mm:ss,SSS");
  
  JobExecutor jobExecutor;
 
  public AcquireJobsCmd(JobExecutor jobExecutor) {
    this.jobExecutor = jobExecutor;
  }
 
  public Collection<Long> execute(Environment environment) throws Exception {
    Collection<Long> acquiredJobDbids = new ArrayList<Long>();
 
    try {
      Collection<JobImpl<?>> acquiredJobs = new ArrayList<JobImpl<?>>();
      
      DbSession dbSession = environment.get(DbSession.class);
      if (log.isTraceEnabled()) log.trace("start querying first acquirable job...");
 
      JobImpl<?> job = dbSession.findFirstAcquirableJob();
 
      if (job!=null) {
        if (job.isExclusive()) {
          if (log.isTraceEnabled()) log.trace("exclusive acquirable job found ("+job+"). querying for other exclusive jobs to lock them all in one tx...");
          List<JobImpl<?>> otherExclusiveJobs = dbSession.findExclusiveJobs(job.getProcessInstance());
          acquiredJobs.addAll(otherExclusiveJobs);
        } else {
          acquiredJobs.add(job);
        }
 
        for (JobImpl<?> acquiredJob: acquiredJobs) {
          long lockExpirationTime = System.currentTimeMillis()+jobExecutor.getLockMillis();
          if (log.isTraceEnabled()) log.trace("trying to obtain a lock for '"+acquiredJob+"' with exp "+timeFormat.format(new Date(lockExpirationTime)));
          acquiredJob.acquire(jobExecutor.getName(), new Date(lockExpirationTime));
          acquiredJobDbids.add(acquiredJob.getDbid());
        }
 
      } else {
        if (log.isTraceEnabled()) log.trace("no acquirable jobs in job table");
      }
      
      if (log.isTraceEnabled()) log.trace("locking jobs "+acquiredJobDbids);
      
    } catch (Exception e) {
      // if jboss is still booting exceptions that are due to improper 
      // dependency configuration can be swallowed
      // https://jira.jboss.org/jira/browse/JBPM-2501
      if (e.getMessage().indexOf("JbpmDS not bound")==-1) {
        throw e;
      }
    }
    
    return acquiredJobDbids;
  }
}