#203 New task for exporting modules

Author: rjrudin

src/main/groovy/com/marklogic/gradle/MarkLogicPlugin.groovy

@@ -24,6 +24,7 @@ import com.marklogic.gradle.task.databases.*
 import com.marklogic.gradle.task.datamovement.AddCollectionsTask
 import com.marklogic.gradle.task.datamovement.AddPermissionsTask
 import com.marklogic.gradle.task.datamovement.DeleteCollectionsTask
+import com.marklogic.gradle.task.datamovement.ExportModulesTask
 import com.marklogic.gradle.task.datamovement.RemoveCollectionsTask
 import com.marklogic.gradle.task.datamovement.RemovePermissionsTask
 import com.marklogic.gradle.task.datamovement.SetCollectionsTask
@@ -181,6 +182,9 @@ class MarkLogicPlugin implements Plugin<Project> {
 		project.task("mlReloadModules", group: modulesGroup, dependsOn: ["mlClearModulesDatabase", "mlLoadModules"], description: "Reloads modules by first clearing the modules database and then loading modules")
 		project.task("mlWatch", type: WatchTask, group: modulesGroup, description: "Run a loop that checks for new/modified modules every second and loads any that it finds. To ignore files that are already dirty and only process new changes, include -PignoreDirty=true . ")
 		project.task("mlDeleteModuleTimestampsFile", type: DeleteModuleTimestampsFileTask, group: modulesGroup, description: "Delete the properties file in the build directory that keeps track of when each module was last loaded")
+		project.task("mlExportModules", type: ExportModulesTask, group: modulesGroup, description: "Export modules matching a URI pattern of ** (can be overridden via -PuriPattern) from the database " +
+			"defined by mlModulesDatabaseName (can be overridden via -PdatabaseName) to the last path defined by mlModulePaths (can be overridden via -PexportPath). For each module that cannot be exported, " +
+			"an error will be logged; an error will be thrown instead by setting -PlogErrors to false.")
 
 		String qconsoleGroup = "ml-gradle qconsole"
 		project.task("mlImportWorkspaces", type: ImportWorkspacesTask, group: qconsoleGroup, description: "Import workspaces into qconsole")

src/main/groovy/com/marklogic/gradle/task/datamovement/ExportModulesTask.groovy

@@ -0,0 +1,59 @@
+package com.marklogic.gradle.task.datamovement
+
+import com.marklogic.client.DatabaseClient
+import com.marklogic.client.datamovement.ExportListener
+import com.marklogic.client.ext.datamovement.QueryBatcherTemplate
+import com.marklogic.client.ext.datamovement.consumer.WriteToFileConsumer
+import org.gradle.api.tasks.TaskAction
+
+class ExportModulesTask extends DataMovementTask {
+
+	@TaskAction
+	void exportModules() {
+		String exportPath = ""
+		String uriPattern = "**"
+		String databaseName = getAppConfig().getModulesDatabaseName()
+		boolean logErrors = true
+
+		if (project.hasProperty("exportPath")) {
+			exportPath = project.property("exportPath")
+		} else {
+			List<String> modulePaths = getAppConfig().getModulePaths()
+			if (modulePaths == null || modulePaths.isEmpty()) {
+				println "Cannot export modules, no module paths are defined; can use -PexportPath= to define a path to export to"
+				return
+			}
+
+			// Use the last path; if there are multiple, the ones at the beginning may be from dependencies
+			exportPath = modulePaths.get(modulePaths.size() - 1) + "/root"
+		}
+
+		if (project.hasProperty("uriPattern")) {
+			uriPattern = project.property("uriPattern")
+		}
+
+		if (project.hasProperty("databaseName")) {
+			databaseName = project.property("databaseName")
+		}
+
+		if (project.hasProperty("logErrors")) {
+			logErrors = Boolean.parseBoolean(project.property("logErrors"))
+		}
+
+		File out = new File(exportPath)
+		println "Will export modules from database '" + databaseName + "' matching pattern '" + uriPattern + "' to path: " + out.getAbsolutePath()
+
+		DatabaseClient client = getAppConfig().newAppServicesDatabaseClient(databaseName)
+		try {
+			WriteToFileConsumer consumer = new WriteToFileConsumer(out)
+			consumer.setLogErrors(logErrors)
+
+			QueryBatcherTemplate template = newQueryBatcherTemplate(client)
+			ExportListener listener = new ExportListener()
+			listener.onDocumentReady(consumer)
+			template.applyOnUriPattern(listener, "**")
+		} finally {
+			client.release()
+		}
+	}
+}