This commit is contained in:
Nunzio Andrea Galante 2017-07-13 16:13:18 +00:00
parent 2d037f21a3
commit 1255938fa2
8 changed files with 115 additions and 24 deletions

View File

@ -8,6 +8,7 @@ import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
@ -118,7 +119,6 @@ public class AnsibleWorker {
}
public int execute(PrintStream ps)
throws IOException, InterruptedException, SVNException {
@ -130,6 +130,8 @@ public class AnsibleWorker {
inheritIO(p.getInputStream(), ps);
inheritIO(p.getErrorStream(), ps);
// writer.println(this.getStatus(p.waitFor()));
// writer.close();
return p.waitFor();

View File

@ -71,6 +71,20 @@ public class RestPoolManager implements PoolManager {
}
@GET
@Path("/monitor")
@Produces("text/plain")
public String getMonitorById(@QueryParam("logUrl") String logUrl) throws IOException {
// TODO Auto-generated method stub
LOGGER.debug("Returning Log =" + logUrl);
return service.getMonitorById(logUrl);
}
@Override
public Algorithm extractAlgorithm(String url) throws IOException {
// TODO Auto-generated method stub

View File

@ -4,6 +4,7 @@ import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.PrintStream;
import java.io.PrintWriter;
import java.util.LinkedList;
import java.util.List;
import java.util.UUID;
@ -16,6 +17,8 @@ import org.gcube.dataanalysis.dataminer.poolmanager.datamodel.Dependency;
import org.gcube.dataanalysis.dataminer.poolmanager.util.SVNUpdater;
import org.tmatesoft.svn.core.SVNException;
import Jampack.Print;
public abstract class DMPMJob {
@ -69,9 +72,14 @@ public abstract class DMPMJob {
protected int executeAnsibleWorker(AnsibleWorker worker) throws IOException, InterruptedException, SVNException{
File path = new File(worker.getWorkdir() + File.separator + "jobs");
path.mkdirs();
File n = new File(this.jobLogs + File.separator + this.id);
FileOutputStream fos = new FileOutputStream(n, true);
PrintStream ps = new PrintStream(fos);
PrintStream ps = new PrintStream(fos);
// File m = new File(this.jobLogs + File.separator + this.id + "_exitStatus");
// PrintWriter fos2 = new PrintWriter(m, "UTF-8");
return worker.execute(ps);
}

View File

@ -56,4 +56,18 @@ public class DataminerPoolManager {
return new Scanner(path).useDelimiter("\\Z").next();
}
public String getMonitorById(String id) throws FileNotFoundException{
//TODO: load dir from configuration file
File path = new File(System.getProperty("user.home") + File.separator + "dataminer-pool-manager/jobs/"
+ id + "_exitStatus");
return new Scanner(path).useDelimiter("\\Z").next();
}
}

View File

@ -2,6 +2,10 @@ package org.gcube.dataanalysis.dataminer.poolmanager.service;
import static org.gcube.common.authorization.client.Constants.authorizationService;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.util.Collection;
import org.gcube.common.authorization.library.AuthorizationEntry;
@ -19,15 +23,20 @@ public class StagingJob extends DMPMJob {
private Algorithm algorithm;
private Cluster stagingCluster;
//private Cluster rProtoCluster;
// private Cluster rProtoCluster;
private String rProtoVREName;
public StagingJob(SVNUpdater svnUpdater, Algorithm algorithm, Cluster stagingCluster, /*Cluster rProtoCluster,*/
public StagingJob(SVNUpdater svnUpdater, Algorithm algorithm,
Cluster stagingCluster, /* Cluster rProtoCluster, */
String rProtoVREName) {
super(svnUpdater);
this.jobLogs = new File(
System.getProperty("user.home") + File.separator + "dataminer-pool-manager" + File.separator + "jobs");
this.jobLogs.mkdirs();
this.algorithm = algorithm;
this.stagingCluster = stagingCluster;
//this.rProtoCluster = rProtoCluster;
// this.rProtoCluster = rProtoCluster;
this.rProtoVREName = rProtoVREName;
}
@ -41,7 +50,8 @@ public class StagingJob extends DMPMJob {
try {
Collection<String> undefinedDependencies = this.svnUpdater.getUndefinedDependencies(
this.svnUpdater.getRProtoDependencyFile(this.algorithm.getLanguage()), this.algorithm.getDependencies());
this.svnUpdater.getRProtoDependencyFile(this.algorithm.getLanguage()),
this.algorithm.getDependencies());
if (!undefinedDependencies.isEmpty()) {
@ -56,27 +66,67 @@ public class StagingJob extends DMPMJob {
int ret = this.executeAnsibleWorker(createWorker(this.algorithm, this.stagingCluster, false, "root"));
Integer s = null;
s = Integer.valueOf(ret);
if (s == null) {
this.getStatus(0);
}
if (ret != 0) {
this.getStatus(2);
sm.sendNotification(nh.getFailedSubject(), nh.getFailedBody("Installation failed. Return code=" + ret));
}
if (ret == 0) {
this.getStatus(0);
if (b.checkMethod(a.getStagingHost(), SecurityTokenProvider.instance.get())
&& (b.algoExists(this.algorithm))) {
sm.sendNotification(nh.getSuccessSubject(), nh.getSuccessBody());
// this.svnUpdater.updateRPRotoDeps(this.algorithm);
this.svnUpdater.updateSVNRProtoAlgorithmList(this.algorithm, this.rProtoVREName,
this.algorithm.getFullname(), "Proto");
// int ret2 = this.executeAnsibleWorker(
// createWorker(this.algorithm, this.rProtoCluster,
// false, "gcube"));
}
} else
sm.sendNotification(nh.getFailedSubject(), nh.getFailedBody("Installation failed. Return code=" + ret));
this.getStatus(9);
sm.sendNotification(nh.getSuccessSubject(), nh.getSuccessBody());
} else
this.getStatus(2);
sm.sendNotification(nh.getFailedSubject(),
nh.getFailedBody("DataMiner Interface not working or files " + this.algorithm.getName()
+ ".jar and " + this.algorithm.getName()
+ "_interface.jar not available at the expected path"));
}
} catch (Exception e) {
e.printStackTrace();
}
}
public String getStatus(int exitstatus) throws FileNotFoundException, UnsupportedEncodingException {
File m = new File(this.jobLogs + File.separator + this.id + "_exitStatus");
PrintWriter writer = new PrintWriter(m, "UTF-8");
String response = "";
if (exitstatus == 0) {
response = "IN PROGRESS";
writer.println(response);
writer.close();
}
if (exitstatus == 9) {
response = "COMPLETED";
writer.println(response);
writer.close();
}
if (exitstatus == 2) {
response = "FAILED";
writer.println(response);
writer.close();
}
return response;
}
}

View File

@ -71,8 +71,8 @@ public class CheckMethod {
public boolean algoExists(Algorithm a){
File file = new File("/home/gcube/wps_algorithms/algorithms/"+a.getName()+".jar");
File file2 = new File(" /home/gcube/wps_algorithms/algorithms/"+a.getName()+"interface_.jar");
File file = new File(System.getProperty("user.home")+File.separator+"wps_algorithms/algorithms/"+a.getName()+".jar");
File file2 = new File(System.getProperty("user.home")+File.separator+"wps_algorithms/algorithms/"+a.getName()+"interface_.jar");
if (file.exists()&&(file2.exists())){
return true;

View File

@ -107,7 +107,7 @@ public class SendMail extends StandardLocalInfraAlgorithm {
//String serviceAddress = ep.getResourceEntyName();//"https://socialnetworking1.d4science.org/social-networking-library-ws/rest/";
serviceAddress = serviceAddress.endsWith("/") ? serviceAddress : serviceAddress + "/";
serviceAddress+= "2/users/get-usernames-by-global-role?";
serviceAddress+= "role-name=Administrator" + "&gcube-token=" + SecurityTokenProvider.instance.get();
serviceAddress+= "role-name=DataMiner-Manager" + "&gcube-token=" + SecurityTokenProvider.instance.get();
System.out.println("Url is " + serviceAddress);

View File

@ -19,10 +19,13 @@ public class SVNTests {
SVNUpdater svnUpdater = new SVNUpdater(new ServiceConfiguration("/home/ngalante/workspace/dataminer-pool-manager/src/main/resources/service.properties"));
Algorithm algo = AlgorithmBuilder.create("http://data.d4science.org/R0ExYjFPVFBrOUlNdEhrTUQyQlZjbUJuQ20rbGFrZ0pHbWJQNStIS0N6Yz0");
Algorithm algo = AlgorithmBuilder.create("http://data.d4science.org/cnFLNHYxR1ZDa1VNdEhrTUQyQlZjaWRBVVZlUHloUitHbWJQNStIS0N6Yz0");
//algo.setClazz(algo.getClazz() + "TEST");
System.out.println(algo.getAlgorithmType());
System.out.println(algo.getCategory());
System.out.println(algo.getSkipJava());
//svnUpdater.updateSVNRProtoAlgorithmList(algo, "/d4science.research-infrastructures.eu/gCubeApps/RPrototypingLab", "Dataminer Pool Manager", "Proto");
svnUpdater.readRPRotoDeps(algo);
//svnUpdater.readRPRotoDeps(algo);
}
}