141 lines
3.5 KiB
Java
Executable File
141 lines
3.5 KiB
Java
Executable File
package org.gcube.dataanalysis.dataminer.poolmanager.ansible;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.PrintStream;
|
|
import java.util.Scanner;
|
|
|
|
import org.gcube.dataanalysis.dataminer.poolmanager.ansible.model.Inventory;
|
|
import org.gcube.dataanalysis.dataminer.poolmanager.ansible.model.Playbook;
|
|
import org.gcube.dataanalysis.dataminer.poolmanager.ansible.model.Role;
|
|
import org.gcube.dataanalysis.dataminer.poolmanager.ansiblebridge.AnsibleSerializeHelper;
|
|
import org.tmatesoft.svn.core.SVNException;
|
|
|
|
/**
|
|
* This class is responsible for the interface with ansible, retrieving log,
|
|
* etc. etc. It's not supposed to access templates and static stuff files. It
|
|
* does not know the service datamodel.
|
|
*
|
|
* @author paolo
|
|
*
|
|
*/
|
|
public class AnsibleWorker {
|
|
|
|
/**
|
|
* The name of the inventory
|
|
*/
|
|
private static String INVENTORY_NAME = "inventory.yaml";
|
|
|
|
/**
|
|
* The directory containing roles
|
|
*/
|
|
private static String ROLES_DIR = "roles";
|
|
|
|
/**
|
|
* The name of the playbook
|
|
*/
|
|
private static String PLAYBOOK_NAME = "playbook.yaml";
|
|
|
|
/**
|
|
* The root of the worker. This corresponds to a standard ansible working dir.
|
|
*/
|
|
private File workerRoot;
|
|
|
|
public AnsibleWorker(File root) {
|
|
this.workerRoot = root;
|
|
this.ensureWorkStructure();
|
|
}
|
|
|
|
// public File getWorkdir() {
|
|
// return this.workerRoot;
|
|
// }
|
|
|
|
public File getRolesDir() {
|
|
return new File(this.workerRoot, ROLES_DIR);
|
|
}
|
|
|
|
public String getWorkerId() {
|
|
return this.workerRoot.getName();
|
|
}
|
|
|
|
public void ensureWorkStructure() {
|
|
// generate root
|
|
this.workerRoot.mkdirs();
|
|
}
|
|
|
|
public void removeWorkStructure() {
|
|
// remove the working dir
|
|
this.workerRoot.delete();
|
|
}
|
|
|
|
public File getPlaybookFile() {
|
|
return new File(this.workerRoot, PLAYBOOK_NAME);
|
|
}
|
|
|
|
public File getInventoryFile() {
|
|
return new File(this.workerRoot, INVENTORY_NAME);
|
|
}
|
|
|
|
|
|
public void setInventory(Inventory inventory) throws IOException {
|
|
// serialize the string to the 'inventory' file
|
|
AnsibleSerializeHelper.serialize(inventory, this.getInventoryFile());
|
|
}
|
|
|
|
public void setPlaybook(Playbook playbook) throws IOException {
|
|
// serialize the string to the 'playbook' file
|
|
AnsibleSerializeHelper.serialize(playbook, this.getPlaybookFile());
|
|
}
|
|
|
|
public void addRole(Role r) throws IOException {
|
|
// Serialize role in the workdir
|
|
AnsibleSerializeHelper.serializeRole(r, this.getRolesDir());
|
|
}
|
|
|
|
|
|
public int execute(PrintStream ps)
|
|
throws IOException, InterruptedException, SVNException {
|
|
|
|
System.out.println(this.workerRoot);
|
|
try {
|
|
Process p = Runtime.getRuntime().exec("ansible-playbook -v -i " + this.getInventoryFile().getAbsolutePath()
|
|
+ " " + this.getPlaybookFile().getAbsolutePath());
|
|
|
|
inheritIO(p.getInputStream(), ps);
|
|
inheritIO(p.getErrorStream(), ps);
|
|
|
|
// writer.println(this.getStatus(p.waitFor()));
|
|
// writer.close();
|
|
|
|
return p.waitFor();
|
|
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
private static void inheritIO(final InputStream src, final PrintStream dest) {
|
|
new Thread(new Runnable() {
|
|
public void run() {
|
|
Scanner sc = new Scanner(src);
|
|
while (sc.hasNextLine()) {
|
|
dest.println(sc.nextLine());
|
|
}
|
|
sc.close();
|
|
}
|
|
}).start();
|
|
}
|
|
|
|
/**
|
|
* Destroy the worker:
|
|
* - remove the working dir
|
|
*/
|
|
public void destroy() {
|
|
this.removeWorkStructure();
|
|
}
|
|
|
|
}
|