engine/api/src/main/java/org/apache/cloudstack/engine/subsystem/api/storage/type/VolumeTypeHelper.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 org.apache.cloudstack.engine.subsystem.api.storage.type;

import java.util.Hashtable;
import java.util.List;
import java.util.Map;

import javax.inject.Inject;

public class VolumeTypeHelper {

    private static VolumeType defaultType = new Unknown();

    private List<VolumeType> types;
    private final Map<String, VolumeType> mapTypes = new Hashtable<String, VolumeType>();

    @Inject
    public void setTypes(List<VolumeType> types) {
        this.types = types;

        mapTypes.clear();
        for (VolumeType ty : this.types) {
            mapTypes.put(ty.getClass().getSimpleName().toUpperCase(), ty);
        }
    }

    public VolumeType getType(String type) {
        if (mapTypes.containsKey(type.toUpperCase())) {
            return mapTypes.get(type.toUpperCase());
        }
        return VolumeTypeHelper.defaultType;
    }
}