com.vmware.thinapp.workpool.CloneRunnerImpl.java Source code

Introduction

Here is the source code for com.vmware.thinapp.workpool.CloneRunnerImpl.java

Source

/* ***********************************************************************
 * VMware ThinApp Factory
 * Copyright (c) 2009-2013 VMware, Inc. All Rights Reserved.
 *
 * Licensed 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 com.vmware.thinapp.workpool;

import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;
import org.springframework.util.FileCopyUtils;
import org.springframework.util.StringUtils;

import com.google.common.base.Charsets;
import com.google.common.io.CharStreams;
import com.google.common.io.Files;

@Component("cloneRunner")
@Scope("prototype")
public class CloneRunnerImpl implements CloneRunner {
    private static final Logger log = LoggerFactory.getLogger(CloneRunnerImpl.class);

    @Value("#{workpoolProperties['clonevm']}")
    private String CLONE_VM;
    private CloneRequest cloneRequest;

    public CloneRunnerImpl(CloneRequest cloneRequest) {
        this.cloneRequest = cloneRequest;
    }

    @Override
    public Result run() throws IOException, InterruptedException {
        log.debug("Using clone-vm script from: {}.", CLONE_VM);
        File ini = File.createTempFile("clone-vm", null);
        try {
            String rawContents = cloneRequest.toIni(CloneRequest.nullScrubber);
            String redactedContents = cloneRequest.toIni(CloneRequest.redactedScrubber);
            log.info("\n" + redactedContents);
            FileCopyUtils.copy(rawContents.getBytes(Charsets.UTF_8), ini);
            log.info("Executing clone-vm request.");

            ProcessBuilder pb = new ProcessBuilder(CLONE_VM, ini.getAbsolutePath());
            // Have to set PYTHON_EGG_CACHE because otherwise Python will try to extract temporary files to
            // /usr/share/tomcat6 which will not work.
            pb.environment().put("PYTHON_EGG_CACHE", "/tmp/tomcat-egg-cache");
            Process p = pb.start();

            int ret = p.waitFor();
            log.info("clone-vm exited with code: {}.", ret);

            String stderr = CharStreams.toString(new InputStreamReader(p.getErrorStream(), Charsets.UTF_8)).trim();

            if (StringUtils.hasLength(stderr)) {
                ret = -1; // suppress content (if any) from stdout and enforce the error handling code path
                log.error(stderr);
            }

            // Relog log file.
            for (String line : Files.readLines(cloneRequest.getLogFile(), Charsets.UTF_8)) {
                log.info(line);
            }

            cloneRequest.getLogFile().delete();

            String moid = "";

            if (ret == 0) {
                moid = CharStreams.toString(new InputStreamReader(p.getInputStream(), Charsets.UTF_8)).trim();
                log.debug("Received clone with moid {}.", moid);
            }

            return new Result(moid, StringUtils.hasLength(moid), stderr);
        } finally {
            if (!ini.delete()) {
                log.error("Failed to delete file: {}.", ini);
            }
        }
    }
}