#212 Can now update the task server

Author: rjrudin

src/main/java/com/marklogic/appdeployer/ConfigDir.java

@@ -156,6 +156,10 @@ public File getTasksDir() {
 		return new File(baseDir, "tasks");
 	}
 
+	public File getTaskServersDir() {
+		return new File(baseDir, "task-servers");
+	}
+
 	public void setDatabasesPath(String databasesPath) {
 		this.databasesPath = databasesPath;
 	}

src/main/java/com/marklogic/appdeployer/command/CommandMapBuilder.java

@@ -26,6 +26,7 @@
 import com.marklogic.appdeployer.command.schemas.LoadSchemasCommand;
 import com.marklogic.appdeployer.command.security.*;
 import com.marklogic.appdeployer.command.tasks.DeployScheduledTasksCommand;
+import com.marklogic.appdeployer.command.taskservers.UpdateTaskServerCommand;
 import com.marklogic.appdeployer.command.temporal.DeployTemporalAxesCommand;
 import com.marklogic.appdeployer.command.temporal.DeployTemporalCollectionsCommand;
 import com.marklogic.appdeployer.command.temporal.DeployTemporalCollectionsLSQTCommand;
@@ -140,6 +141,7 @@ public Map<String, List<Command>> buildCommandMap() {
 		// Tasks
 		List<Command> taskCommands = new ArrayList<Command>();
 		taskCommands.add(new DeployScheduledTasksCommand());
+		taskCommands.add(new UpdateTaskServerCommand());
 		map.put("mlTaskCommands", taskCommands);
 
 		// Temporal

src/main/java/com/marklogic/appdeployer/command/SortOrderConstants.java

@@ -46,6 +46,7 @@ public abstract class SortOrderConstants {
 	public static Integer DEPLOY_TEMPORAL_COLLECTIONS_LSQT = 752;
 
     public static Integer DEPLOY_SCHEDULED_TASKS = 800;
+    public static Integer UPDATE_TASK_SERVER = 810;
 
     public static Integer DEPLOY_DEFAULT_PIPELINES = 900;
     public static Integer DEPLOY_PIPELINES = 905;

src/main/java/com/marklogic/appdeployer/command/taskservers/UpdateTaskServerCommand.java

@@ -0,0 +1,46 @@
+package com.marklogic.appdeployer.command.taskservers;
+
+import com.marklogic.appdeployer.command.AbstractCommand;
+import com.marklogic.appdeployer.command.CommandContext;
+import com.marklogic.appdeployer.command.SortOrderConstants;
+import com.marklogic.mgmt.resource.taskservers.TaskServerManager;
+
+import java.io.File;
+
+/**
+ * As of ML 9.0-3, the Manage API only supports a PUT call to update the single task server within a cluster.
+ * <p>
+ * In addition, the single task server is named "TaskServer" and there's no apparent way to change that. So this
+ * command assumes that name as well.
+ */
+public class UpdateTaskServerCommand extends AbstractCommand {
+
+	private String taskServerName = "TaskServer";
+
+	public UpdateTaskServerCommand() {
+		setExecuteSortOrder(SortOrderConstants.UPDATE_TASK_SERVER);
+	}
+
+	@Override
+	public void execute(CommandContext context) {
+		File dir = context.getAppConfig().getConfigDir().getTaskServersDir();
+		if (dir.exists()) {
+			if (logger.isInfoEnabled()) {
+				logger.info("Processing files in directory: " + dir.getAbsolutePath());
+			}
+
+			TaskServerManager mgr = new TaskServerManager(context.getManageClient());
+			for (File f : listFilesInDirectory(dir)) {
+				if (logger.isInfoEnabled()) {
+					logger.info("Processing file: " + f.getAbsolutePath());
+				}
+				String payload = copyFileToString(f, context);
+				mgr.updateTaskServer(taskServerName, payload);
+			}
+		}
+	}
+
+	public void setTaskServerName(String taskServerName) {
+		this.taskServerName = taskServerName;
+	}
+}

src/main/java/com/marklogic/mgmt/resource/taskservers/TaskServerManager.java

@@ -0,0 +1,31 @@
+package com.marklogic.mgmt.resource.taskservers;
+
+import com.marklogic.mgmt.AbstractManager;
+import com.marklogic.mgmt.ManageClient;
+import com.marklogic.rest.util.Fragment;
+
+public class TaskServerManager extends AbstractManager {
+
+	private ManageClient manageClient;
+
+	public TaskServerManager(ManageClient manageClient) {
+		this.manageClient = manageClient;
+	}
+
+	public void updateTaskServer(String taskServerName, String payload) {
+		String path = format("/manage/v2/task-servers/%s/properties", taskServerName);
+		if (payloadParser.isJsonPayload(payload)) {
+			manageClient.putJson(path, payload);
+		} else {
+			manageClient.putXml(path, payload);
+		}
+	}
+
+	public Fragment getPropertiesAsXml() {
+		return getPropertiesAsXml("TaskServer");
+	}
+
+	public Fragment getPropertiesAsXml(String taskServerName) {
+		return manageClient.getXml(format("/manage/v2/task-servers/%s/properties", taskServerName));
+	}
+}

src/test/java/com/marklogic/appdeployer/command/taskservers/UpdateTaskServerTest.java

@@ -0,0 +1,36 @@
+package com.marklogic.appdeployer.command.taskservers;
+
+import com.marklogic.appdeployer.AbstractAppDeployerTest;
+import com.marklogic.mgmt.resource.taskservers.TaskServerManager;
+import com.marklogic.rest.util.Fragment;
+import org.junit.Test;
+
+public class UpdateTaskServerTest extends AbstractAppDeployerTest {
+
+	@Test
+	public void test() {
+		TaskServerManager mgr = new TaskServerManager(manageClient);
+
+		initializeAppDeployer(new UpdateTaskServerCommand());
+		deploySampleApp();
+
+		try {
+			Fragment xml = mgr.getPropertiesAsXml();
+			assertEquals("false", xml.getElementValue("/m:task-server-properties/m:log-errors"));
+			assertEquals("false", xml.getElementValue("/m:task-server-properties/m:debug-allow"));
+			assertEquals("false", xml.getElementValue("/m:task-server-properties/m:profile-allow"));
+		} finally {
+			String payload = "{\n" +
+				"\t\"log-errors\": true,\n" +
+				"\t\"debug-allow\": true,\n" +
+				"\t\"profile-allow\": true\n" +
+				"}";
+
+			mgr.updateTaskServer("TaskServer", payload);
+			Fragment xml = mgr.getPropertiesAsXml();
+			assertEquals("true", xml.getElementValue("/m:task-server-properties/m:log-errors"));
+			assertEquals("true", xml.getElementValue("/m:task-server-properties/m:debug-allow"));
+			assertEquals("true", xml.getElementValue("/m:task-server-properties/m:profile-allow"));
+		}
+	}
+}

src/test/resources/sample-app/src/main/ml-config/task-servers/task-server.json

@@ -0,0 +1,5 @@
+{
+	"log-errors": false,
+	"debug-allow": false,
+	"profile-allow": false
+}