1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * Permission is hereby granted, free of charge, to any person obtaining a copy
5    * of this software and associated documentation files (the "Software"), to deal
6    * in the Software without restriction, including without limitation the rights
7    * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8    * copies of the Software, and to permit persons to whom the Software is
9    * furnished to do so, subject to the following conditions:
10   *
11   * The above copyright notice and this permission notice shall be included in
12   * all copies or substantial portions of the Software.
13   *
14   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20   * SOFTWARE.
21   */
22  
23  package com.liferay.portal.scheduler.messaging;
24  
25  import com.liferay.portal.kernel.log.Log;
26  import com.liferay.portal.kernel.log.LogFactoryUtil;
27  import com.liferay.portal.kernel.messaging.Message;
28  import com.liferay.portal.kernel.messaging.MessageBusUtil;
29  import com.liferay.portal.kernel.messaging.MessageListener;
30  import com.liferay.portal.kernel.messaging.sender.MessageSender;
31  import com.liferay.portal.kernel.scheduler.SchedulerEngine;
32  import com.liferay.portal.kernel.scheduler.TriggerType;
33  import com.liferay.portal.kernel.scheduler.messaging.SchedulerRequest;
34  
35  import java.util.List;
36  
37  /**
38   * <a href="SchedulerMessageListener.java.html"><b><i>View Source</i></b></a>
39   *
40   * @author Michael C. Han
41   * @author Bruno Farache
42   */
43  public class SchedulerMessageListener implements MessageListener {
44  
45      public SchedulerMessageListener() {
46      }
47  
48      /**
49       * @deprecated
50       */
51      public SchedulerMessageListener(
52          MessageSender messageSender, SchedulerEngine schedulerEngine) {
53  
54          _messageSender = messageSender;
55          _schedulerEngine = schedulerEngine;
56      }
57  
58      public void receive(Message message) {
59          try {
60              doReceive(message);
61          }
62          catch (Exception e) {
63              _log.error("Unable to process message " + message, e);
64          }
65      }
66  
67      public void setMessageSender(MessageSender messageSender) {
68          _messageSender = messageSender;
69      }
70  
71      public void setSchedulerEngine(SchedulerEngine schedulerEngine) {
72          _schedulerEngine = schedulerEngine;
73      }
74  
75      protected void doReceive(Message message) throws Exception {
76          SchedulerRequest schedulerRequest =
77              (SchedulerRequest)message.getPayload();
78  
79          String command = schedulerRequest.getCommand();
80  
81          if (command.equals(SchedulerRequest.COMMAND_REGISTER)) {
82              TriggerType triggerType = schedulerRequest.getTriggerType();
83  
84              if (triggerType.equals(TriggerType.CRON)) {
85                  _schedulerEngine.schedule(
86                      schedulerRequest.getJobName(),
87                      schedulerRequest.getGroupName(),
88                      schedulerRequest.getCronText(),
89                      schedulerRequest.getStartDate(),
90                      schedulerRequest.getEndDate(),
91                      schedulerRequest.getDescription(),
92                      schedulerRequest.getDestination(),
93                      schedulerRequest.getMessageBody());
94              }
95              else if (triggerType.equals(TriggerType.SIMPLE)) {
96                  _schedulerEngine.schedule(
97                      schedulerRequest.getJobName(),
98                      schedulerRequest.getGroupName(),
99                      schedulerRequest.getInterval(),
100                     schedulerRequest.getStartDate(),
101                     schedulerRequest.getEndDate(),
102                     schedulerRequest.getDescription(),
103                     schedulerRequest.getDestination(),
104                     schedulerRequest.getMessageBody());
105             }
106         }
107         else if (command.equals(SchedulerRequest.COMMAND_RETRIEVE)) {
108             doCommandRetrieve(message, schedulerRequest);
109         }
110         else if (command.equals(SchedulerRequest.COMMAND_SHUTDOWN)) {
111             _schedulerEngine.shutdown();
112         }
113         else if (command.equals(SchedulerRequest.COMMAND_STARTUP)) {
114             _schedulerEngine.start();
115         }
116         else if (command.equals(SchedulerRequest.COMMAND_UNREGISTER)) {
117             _schedulerEngine.unschedule(
118                 schedulerRequest.getJobName(), schedulerRequest.getGroupName());
119         }
120     }
121 
122     protected void doCommandRetrieve(
123             Message message, SchedulerRequest schedulerRequest)
124         throws Exception {
125 
126         List<SchedulerRequest> schedulerRequests =
127             _schedulerEngine.getScheduledJobs(schedulerRequest.getGroupName());
128 
129         Message responseMessage = MessageBusUtil.createResponseMessage(
130             message, schedulerRequests);
131 
132         _messageSender.send(
133             responseMessage.getDestinationName(), responseMessage);
134     }
135 
136     private static Log _log =
137         LogFactoryUtil.getLog(SchedulerMessageListener.class);
138 
139     private MessageSender _messageSender;
140     private SchedulerEngine _schedulerEngine;
141 
142 }