Scheduler in progress

Author: cgivre

exec/java-exec/src/main/java/org/apache/drill/exec/server/Drillbit.java

@@ -20,6 +20,7 @@
 import org.apache.curator.framework.api.ACLProvider;
 import org.apache.curator.utils.ZKPaths;
 import org.apache.drill.common.AutoCloseables;
+import org.apache.drill.exec.server.rest.ScheduleManager;
 import org.apache.drill.common.StackTrace;
 import org.apache.drill.common.concurrent.ExtendedLatch;
 import org.apache.drill.common.config.DrillConfig;
@@ -233,6 +234,16 @@ public void run() throws Exception {
     javaPropertiesToSystemOptions();
     manager.getContext().getRemoteFunctionRegistry().init(context.getConfig(), storeProvider, coord);
     webServer.start();
+
+    // Start the SQL Lab scheduled query manager.
+    // Runs on a background daemon thread, checking every 60 seconds for due queries.
+    try {
+      ScheduleManager.getOrCreate(manager, storeProvider).start();
+      logger.info("ScheduleManager started for SQL Lab scheduled queries");
+    } catch (Exception e) {
+      logger.warn("Failed to start ScheduleManager (scheduled queries will not run): {}", e.getMessage());
+    }
+
     //Discovering HTTP port (in case of port hunting)
     int httpPort = -1;
     if (webServer.isRunning()) {
@@ -317,6 +328,7 @@ public synchronized void close() {
 
     try {
       AutoCloseables.close(
+          ScheduleManager.getInstance(),
           webServer,
           engine,
           storeProvider,

exec/java-exec/src/main/java/org/apache/drill/exec/server/rest/DrillRestServer.java

@@ -113,8 +113,11 @@ public DrillRestServer(final WorkManager workManager, final ServletContext servl
     register(NotebookResources.class);
     register(ResultCacheResources.class);
     register(ProfileConfigResources.class);
+    register(UserResources.class);
+    register(WorkflowConfigResources.class);
+    register(ScheduleResources.class);
 
-    logger.info("Registered {} resource classes", 23);
+    logger.info("Registered {} resource classes", 26);
 
     property(FreemarkerMvcFeature.TEMPLATE_OBJECT_FACTORY, getFreemarkerConfiguration(servletContext));
     register(FreemarkerMvcFeature.class);