在Java中创建权限位掩码

17

我想做类似于这样的事情:

public enum Permissions
{
    CanBlah1,
    CanBlah2,
    CanBlah3
}

byte[] userPerm = Permissions.CanBlah1 | Permissions.CanBlah2;

// check permssions
//
if(userPerm && Permissions.CanBlah1 == Permissions.CanBlah1)
{
      // do something
}

你能用Java这样做吗?(我之前是C#背景)

5个回答

38

你可以使用EnumSet轻松地实现它。

import java.util.EnumSet;

import static java.util.EnumSet.of;
import static java.util.EnumSet.range;
import static so.User.Permissions.CanBlah1;
import static so.User.Permissions.CanBlah2;
import static so.User.Permissions.CanBlah3;

public class User {
    public enum Permissions {
        CanBlah1,
        CanBlah2,
        CanBlah3
    }

    public static void main(String[] args) throws Exception {
        EnumSet<Permissions> userPerms = of(CanBlah1, CanBlah2);
        System.out.println(userPerms.contains(CanBlah1)); //true
        System.out.println(userPerms.contains(CanBlah2)); //true
        System.out.println(userPerms.contains(CanBlah3)); //false
        System.out.println(userPerms.containsAll(of(CanBlah1, CanBlah3))); //false
        System.out.println(userPerms.containsAll(range(CanBlah1, CanBlah2))); //true
        System.out.println(userPerms.containsAll(range(CanBlah1, CanBlah3))); //false
    }

}

9
当然,EnumSet在底层是以位掩码的形式实现的。 - Ross
它们是。它使用long来处理最多64个Enums字节数组,被称为JubmoSet - Chandra Patni

7
这是另一种选项,与顺序解决方案类似,但您可以使用|和&运算符来完成:
 public enum Permissions {
     CanBlah1(1),
     CanBlah2(2),
     CanBlah3(4);

     public int value;

     Permissions(int value) {
         this.value = value;
     }
     public int value() {
      return value;
     }
 }

 public static void main(String[] args) {  
    int userPerm = Permissions.CanBlah1.value() | Permissions.CanBlah2.value();
    // check permssions
    //
    if((userPerm & Permissions.CanBlah1.value()) == Permissions.CanBlah1.value())
    {
        // do something
    }
 }

或者:

 public enum Permissions {
         CanBlah1,
         CanBlah2,
         CanBlah3;

         public int value() {
            return 1<<ordinal();
         }
     }

     public static void main(String[] args) {  
        int userPerm = Permissions.CanBlah1.value() | Permissions.CanBlah2.value();
        // check permssions
        //
        if((userPerm & Permissions.CanBlah1.value()) == Permissions.CanBlah1.value())
        {
            // do something
        }
     }

3

虽然我不建议这样做,但您可以请求枚举的ordinal()并将其用于位操作。当然,由于您无法定义枚举的ordinal是什么,因此必须插入虚假值以获取正确的序号。

enum Example {
   Bogus,            --> 0
   This,             --> 1
   That,             --> 2
   ThisOrThat        --> 3
};

注意,需要引入一个虚假的枚举类型,以便

ThisOrThat.ordinal() == This.ordinal() | That.ordinal()

2
如果你被困在 Java 7 以前的时代(Android),可以尝试以下代码:
public enum STUFF_TO_BIT_BASK {
THIS,THAT,OTHER;

public static int getBitMask(STUFF_TO_BIT_BASK... masks) {
    int res = 0;

    for (STUFF_TO_BIT_BASK cap : masks) {
        res |= (int) Math.pow(2, cap.ordinal());
    }

    return res;
}

public boolean is(int maskToCheck){
    return maskToCheck | (int) Math.pow(2, this.ordinal());
}

}


0
据我所知,对于枚举类型,按位运算符是未定义的。

是的,但是EnumSet具有基本的集合操作,例如通过addAll()进行并集操作,通过retainAll()进行交集操作。另请参见:http://en.wikipedia.org/wiki/Set_(mathematics)。 - trashgod

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接