Java tutorial
/* * 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 brooklyn.entity.chef; import java.util.Map; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import brooklyn.entity.Entity; import brooklyn.entity.effector.EffectorTasks; import brooklyn.entity.software.SshEffectorTasks; import brooklyn.management.TaskAdaptable; import brooklyn.management.TaskFactory; import brooklyn.util.collections.MutableMap; import brooklyn.util.file.ArchiveTasks; import brooklyn.util.file.ArchiveUtils.ArchiveType; import brooklyn.util.net.Urls; import brooklyn.util.ssh.BashCommands; import brooklyn.util.task.DynamicTasks; import brooklyn.util.task.TaskBuilder; import brooklyn.util.task.Tasks; import brooklyn.util.text.Identifiers; import brooklyn.util.text.Strings; import com.google.common.annotations.Beta; import com.google.common.collect.ImmutableList; import com.google.gson.Gson; import com.google.gson.GsonBuilder; @Beta public class ChefTasks { private static final Logger log = LoggerFactory.getLogger(ChefTasks.class); public static TaskFactory<?> installChef(String chefDirectory, boolean force) { // TODO check on entity whether it is chef _server_ String installCmd = cdAndRun(chefDirectory, ChefBashCommands.INSTALL_FROM_OPSCODE); if (!force) installCmd = BashCommands.alternatives("which chef-solo", installCmd); return SshEffectorTasks.ssh(installCmd).summary("install chef"); } public static TaskFactory<?> installCookbooks(final String chefDirectory, final Map<String, String> cookbooksAndUrls, final boolean force) { return Tasks.<Void>builder().name("install " + (cookbooksAndUrls == null ? "0" : cookbooksAndUrls.size()) + " cookbook" + Strings.s(cookbooksAndUrls)).body(new Runnable() { public void run() { Entity e = EffectorTasks.findEntity(); if (cookbooksAndUrls == null) throw new IllegalStateException("No cookbooks defined to install at " + e); for (String cookbook : cookbooksAndUrls.keySet()) DynamicTasks.queue(installCookbook(chefDirectory, cookbook, cookbooksAndUrls.get(cookbook), force)); } }).buildFactory(); } public static TaskFactory<?> installCookbook(final String chefDirectory, final String cookbookName, final String cookbookArchiveUrl, final boolean force) { return new TaskFactory<TaskAdaptable<?>>() { @Override public TaskAdaptable<?> newTask() { TaskBuilder<Void> tb = Tasks.<Void>builder().name("install cookbook " + cookbookName); String cookbookDir = Urls.mergePaths(chefDirectory, cookbookName); String privateTmpDirContainingUnpackedCookbook = Urls.mergePaths(chefDirectory, "tmp-" + Strings.makeValidFilename(cookbookName) + "-" + Identifiers.makeRandomId(4)); // TODO - skip the install earlier if it exists and isn't forced // if (!force) { // // in builder.body, check // // "ls "+cookbookDir // // and stop if it's zero // // remove reference to 'force' below // } String destName = null; if (ArchiveType.of(cookbookArchiveUrl) == ArchiveType.UNKNOWN) { destName = cookbookName + ".tgz"; log.debug("Assuming TGZ type for chef cookbook url " + cookbookArchiveUrl + "; it will be downloaded as " + destName); } tb.add(ArchiveTasks.deploy(null, null, cookbookArchiveUrl, EffectorTasks.findSshMachine(), privateTmpDirContainingUnpackedCookbook, false, null, destName).newTask()); String installCmd = BashCommands.chain("cd " + privateTmpDirContainingUnpackedCookbook, "COOKBOOK_EXPANDED_DIR=`ls`", BashCommands.requireTest("`ls | wc -w` -eq 1", "The deployed archive " + cookbookArchiveUrl + " must contain exactly one directory"), "mv $COOKBOOK_EXPANDED_DIR '../" + cookbookName + "'", "cd ..", "rm -rf '" + privateTmpDirContainingUnpackedCookbook + "'"); installCmd = force ? BashCommands.alternatives("rm -rf " + cookbookDir, installCmd) : BashCommands.alternatives("ls " + cookbookDir + " > /dev/null 2> /dev/null", installCmd); tb.add(SshEffectorTasks.ssh(installCmd).summary("renaming cookbook dir").requiringExitCodeZero() .newTask()); return tb.build(); } }; } protected static String cdAndRun(String targetDirectory, String command) { return BashCommands.chain("mkdir -p '" + targetDirectory + "'", "cd '" + targetDirectory + "'", command); } public static TaskFactory<?> buildChefFile(String runDirectory, String chefDirectory, String phase, Iterable<? extends String> runList, Map<String, Object> optionalAttributes) { // TODO if it's server, try knife first // TODO configure add'l properties String phaseRb = "root = " + "'" + runDirectory + "'" // recommended alternate to runDir is the following, but it is not available in some rubies //+ File.absolute_path(File.dirname(__FILE__))"+ + "\n" + "file_cache_path root\n" + // "cookbook_path root + '/cookbooks'\n"; "cookbook_path '" + chefDirectory + "'\n"; Map<String, Object> phaseJsonMap = MutableMap.of(); if (optionalAttributes != null) phaseJsonMap.putAll(optionalAttributes); if (runList != null) phaseJsonMap.put("run_list", ImmutableList.copyOf(runList)); Gson json = new GsonBuilder().create(); String phaseJson = json.toJson(phaseJsonMap); return Tasks.sequential("build chef files for " + phase, SshEffectorTasks.put(Urls.mergePaths(runDirectory) + "/" + phase + ".rb").contents(phaseRb) .createDirectory(), SshEffectorTasks.put(Urls.mergePaths(runDirectory) + "/" + phase + ".json").contents(phaseJson)); } public static TaskFactory<?> runChef(String runDir, String phase) { // TODO chef server return SshEffectorTasks .ssh(cdAndRun(runDir, "sudo chef-solo -c " + phase + ".rb -j " + phase + ".json -ldebug")) .summary("run chef for " + phase).requiringExitCodeZero(); } }