Matching enum Strings in Java in If conditions

Question

I have an enum:

public enum Status{
   A("A"),
   B("B"),
   C("C"),
   D("D"),
   E("E")
}

A,B,C are in one category and D&E are a different category. Now when I get a string, I want to decide which category it falls in. Basically,

String s;
if   s.equals( Status.A.toString()) || s.equals(Status.B.toString()) || s.equals( Status.C.toString()) 
            return 1; 

        else return 2;

Now, this is manageable with 5 letters. If I have 26 letters, the number of if conditions will be unmanageable. Is there a better way to write this?

@John3136 Yes I do know. Just wrote it quickly. Updated question — user1692342
– user1692342, Commented Mar 29, 2017 at 3:18

Sergey Kalinichenko · Accepted Answer · 2017-03-29 03:22:12Z

5

Rather than deciding the category in the code, store it in the enum itself. You are not limited to a single stored attribute; you can have as many as you wish.

Change the constructor to take the category in addition to the letter, and store it in the enum:

public enum Status {
    A("A", 1),
    B("B", 1),
    C("C", 1),
    D("D", 2),
    E("E", 2);
    private String name;
    private int category;
    public String getName() { return name; }
    public int getCategory() { return category; }
    Status(String name, int category) {
        this.name = name;
        this.category = category;
    }
}

Now it is very clear which enum belongs to what category, because categories are assigned declaratively rather than algorithmically.

answered Mar 29, 2017 at 3:22

Sergey Kalinichenko

729k85 gold badges1.2k silver badges1.6k bronze badges

Sign up to request clarification or add additional context in comments.

4 Comments

GhostCat Over a year ago

I would suggest there to make the category an enum itself instead of using raw ints.

Sergey Kalinichenko Over a year ago

@GhostCat Absolutely! Assuming that category has some meaning (as opposed to being, say, an indexer into some array) using an enum for category would be a good idea.

user1692342 Over a year ago

@dasblinkenlight I like this solution. However, is there no way to take advantage of the fact that a letter is mapped to the value. so say letter B = 2. Is there a way to do something like if(a.value() < s.value() < e.value()) return category1 ?

Sergey Kalinichenko Over a year ago

@user1692342 You can definitely rely on enums ordering: .ordinal() is always there, so you can compare them as needed. However, as soon as you do that, the definition of category becomes implicit in code again, rather than being declarative, and it also becomes dependent on the declaration order of enum values, which is less than ideal.

VHS · Accepted Answer · 2017-03-29 03:32:41Z

2

How about you maintain a list of enums per category and then compare your string with the enums in a category list? See the following example.

    List<Status> category1 = new ArrayList<Status>();
    category1.add(Status.A);
    category1.add(Status.B);
    category1.add(Status.C);
    List<Status> category2 = new ArrayList<Status>();
    category2.add(Status.D);
    category2.add(Status.E);

    for(Status e : category1) {
        if(s.equals(e.toString()))
            return true;
    }

    for(Status e : category2) {
        if(s.equals(e.toString()))
            return true;
    }

    return false;

answered Mar 29, 2017 at 3:32

VHS

10.2k3 gold badges25 silver badges48 bronze badges

1 Comment

GhostCat Over a year ago

Also a good approach as it decouples the category from the enum.

Collectives™ on Stack Overflow

Matching enum Strings in Java in If conditions

2 Answers 2

4 Comments

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

4 Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related