DescribeLoadBalancer.java

/*
 * Copyright 2014 by SCSK Corporation.
 * 
 * This file is part of PrimeCloud Controller(TM).
 * 
 * PrimeCloud Controller(TM) 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, either version 2 of the License, or
 * (at your option) any later version.
 * 
 * PrimeCloud Controller(TM) 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 PrimeCloud Controller(TM). If not, see <http://www.gnu.org/licenses/>.
 */
package jp.primecloud.auto.api.lb;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.MediaType;

import jp.primecloud.auto.api.ApiSupport;
import jp.primecloud.auto.api.ApiValidate;
import jp.primecloud.auto.api.response.lb.AwsLoadBalancerResponse;
import jp.primecloud.auto.api.response.lb.DescribeLoadBalancerResponse;
import jp.primecloud.auto.api.response.lb.LoadBalancerHealthCheckResponse;
import jp.primecloud.auto.api.response.lb.LoadBalancerInstanceResponse;
import jp.primecloud.auto.api.response.lb.LoadBalancerListenerResponse;
import jp.primecloud.auto.api.response.lb.LoadBalancerResponse;
import jp.primecloud.auto.common.status.ComponentInstanceStatus;
import jp.primecloud.auto.common.status.LoadBalancerInstanceStatus;
import jp.primecloud.auto.entity.crud.AwsLoadBalancer;
import jp.primecloud.auto.entity.crud.ComponentInstance;
import jp.primecloud.auto.entity.crud.LoadBalancer;
import jp.primecloud.auto.entity.crud.LoadBalancerHealthCheck;
import jp.primecloud.auto.entity.crud.LoadBalancerInstance;
import jp.primecloud.auto.entity.crud.LoadBalancerListener;
import jp.primecloud.auto.entity.crud.User;
import jp.primecloud.auto.service.impl.Comparators;

import org.apache.commons.lang.BooleanUtils;
import org.apache.commons.lang.StringUtils;

import com.amazonaws.services.ec2.model.Subnet;

@Path("/DescribeLoadBalancer")
public class DescribeLoadBalancer extends ApiSupport {

    /**
     * ロードバランサ情報取得
     *
     * @param loadBalancerNo ロードバランサ番号
     * @return DescribeLoadBalancerResponse
     */
    @GET
    @Produces(MediaType.APPLICATION_JSON)
    public DescribeLoadBalancerResponse describeLoadBalancer(
            @QueryParam(PARAM_NAME_LOAD_BALANCER_NO) String loadBalancerNo) {

        // 入力チェック
        // LoadBalancerNo
        ApiValidate.validateLoadBalancerNo(loadBalancerNo);

        // ロードバランサ取得
        LoadBalancer loadBalancer = getLoadBalancer(Long.parseLong(loadBalancerNo));

        // 権限チェック
        User user = checkAndGetUser(loadBalancer);

        //ロードバランサ情報設定
        LoadBalancerResponse loadBalancerResponse = new LoadBalancerResponse(loadBalancer);

        //リスナー取得
        List<LoadBalancerListener> listeners = loadBalancerListenerDao.readByLoadBalancerNo(Long
                .parseLong(loadBalancerNo));
        if (listeners.isEmpty() == false) {
            //ソート
            Collections.sort(listeners, Comparators.COMPARATOR_LOAD_BALANCER_LISTENER);
        }
        for (LoadBalancerListener listener : listeners) {
            //リスナー情報設定
            loadBalancerResponse.getListeners().add(new LoadBalancerListenerResponse(listener));
        }

        //ヘルスチェック取得
        LoadBalancerHealthCheck healthCheck = loadBalancerHealthCheckDao.read(Long.parseLong(loadBalancerNo));
        if (healthCheck != null) {
            //ヘルスチェック情報設定
            loadBalancerResponse.setHealthCheck(new LoadBalancerHealthCheckResponse(healthCheck));
        }

        //コンポーネントインスタンス情報取得
        List<ComponentInstance> componentInstances = componentInstanceDao.readByComponentNo(loadBalancer
                .getComponentNo());
        if (componentInstances.isEmpty() == false) {
            //ソート
            Collections.sort(componentInstances, Comparators.COMPARATOR_COMPONENT_INSTANCE);
        }

        for (ComponentInstance componentInstance : componentInstances) {
            // 関連付けが無効で停止している場合は除外
            if (BooleanUtils.isNotTrue(componentInstance.getAssociate())) {
                ComponentInstanceStatus status = ComponentInstanceStatus.fromStatus(componentInstance.getStatus());
                if (status == ComponentInstanceStatus.STOPPED) {
                    continue;
                }
            }
            //ロードバランサインスタンス取得
            LoadBalancerInstance loadBalancerInstance = loadBalancerInstanceDao.read(Long.parseLong(loadBalancerNo),
                    componentInstance.getInstanceNo());

            LoadBalancerInstanceResponse loadBalancerInstanceResponse;
            if (loadBalancerInstance == null) {
                //ロードバランサインスタンスが存在しない場合、ロードバランサインスタンス作成時のデータを設定
                LoadBalancerInstance loadBalancerInstance2 = new LoadBalancerInstance();
                loadBalancerInstance2.setInstanceNo(componentInstance.getInstanceNo());
                loadBalancerInstance2.setEnabled(false);
                loadBalancerInstance2.setStatus(LoadBalancerInstanceStatus.STOPPED.toString());
                loadBalancerInstanceResponse = new LoadBalancerInstanceResponse(loadBalancerInstance2);
            } else {
                loadBalancerInstanceResponse = new LoadBalancerInstanceResponse(loadBalancerInstance);
            }
            //ロードバランサインスタンス情報設定
            loadBalancerResponse.getInstances().add(loadBalancerInstanceResponse);
        }

        // AWS
        if (LB_TYPE_ELB.equals(loadBalancer.getType())) {
            AwsLoadBalancer awsLoadBalancer = awsLoadBalancerDao.read(loadBalancer.getLoadBalancerNo());
            AwsLoadBalancerResponse awsLoadBalancerResponse = new AwsLoadBalancerResponse(awsLoadBalancer);

            // Subnet
            if (StringUtils.isNotEmpty(awsLoadBalancer.getSubnetId())) {
                List<Subnet> subnets = awsDescribeService.getSubnets(user.getUserNo(), loadBalancer.getPlatformNo());

                List<String> cidrBlocks = new ArrayList<String>();
                for (String subnetId : StringUtils.split(awsLoadBalancer.getSubnetId(), ",")) {
                    for (Subnet subnet : subnets) {
                        if (StringUtils.equals(subnet.getSubnetId(), subnetId)) {
                            cidrBlocks.add(subnet.getCidrBlock());
                            break;
                        }
                    }
                }

                awsLoadBalancerResponse.setSubnets(StringUtils.join(cidrBlocks, ","));
            }

            loadBalancerResponse.setAws(awsLoadBalancerResponse);
        }

        DescribeLoadBalancerResponse response = new DescribeLoadBalancerResponse(loadBalancerResponse);

        return response;
    }

}