/*
* RHQ Management Platform
* Copyright (C) 2005-2014 Red Hat, Inc.
* All rights reserved.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation version 2 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
package org.rhq.modules.plugins.wildfly10;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.codehaus.jackson.JsonNode;
import org.rhq.core.domain.configuration.Configuration;
import org.rhq.core.domain.configuration.PropertySimple;
import org.rhq.core.domain.configuration.definition.PropertyDefinitionSimple;
import org.rhq.core.domain.content.PackageDetailsKey;
import org.rhq.core.domain.content.PackageType;
import org.rhq.core.domain.content.transfer.ContentResponseResult;
import org.rhq.core.domain.content.transfer.DeployIndividualPackageResponse;
import org.rhq.core.domain.content.transfer.DeployPackageStep;
import org.rhq.core.domain.content.transfer.DeployPackagesResponse;
import org.rhq.core.domain.content.transfer.RemovePackagesResponse;
import org.rhq.core.domain.content.transfer.ResourcePackageDetails;
import org.rhq.core.pluginapi.content.ContentContext;
import org.rhq.core.pluginapi.content.ContentFacet;
import org.rhq.core.pluginapi.content.ContentServices;
import org.rhq.core.pluginapi.inventory.CreateChildResourceFacet;
import org.rhq.core.pluginapi.operation.OperationFacet;
import org.rhq.core.pluginapi.operation.OperationResult;
import org.rhq.modules.plugins.wildfly10.json.Address;
import org.rhq.modules.plugins.wildfly10.json.CompositeOperation;
import org.rhq.modules.plugins.wildfly10.json.Operation;
import org.rhq.modules.plugins.wildfly10.json.PROPERTY_VALUE;
import org.rhq.modules.plugins.wildfly10.json.ReadChildrenNames;
import org.rhq.modules.plugins.wildfly10.json.Result;
/**
* Component dealing with server group specific things
* @author Heiko W. Rupp
*/
public class ServerGroupComponent extends BaseComponent implements ContentFacet, CreateChildResourceFacet,
OperationFacet {
private static final String OUTCOME = "outcome";
@Override
public OperationResult invokeOperation(String name, Configuration parameters) throws InterruptedException,
Exception {
Operation op = new Operation(name, getAddress());
PropertyDefinitionSimple responseTimeoutDefinition = getParameterDefinitionsForOperation(name).getPropertyDefinitionSimple("responseTimeout");
int timeout = Integer.parseInt(responseTimeoutDefinition.getDefaultValue());
PropertySimple responseTimeout = parameters.getSimple("responseTimeout");
if(responseTimeout != null && responseTimeout.getStringValue() != null) {
timeout = Integer.parseInt(responseTimeout.getStringValue());
}
Result res = getASConnection().execute(op, timeout);
OperationResult result = new OperationResult();
if (res.isSuccess()) {
result.setSimpleResult(Result.SUCCESS);
} else {
result.setErrorMessage(res.getFailureDescription());
}
return result;
}
@Override
public List<DeployPackageStep> generateInstallationSteps(ResourcePackageDetails packageDetails) {
return null; // TODO: Customise this generated block
}
// TODO I think this package code is not used.
@Override
public DeployPackagesResponse deployPackages(Set<ResourcePackageDetails> packages, ContentServices contentServices) {
ContentContext cctx = context.getContentContext();
DeployPackagesResponse response = new DeployPackagesResponse();
for (ResourcePackageDetails details : packages) {
String packageName = details.getName();
ASUploadConnection uploadConnection = new ASUploadConnection(getServerComponent().getASConnection(),
packageName);
OutputStream out = uploadConnection.getOutputStream();
if (out == null) {
response.addPackageResponse(new DeployIndividualPackageResponse(details.getKey(),
ContentResponseResult.FAILURE));
continue;
}
try {
contentServices.downloadPackageBits(cctx, details.getKey(), out, false);
} catch (Exception e) {
uploadConnection.cancelUpload();
response.addPackageResponse(new DeployIndividualPackageResponse(details.getKey(),
ContentResponseResult.FAILURE));
continue;
}
JsonNode uploadResult = uploadConnection.finishUpload();
if (uploadResult.has(OUTCOME)) {
// TODO use Deployer class
String outcome = uploadResult.get(OUTCOME).getTextValue();
if (outcome.equals(Result.SUCCESS)) { // Upload was successful, so now add the file to the server group
JsonNode resultNode = uploadResult.get("result");
String hash = resultNode.get("BYTES_VALUE").getTextValue();
ASConnection connection = getASConnection();
Address deploymentsAddress = new Address();
deploymentsAddress.add("deployment", packageName);
Operation step1 = new Operation("add", deploymentsAddress);
// step1.addAdditionalProperty("hash", new PROPERTY_VALUE("BYTES_VALUE", hash));
List<Object> content = new ArrayList<Object>(1);
Map<String, Object> contentValues = new HashMap<String, Object>();
contentValues.put("hash", new PROPERTY_VALUE("BYTES_VALUE", hash));
content.add(contentValues);
step1.addAdditionalProperty("content", content);
step1.addAdditionalProperty("name", packageName);
Address serverGroupAddress = new Address(context.getResourceKey());
serverGroupAddress.add("deployment", packageName);
Operation step2 = new Operation("add", serverGroupAddress);
Operation step3 = new Operation("deploy", serverGroupAddress);
CompositeOperation cop = new CompositeOperation();
cop.addStep(step1);
cop.addStep(step2);
cop.addStep(step3);
Result result = connection.execute(cop);
if (!result.isSuccess()) // TODO get failure message into response
response.addPackageResponse(new DeployIndividualPackageResponse(details.getKey(),
ContentResponseResult.FAILURE));
else {
DeployIndividualPackageResponse individualPackageResponse = new DeployIndividualPackageResponse(
details.getKey(), ContentResponseResult.SUCCESS);
response.addPackageResponse(individualPackageResponse);
response.setOverallRequestResult(ContentResponseResult.SUCCESS);
}
} else
response.addPackageResponse(new DeployIndividualPackageResponse(details.getKey(),
ContentResponseResult.FAILURE));
} else {
response.addPackageResponse(new DeployIndividualPackageResponse(details.getKey(),
ContentResponseResult.FAILURE));
}
}
return response;
}
@Override
public RemovePackagesResponse removePackages(Set<ResourcePackageDetails> packages) {
return null; // TODO: Customise this generated block
}
@Override
public Set<ResourcePackageDetails> discoverDeployedPackages(PackageType type) {
Operation op = new ReadChildrenNames(address, "deployment"); // TODO read full packages not only names
Result node = getASConnection().execute(op);
if (!node.isSuccess())
return null;
List<String> resultList = (List<String>) node.getResult(); // TODO needs checking
Set<ResourcePackageDetails> details = new HashSet<ResourcePackageDetails>();
for (String file : resultList) {
String t;
if (file.contains("."))
t = file.substring(file.lastIndexOf(".") + 1);
else
t = "-none-";
ResourcePackageDetails detail = new ResourcePackageDetails(new PackageDetailsKey(file, "1.0", t, "all"));
details.add(detail);
}
return details;
}
@Override
public InputStream retrievePackageBits(ResourcePackageDetails packageDetails) {
return null; // TODO: Customise this generated block
}
@SuppressWarnings("unused")
private String serverGroupFromKey() {
String key1 = context.getResourceKey();
return key1.substring(key1.lastIndexOf("/") + 1);
}
}