123 lines
3.9 KiB
Java
123 lines
3.9 KiB
Java
package org.gcube.dataanalysis.dataminer.poolmanager.ansible;
|
|
|
|
import java.io.BufferedReader;
|
|
import java.io.File;
|
|
***REMOVED***
|
|
import java.io.InputStreamReader;
|
|
import java.io.OutputStream;
|
|
|
|
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;
|
|
|
|
/**
|
|
* 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 ***REMOVED***
|
|
|
|
/**
|
|
* 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) ***REMOVED***
|
|
this.workerRoot = root;
|
|
this.ensureWorkStructure();
|
|
***REMOVED***
|
|
|
|
public File getWorkdir() ***REMOVED***
|
|
return this.workerRoot;
|
|
***REMOVED***
|
|
|
|
public File getRolesDir() ***REMOVED***
|
|
return new File(this.getWorkdir(), ROLES_DIR);
|
|
***REMOVED***
|
|
|
|
public String getWorkerId() ***REMOVED***
|
|
return this.workerRoot.getName();
|
|
***REMOVED***
|
|
|
|
public void ensureWorkStructure() ***REMOVED***
|
|
***REMOVED*** generate root
|
|
this.getWorkdir().mkdirs();
|
|
***REMOVED***
|
|
|
|
public void removeWorkStructure() ***REMOVED***
|
|
***REMOVED*** remove the working dir
|
|
***REMOVED***this.getWorkdir().delete();
|
|
***REMOVED***
|
|
|
|
public File getPlaybookFile() ***REMOVED***
|
|
return new File(this.getWorkdir(), PLAYBOOK_NAME);
|
|
***REMOVED***
|
|
|
|
public File getInventoryFile() ***REMOVED***
|
|
return new File(this.getWorkdir(), INVENTORY_NAME);
|
|
***REMOVED***
|
|
|
|
|
|
public void setInventory(Inventory inventory) throws IOException ***REMOVED***
|
|
***REMOVED*** serialize the string to the 'inventory' file
|
|
AnsibleSerializeHelper.serialize(inventory, this.getInventoryFile());
|
|
***REMOVED***
|
|
|
|
public void setPlaybook(Playbook playbook) throws IOException ***REMOVED***
|
|
***REMOVED*** serialize the string to the 'playbook' file
|
|
AnsibleSerializeHelper.serialize(playbook, this.getPlaybookFile());
|
|
***REMOVED***
|
|
|
|
public void addRole(Role r) throws IOException ***REMOVED***
|
|
***REMOVED*** Serialize role in the workdir
|
|
AnsibleSerializeHelper.serializeRole(r, this.getRolesDir());
|
|
***REMOVED***
|
|
|
|
|
|
|
|
public void apply() throws IOException ***REMOVED***
|
|
***REMOVED*** TODO execute the playbook and return output
|
|
try ***REMOVED***
|
|
System.out.println("ansible-playbook -v -i " + this.getInventoryFile().getName() + " " + this.getPlaybookFile().getName());
|
|
Process p = Runtime.getRuntime().exec("ansible-playbook -v -i " + this.getInventoryFile().getName() + " " + this.getPlaybookFile().getName());
|
|
BufferedReader in = new BufferedReader(
|
|
new InputStreamReader(p.getInputStream()));
|
|
String line = null;
|
|
while ((line = in.readLine()) != null) ***REMOVED***
|
|
System.out.println(line);
|
|
***REMOVED***
|
|
***REMOVED*** catch (IOException e) ***REMOVED***
|
|
e.printStackTrace();
|
|
***REMOVED***
|
|
***REMOVED***System.out.println("TODO: execute: ansible-playbook -v -i " + this.getInventoryFile().getName() + " " + this.getPlaybookFile().getName());
|
|
***REMOVED***
|
|
|
|
/**
|
|
* Destroy the worker:
|
|
* - remove the working dir
|
|
*/
|
|
public void destroy() ***REMOVED***
|
|
this.removeWorkStructure();
|
|
***REMOVED***
|
|
|
|
***REMOVED***
|