| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package controllers; |
| |
| import java.io.BufferedWriter; |
| import java.io.File; |
| import java.io.FileNotFoundException; |
| import java.io.FileWriter; |
| import java.io.IOException; |
| import java.io.PrintWriter; |
| import java.text.SimpleDateFormat; |
| import java.util.ArrayList; |
| import java.util.Date; |
| import java.util.HashSet; |
| import java.util.Iterator; |
| import java.util.List; |
| import java.util.Set; |
| |
| import javax.inject.Inject; |
| import javax.inject.Named; |
| import javax.inject.Singleton; |
| import javax.persistence.PersistenceException; |
| |
| import com.fasterxml.jackson.databind.JsonNode; |
| import com.google.gson.Gson; |
| |
| import models.ClimateService; |
| import models.ClimateServiceRepository; |
| import models.DatasetEntryRepository; |
| import models.DatasetLogRepository; |
| import models.DatasetRepository; |
| import models.Parameter; |
| import models.ParameterRepository; |
| import models.ServiceConfiguration; |
| import models.ServiceConfigurationItem; |
| import models.ServiceConfigurationItemRepository; |
| import models.ServiceConfigurationRepository; |
| import models.ServiceEntryRepository; |
| import models.ServiceExecutionLog; |
| import models.ServiceExecutionLogRepository; |
| import models.User; |
| import models.UserRepository; |
| import models.Workflow; |
| import models.WorkflowRepository; |
| import play.mvc.*; |
| import workflow.VisTrailJson; |
| |
| @Named |
| @Singleton |
| public class WorkflowController extends Controller { |
| public static final String WILDCARD = "%"; |
| |
| private final ClimateServiceRepository climateServiceRepository; |
| private final UserRepository userRepository; |
| private final WorkflowRepository workflowRepository; |
| private final ParameterRepository parameterRepository; |
| private final ServiceExecutionLogRepository serviceExecutionLogRepository; |
| private final ServiceConfigurationItemRepository serviceConfigurationItemRepository; |
| private final DatasetLogRepository datasetLogRepository; |
| private final ServiceConfigurationRepository serviceConfigurationRepository; |
| private final ServiceEntryRepository serviceEntryRepository; |
| private final DatasetEntryRepository datasetEntryRepository; |
| private final DatasetRepository datasetRepository; |
| |
| @Inject |
| public WorkflowController( |
| ClimateServiceRepository climateServiceRepository, |
| UserRepository userRepository, |
| WorkflowRepository workflowRepository, |
| ParameterRepository parameterRepository, |
| ServiceExecutionLogRepository serviceExecutionLogRepository, |
| ServiceConfigurationItemRepository serviceConfigurationItemRepository, |
| DatasetLogRepository datasetLogRepository, |
| ServiceConfigurationRepository serviceConfigurationRepository, |
| ServiceEntryRepository serviceEntryRepository, |
| DatasetEntryRepository datasetEntryRepository, |
| DatasetRepository datasetRepository) { |
| this.climateServiceRepository = climateServiceRepository; |
| this.userRepository = userRepository; |
| this.workflowRepository = workflowRepository; |
| this.parameterRepository = parameterRepository; |
| this.serviceExecutionLogRepository = serviceExecutionLogRepository; |
| this.serviceConfigurationItemRepository = serviceConfigurationItemRepository; |
| this.datasetLogRepository = datasetLogRepository; |
| this.serviceConfigurationRepository = serviceConfigurationRepository; |
| this.serviceEntryRepository = serviceEntryRepository; |
| this.datasetEntryRepository = datasetEntryRepository; |
| this.datasetRepository = datasetRepository; |
| } |
| |
| public Result addWorkflow() { |
| JsonNode json = request().body().asJson(); |
| if (json == null) { |
| System.out.println("Dataset not saved, expecting Json data"); |
| return badRequest("Dataset not saved, expecting Json data"); |
| } |
| String name = json.findPath("name").asText(); |
| String purpose = json.findPath("purpose").asText(); |
| long createTimeNumber = json.findPath("createTime").asLong(); |
| Date createTime = new Date(createTimeNumber); |
| String versionNo = json.findPath("versionNo").asText(); |
| long rootWorkflowId = json.findPath("rootWorkflowId").asLong(); |
| |
| JsonNode users = json.findPath("userId"); |
| List<User> userSet = new ArrayList<User>(); |
| for (int i = 0; i < users.size(); i++) { |
| userSet.add(userRepository.findOne(users.get(i).asLong())); |
| } |
| JsonNode climateServices = json.findPath("climateServiceId"); |
| List<ClimateService> climateServiceSet = new ArrayList<ClimateService>(); |
| for (int i = 0; i < climateServices.size(); i++) { |
| climateServiceSet.add(climateServiceRepository |
| .findOne(climateServices.get(i).asLong())); |
| } |
| try { |
| Workflow workflow = new Workflow(name, purpose, createTime, |
| versionNo, rootWorkflowId, userSet, climateServiceSet); |
| Workflow savedWorkflow = workflowRepository.save(workflow); |
| System.out.println("Workflow saved: " + savedWorkflow.getId()); |
| return created(new Gson().toJson("Workflow saved: " |
| + savedWorkflow.getId())); |
| } catch (PersistenceException pe) { |
| pe.printStackTrace(); |
| System.out.println("Workflow not created"); |
| return badRequest("Workflow not created"); |
| } |
| |
| } |
| |
| public Result updateWorkflowById(long id) { |
| if (id < 0) { |
| System.out.println("id is negative!"); |
| return badRequest("id is negative!"); |
| } |
| JsonNode json = request().body().asJson(); |
| if (json == null) { |
| System.out.println("Workflow not saved, expecting Json data"); |
| return badRequest("Workflow not saved, expecting Json data"); |
| } |
| String name = json.findPath("name").asText(); |
| String purpose = json.findPath("purpose").asText(); |
| long createTimeNumber = json.findPath("createTime").asLong(); |
| Date createTime = new Date(createTimeNumber); |
| String versionNo = json.findPath("versionNo").asText(); |
| long rootWorkflowId = json.findPath("rootWorkflowId").asLong(); |
| |
| JsonNode users = json.findPath("userId"); |
| List<User> userSet = new ArrayList<User>(); |
| for (int i = 0; i < users.size(); i++) { |
| userSet.add(userRepository.findOne(users.get(i).asLong())); |
| } |
| JsonNode climateServices = json.findPath("climateServiceId"); |
| List<ClimateService> climateServiceSet = new ArrayList<ClimateService>(); |
| for (int i = 0; i < climateServices.size(); i++) { |
| climateServiceSet.add(climateServiceRepository |
| .findOne(climateServices.get(i).asLong())); |
| } |
| |
| try { |
| Workflow workflow = workflowRepository.findOne(id); |
| |
| workflow.setClimateServiceSet(climateServiceSet); |
| workflow.setCreateTime(createTime); |
| workflow.setName(name); |
| workflow.setPurpose(purpose); |
| workflow.setRootWorkflowId(rootWorkflowId); |
| workflow.setUserSet(userSet); |
| workflow.setVersionNo(versionNo); |
| |
| Workflow savedWorkflow = workflowRepository.save(workflow); |
| |
| System.out.println("Workflow updated: " + savedWorkflow.getId()); |
| return created("Workflow updated: " + savedWorkflow.getId()); |
| } catch (PersistenceException pe) { |
| pe.printStackTrace(); |
| System.out.println("Workflow not saved: " + id); |
| return badRequest("Workflow not saved: " + id); |
| } |
| } |
| |
| public Result deleteWorkflowById(long id) { |
| if (id < 0) { |
| System.out.println("id is negative!"); |
| return badRequest("id is negative!"); |
| } |
| Workflow workflow = workflowRepository.findOne(id); |
| if (workflow == null) { |
| System.out.println("Workflow not found with id: " + id); |
| return notFound("Workflow not found with id: " + id); |
| } |
| workflowRepository.delete(workflow); |
| System.out.println("Workflow is deleted: " + id); |
| return ok("Workflow is deleted: " + id); |
| } |
| |
| public Result getWorkflowById(long id, String format) { |
| if (id < 0) { |
| System.out.println("id is negative!"); |
| return badRequest("id is negative!"); |
| } |
| Workflow workflow = workflowRepository.findOne(id); |
| if (workflow == null) { |
| System.out.println("Workflow not found with name: " + id); |
| return notFound("Workflow not found with name: " + id); |
| } |
| |
| String result = new String(); |
| if (format.equals("json")) { |
| result = new Gson().toJson(workflow); |
| } |
| |
| return ok(result); |
| } |
| |
| public Result getAllWorkflows(String format) { |
| try { |
| Iterable<Workflow> workflows = workflowRepository.findAll(); |
| String result = new String(); |
| result = new Gson().toJson(workflows); |
| return ok(result); |
| } catch (Exception e) { |
| return badRequest("Workflows not found"); |
| } |
| } |
| |
| public Result generateWorkflowJson() { |
| ServiceExecutionLogController serviceExecutionLogController = new ServiceExecutionLogController( |
| serviceExecutionLogRepository, parameterRepository, |
| serviceConfigurationItemRepository, userRepository, |
| climateServiceRepository, datasetLogRepository, |
| serviceConfigurationRepository, serviceEntryRepository, |
| datasetEntryRepository, datasetRepository); |
| List<ServiceExecutionLog> list = serviceExecutionLogController |
| .queryServiceExecutionLogsAsList(); |
| String result = VisTrailJson.getVisTrailJson(list); |
| String timeStamp = new SimpleDateFormat("yyyy_MM_dd_HH_mm_ss") |
| .format(new Date()); |
| String location = "workflowRepository/" + timeStamp + ".json"; |
| |
| File theDir = new File("workflowRepository"); |
| |
| // if the directory does not exist, create it |
| if (!theDir.exists()) { |
| System.out.println("creating directory: " + workflowRepository); |
| boolean create = false; |
| |
| try { |
| theDir.mkdir(); |
| create = true; |
| } catch (SecurityException se) { |
| // handle it |
| } |
| if (create) { |
| System.out.println("DIR created"); |
| } |
| } else { |
| System.out.println("No"); |
| } |
| |
| try { |
| File file = new File(location); |
| BufferedWriter output = new BufferedWriter(new FileWriter(file)); |
| output.write(result); |
| output.close(); |
| } catch (FileNotFoundException e) { |
| // TODO Auto-generated catch block |
| e.printStackTrace(); |
| } catch (IOException e) { |
| // TODO Auto-generated catch block |
| e.printStackTrace(); |
| } |
| return ok(result); |
| } |
| } |
