core/src/main/java/com/cloud/info/RunningHostInfoAgregator.java

//
// 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 com.cloud.info;

import java.util.HashMap;
import java.util.Map;

import com.cloud.host.Host;

public class RunningHostInfoAgregator {

    public static class ZoneHostInfo {
        public static final int ROUTING_HOST_MASK = 2;
        public static final int STORAGE_HOST_MASK = 4;
        public static final int ALL_HOST_MASK = ROUTING_HOST_MASK | STORAGE_HOST_MASK;

        private long dcId;

        // (1 << 1) : at least one routing host is running in the zone
        // (1 << 2) : at least one storage host is running in the zone
        private int flags = 0;

        public long getDcId() {
            return dcId;
        }

        public void setDcId(long dcId) {
            this.dcId = dcId;
        }

        public void setFlag(int flagMask) {
            flags |= flagMask;
        }

        public int getFlags() {
            return flags;
        }
    }

    private final Map<Long, ZoneHostInfo> zoneHostInfoMap = new HashMap<Long, ZoneHostInfo>();

    public RunningHostInfoAgregator() {
    }

    public void aggregate(RunningHostCountInfo countInfo) {
        if (countInfo.getCount() > 0) {
            ZoneHostInfo zoneInfo = getZoneHostInfo(countInfo.getDcId());

            Host.Type type = Enum.valueOf(Host.Type.class, countInfo.getHostType());
            if (type == Host.Type.Routing) {
                zoneInfo.setFlag(ZoneHostInfo.ROUTING_HOST_MASK);
            } else if (type == Host.Type.Storage || type == Host.Type.SecondaryStorage) {
                zoneInfo.setFlag(ZoneHostInfo.STORAGE_HOST_MASK);
            }
        }
    }

    public Map<Long, ZoneHostInfo> getZoneHostInfoMap() {
        return zoneHostInfoMap;
    }

    private ZoneHostInfo getZoneHostInfo(long dcId) {
        if (zoneHostInfoMap.containsKey(dcId))
            return zoneHostInfoMap.get(dcId);

        ZoneHostInfo info = new ZoneHostInfo();
        info.setDcId(dcId);
        zoneHostInfoMap.put(dcId, info);
        return info;
    }
}