Как вернуть Enum, если совпадает с заданной строкой - PullRequest
Новая
       24

Как вернуть Enum, если совпадает с заданной строкой

0 голосов
/ 20 ноября 2018

У меня есть класс Enum, как указано ниже 

public enum AlgorithmEnum {

    SHA512("RSA", "SHA512", 1), SHA1("RSA", "SHA1", 1), SHA384("RSA", "SHA384", 1);

    private String keyAlgorithm;
    private String hashAlgorithm;
    private Integer key;

    private AlgorithmEnum(String keyAlgorithm, String hashAlgorithm, Integer key) {
        this.keyAlgorithm = keyAlgorithm;
        this.hashAlgorithm = hashAlgorithm;
        this.key = key;
    }

    public String getKeyAlgorithm() {
        return keyAlgorithm;
    }

    public void setKeyAlgorithm(String keyAlgorithm) {
        this.keyAlgorithm = keyAlgorithm;
    }

    public String getHashAlgorithm() {
        return hashAlgorithm;
    }

    public void setHashAlgorithm(String hashAlgorithm) {
        this.hashAlgorithm = hashAlgorithm;
    }

    public Integer getKey() {
        return key;
    }

    public void setKey(Integer key) {
        this.key = key;
    }
}

Мне нужно иметь метод, подобный приведенному ниже, который принимает входные данные в виде строки и возвращает Enum 

public AlgorithmEnum getAlgorithm(String algorithm){
        //returns AlgorithmEnum object
    }

Я бы вызвал метод вышепередача "SHA512withRSA" в качестве входных данных для метода getAlgorithm.

Мне нужна помощь в реализации метода getAlgorithm.

Ответы [ 4 ]

0 голосов
/ 20 ноября 2018

Я оставляю вам пример того, как я делал в подобных случаях, вы можете легко адаптировать его к вашим потребностям: 

private static Map<Integer, YourEnum> valuesById = new HashMap<>();
private static Map<String, YourEnum> valuesByCode = new HashMap<>();

    static {
        Arrays.stream(YourEnum.values()).forEach(value -> valuesById.put(value.reasonId, value));
        Arrays.stream(YourEnum.values()).forEach(value -> valuesByCode.put(value.reasonCode, value));
    }


    public static YourEnum getByReasonId(int endReason) {
        return valuesById.get(endReason);
    }
0 голосов
/ 20 ноября 2018

Вы можете иметь что-то вроде: 

public static AlgorithmEnum getAlgorithm(final String algorithm)
        throws IllegalArgumentException
    {
        for (final AlgorithmEnum algorithmEnum : AlgorithmEnum.values())
        {
            if (algorithm.equalsIgnoreCase(String.format("%swith%s", algorithmEnum.getHashAlgorithm(), algorithmEnum.getKeyAlgorithm())))
            {
                return algorithmEnum;
            }
        }
        throw new IllegalArgumentException("Unknown algorithm: " + algorithm);
    }

Однако я не буду предлагать использовать этот подход.Вместо этого используйте 2 разных аргумента вместо одной строки.

0 голосов
/ 20 ноября 2018

Вы можете проверить, содержит ли данное String значение, которое соответствует одному из атрибутов enum с некоторыми if утверждениями: 

public AlgorithmEnum getAlgorithm(String algorithm) {
    if (algorithm.contains("SHA1")) {
        return SHA1;
    } else if (algorithm.contains("SHA512")) {
        return SHA512;
    } else if (algorithm.contains("SHA384")) {
        return SHA384;
    } else {
        return null;
    }
}

Обратите внимание, что это будет соответствовать String s, как "SHA512withoutRSA", тоже ...

Возможно, такой метод, как 

public AlgorithmEnum getAlgorithm(String keyAlgorithm, String hashAlgorithm)

, будет лучше.Однако тогда вам нужно будет указать два параметра.

0 голосов
/ 20 ноября 2018

Предполагая, что все строковые значения, переданные в ваш метод getAlgorithm() оканчиваются на withRSA, вы можете использовать следующее для извлечения значений перечисления: 

public AlgorithmEnum getAlgorithm(String algorithm) {
    return AlgorithmEnum.valueOf(algorithm.substring(0, algorithm.indexOf("withRSA")));
}
...