| /* |
| * 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 org.jclouds.chef.strategy.internal; |
| |
| import static com.google.common.base.Preconditions.checkNotNull; |
| import static com.google.common.collect.Iterables.filter; |
| import static com.google.common.collect.Iterables.transform; |
| import static com.google.common.util.concurrent.Futures.allAsList; |
| import static com.google.common.util.concurrent.Futures.getUnchecked; |
| |
| import java.util.List; |
| import java.util.concurrent.Callable; |
| |
| import javax.annotation.Resource; |
| import javax.inject.Named; |
| import javax.inject.Singleton; |
| |
| import org.jclouds.Constants; |
| import org.jclouds.chef.ChefApi; |
| import org.jclouds.chef.config.ChefProperties; |
| import org.jclouds.chef.domain.Node; |
| import org.jclouds.chef.strategy.ListNodesInEnvironment; |
| import org.jclouds.logging.Logger; |
| |
| import com.google.common.base.Function; |
| import com.google.common.base.Joiner; |
| import com.google.common.base.Predicate; |
| import com.google.common.util.concurrent.ListenableFuture; |
| import com.google.common.util.concurrent.ListeningExecutorService; |
| import com.google.inject.Inject; |
| |
| /** |
| * |
| * |
| * @author Noorul Islam K M |
| */ |
| @Singleton |
| public class ListNodesInEnvironmentImpl implements ListNodesInEnvironment { |
| |
| protected final ChefApi api; |
| protected final ListeningExecutorService userExecutor; |
| @Resource |
| @Named(ChefProperties.CHEF_LOGGER) |
| protected Logger logger = Logger.NULL; |
| |
| @Inject |
| ListNodesInEnvironmentImpl(@Named(Constants.PROPERTY_USER_THREADS) ListeningExecutorService userExecutor, ChefApi api) { |
| this.userExecutor = checkNotNull(userExecutor, "userExecuor"); |
| this.api = checkNotNull(api, "api"); |
| } |
| |
| @Override |
| public Iterable<? extends Node> execute(String environmentName) { |
| return execute(userExecutor, environmentName); |
| } |
| |
| @Override |
| public Iterable<? extends Node> execute(String environmentName, Predicate<String> nodeNameSelector) { |
| return execute(userExecutor, environmentName, nodeNameSelector); |
| } |
| |
| @Override |
| public Iterable<? extends Node> execute(String environmentName, Iterable<String> toGet) { |
| return execute(userExecutor, environmentName, toGet); |
| } |
| |
| @Override |
| public Iterable<? extends Node> execute(ListeningExecutorService executor, String environmentName) { |
| return execute(executor, environmentName, api.listNodesInEnvironment(environmentName)); |
| } |
| |
| @Override |
| public Iterable<? extends Node> execute(ListeningExecutorService executor, String environmentName, Predicate<String> nodeNameSelector) { |
| return execute(executor, environmentName, filter(api.listNodesInEnvironment(environmentName), nodeNameSelector)); |
| } |
| |
| @Override |
| public Iterable<? extends Node> execute(final ListeningExecutorService executor, String environmentName, Iterable<String> toGet) { |
| ListenableFuture<List<Node>> futures = allAsList(transform(toGet, new Function<String, ListenableFuture<Node>>() { |
| @Override |
| public ListenableFuture<Node> apply(final String input) { |
| return executor.submit(new Callable<Node>() { |
| @Override |
| public Node call() throws Exception { |
| return api.getNode(input); |
| } |
| }); |
| } |
| })); |
| |
| logger.trace(String.format("getting nodes in environment %s: %s", environmentName, Joiner.on(',').join(toGet))); |
| return getUnchecked(futures); |
| } |
| |
| } |