DiscoveryITBMServer.java

package org.hyperic.plugin.vrealize.automation;

import static org.hyperic.plugin.vrealize.automation.VraConstants.KEY_APPLICATION_NAME;
import static org.hyperic.plugin.vrealize.automation.VraConstants.TYPE_ITBM_SERVER_GROUP;
import static org.hyperic.plugin.vrealize.automation.VraConstants.TYPE_VRA_APPLICATION;

import java.util.List;
import java.util.Properties;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hyperic.hq.product.PluginException;
import org.hyperic.hq.product.ServerResource;
import org.hyperic.util.config.ConfigResponse;

import com.vmware.hyperic.model.relations.ObjectFactory;
import com.vmware.hyperic.model.relations.RelationType;
import com.vmware.hyperic.model.relations.Resource;
import com.vmware.hyperic.model.relations.ResourceTier;

/**
 * @author Sharon Rozinsky
 */
public class DiscoveryITBMServer extends Discovery {
    private static final String CSP_COMPONENT_REGISTRY_URL = "csp.component.registry.url";
    private static final String SECURITY_PROPERTIES_FILE_PATH =
                "/usr/local/tcserver/vfabric-tc-server-standard/itbm-server/conf/security.properties";
    private static final Log log = LogFactory.getLog(DiscoveryITBMServer.class);

    @Override
    public List<ServerResource> getServerResources(ConfigResponse platformConfig)
                throws PluginException {
        log.debug("[getServerResources] platformConfig=" + platformConfig);
        String platformFqdn = platformConfig.getValue("platform.fqdn");
        VRAUtils.setLocalFqdn(platformFqdn);
        @SuppressWarnings("unchecked")
        List<ServerResource> servers = super.getServerResources(platformConfig);

        Properties props = VRAUtils.configFile(SECURITY_PROPERTIES_FILE_PATH);
        String applicationName = VRAUtils.getFqdn(props.getProperty(CSP_COMPONENT_REGISTRY_URL));
        log.debug("[getServerResources] " + CSP_COMPONENT_REGISTRY_URL + " = " + applicationName);

        if (applicationName != null) {
            for (ServerResource serverResource : servers) {
                Resource modelResource = getCommonModel(serverResource, applicationName);
                String modelXml = VRAUtils.marshallResource(modelResource);
                VRAUtils.setModelProperty(serverResource, modelXml);
            }
        }
        return servers;
    }

    private Resource getCommonModel(
                ServerResource server, String applicationName) {
        ObjectFactory objectFactory = new ObjectFactory();

        Resource itbmServer =
                    objectFactory.createResource(false, server.getType(), server.getName(), ResourceTier.SERVER);
        Resource itbmGroup = objectFactory.createLogicalResource(TYPE_ITBM_SERVER_GROUP, applicationName);
        Resource application = objectFactory.createApplicationResource(TYPE_VRA_APPLICATION, applicationName);
        application.addProperty(objectFactory.createProperty(KEY_APPLICATION_NAME, applicationName));

        itbmServer.addRelations(objectFactory.createRelation(itbmGroup, RelationType.PARENT));
        itbmGroup.addRelations(objectFactory.createRelation(application, RelationType.PARENT));

        return itbmServer;
    }

}