forked from jkreps/azkaban
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
AZK-123: Add jmx hooks for Schedule-Unschedule commands
- Loading branch information
Ibrahim Ulukaya
committed
Sep 21, 2011
1 parent
35dea11
commit 7d0f432
Showing
8 changed files
with
853 additions
and
224 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,36 @@ | ||
| /* | ||
| * Copyright 2011 Adconion, 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. | ||
| */ | ||
| package azkaban.app.jmx; | ||
|
|
||
| import java.lang.annotation.Documented; | ||
| import java.lang.annotation.ElementType; | ||
| import java.lang.annotation.Retention; | ||
| import java.lang.annotation.RetentionPolicy; | ||
| import java.lang.annotation.Target; | ||
|
|
||
| import javax.management.DescriptorKey; | ||
|
|
||
| /** | ||
| * DisplayName - This annotation allows to supply | ||
| * a display name for a method in the MBean interface. | ||
| */ | ||
| @Documented | ||
| @Target(ElementType.METHOD) | ||
| @Retention(RetentionPolicy.RUNTIME) | ||
| public @interface DisplayName { | ||
| @DescriptorKey("displayName") | ||
| String value(); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,141 @@ | ||
| /* | ||
| * Copyright 2011 Adconion, 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. | ||
| */ | ||
| package azkaban.app.jmx; | ||
|
|
||
| import org.apache.log4j.Logger; | ||
| import org.joda.time.DateTime; | ||
| import org.joda.time.Days; | ||
| import org.joda.time.Hours; | ||
| import org.joda.time.IllegalFieldValueException; | ||
| import org.joda.time.LocalDateTime; | ||
| import org.joda.time.Minutes; | ||
| import org.joda.time.ReadablePeriod; | ||
| import org.joda.time.Seconds; | ||
| import org.joda.time.format.DateTimeFormat; | ||
|
|
||
| import azkaban.app.JobDescriptor; | ||
| import azkaban.app.JobManager; | ||
| import azkaban.scheduler.ScheduleManager; | ||
|
|
||
| /** | ||
| * @author ibrahimulukaya | ||
| * Implements the JobSchedulerMBean | ||
| */ | ||
| public class JobScheduler implements JobSchedulerMBean { | ||
| private static Logger logger = Logger.getLogger(JobScheduler.class); | ||
| private ScheduleManager scheduler; | ||
| private JobManager jobManager; | ||
|
|
||
| public JobScheduler(ScheduleManager scheduler, JobManager jobManager) { | ||
| this.scheduler = scheduler; | ||
| this.jobManager = jobManager; | ||
| } | ||
|
|
||
| public String scheduleWorkflow(String jobName, boolean ignoreDeps, | ||
| int hour, int minutes, int seconds, String scheduledDate, | ||
| boolean isRecurring, int period, String periodUnits) { | ||
| String errorMsg = null; | ||
| if (jobName == null || jobName.trim().length() == 0) { | ||
| errorMsg = "You must select at least one job to run."; | ||
| logger.error(errorMsg); | ||
| return errorMsg; | ||
| } | ||
| JobDescriptor descriptor = jobManager.getJobDescriptor(jobName); | ||
| if (descriptor == null) { | ||
| errorMsg = "Job: '" + jobName + "' doesn't exist."; | ||
| logger.error(errorMsg); | ||
| return errorMsg; | ||
| } | ||
|
|
||
| DateTime day = null; | ||
| DateTime time = null; | ||
| try { | ||
| if (scheduledDate == null || scheduledDate.trim().length() == 0) { | ||
| day = new LocalDateTime().toDateTime(); | ||
| time = day.withHourOfDay(hour).withMinuteOfHour(minutes) | ||
| .withSecondOfMinute(seconds); | ||
| if (day.isAfter(time)) { | ||
| time = time.plusDays(1); | ||
| } | ||
| } else { | ||
| try { | ||
| day = DateTimeFormat.forPattern("MM-dd-yyyy") | ||
| .parseDateTime(scheduledDate); | ||
| } catch (IllegalArgumentException e) { | ||
| logger.error(e); | ||
| return "Invalid date: '" + scheduledDate | ||
| + "', \"MM-dd-yyyy\" format is expected."; | ||
| } | ||
| time = day.withHourOfDay(hour).withMinuteOfHour(minutes) | ||
| .withSecondOfMinute(seconds); | ||
| } | ||
| } catch (IllegalFieldValueException e) { | ||
| logger.error(e); | ||
| return "Invalid schedule time (see logs): " + e.getMessage(); | ||
|
|
||
| } | ||
| ReadablePeriod thePeriod = null; | ||
| if (isRecurring) { | ||
| if ("d".equals(periodUnits)) { | ||
| thePeriod = Days.days(period); | ||
| } else if ("h".equals(periodUnits)) { | ||
| thePeriod = Hours.hours(period); | ||
| } else if ("m".equals(periodUnits)) { | ||
| thePeriod = Minutes.minutes(period); | ||
| } else if ("s".equals(periodUnits)) { | ||
| thePeriod = Seconds.seconds(period); | ||
| } else { | ||
| errorMsg = "Unknown period unit: " + periodUnits; | ||
| logger.error(errorMsg); | ||
| return errorMsg; | ||
| } | ||
| } | ||
| try { | ||
| if (thePeriod == null) { | ||
| scheduler.schedule(jobName, time, ignoreDeps); | ||
| } else { | ||
| scheduler.schedule(jobName, time, thePeriod, ignoreDeps); | ||
| } | ||
| return "Schedule Successful!"; | ||
| } catch (Exception e) { | ||
| logger.error(e); | ||
| return "Schedule Failed (see logs): " + e.getMessage(); | ||
| } | ||
| } | ||
|
|
||
| public String removeScheduledWorkflow(String jobName) { | ||
| String errorMsg = null; | ||
| if (jobName == null || jobName.trim().length() == 0) { | ||
| errorMsg = "You must select at least one job to remove."; | ||
| logger.error(errorMsg); | ||
| return errorMsg; | ||
| } | ||
| if (scheduler.getSchedule(jobName) == null) { | ||
| errorMsg = "Job: '"+ jobName + "' doesn't exist in schedule."; | ||
| logger.error(errorMsg); | ||
| return errorMsg; | ||
| } | ||
| try { | ||
| scheduler.removeScheduledJob(jobName); | ||
| return "Job: '" + jobName + "' is successfully removed from " + | ||
| "schedule."; | ||
| } catch (Exception e) { | ||
| logger.error(e); | ||
| return "Removing From Schedule Failed (see logs): " + | ||
| e.getMessage(); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.