org.joyqueue.model.domain.Permission Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2019 The JoyQueue Authors.
 *
 * Licensed 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.joyqueue.model.domain;

/**
 * 权限
 */
public enum Permission {
    /**
     * 无读写权限
     */
    NONE,
    /**
     * 只读
     */
    READ,
    /**
     * 只写
     */
    WRITE,
    /**
     * 读写权限
     */
    FULL;

    public static Permission valueOf(int ordinal) {
        if (ordinal < 0 || ordinal >= values().length) {
            throw new IndexOutOfBoundsException("Invalid ordinal");
        }
        return values()[ordinal];
    }

    /**
     * 包含指定的权限
     *
     * @param permission 权限
     * @return
     */
    public boolean contain(Permission permission) {
        return this == permission || this == FULL && (permission != NONE);
    }


    /**
     * 添加写权限
     *
     * @return 返回修改后的权限
     */
    public Permission addWrite() {
        if (this == null || this == Permission.NONE) {
            return WRITE;
        } else if (this == READ) {
            return FULL;
        }
        return this;
    }

    /**
     * 删除写权限
     *
     * @return 返回修改后的权限
     */
    public Permission removeWrite() {
        if (this == null) {
            return NONE;
        }
        if (this == Permission.NONE || this == READ) {
            return this;
        } else if (this == WRITE) {
            return NONE;
        }
        return READ;
    }

    /**
     * 添加读权限
     *
     * @return 返回修改后的权限
     */
    public Permission addRead() {
        if (this == null || this == Permission.NONE) {
            return READ;
        } else if (this == WRITE) {
            return FULL;
        }
        return this;
    }

    /**
     * 删除读权限
     *
     * @return 返回修改后的权限
     */
    public Permission removeRead() {
        if (this == null) {
            return NONE;
        }
        if (this == Permission.NONE || this == WRITE) {
            return this;
        } else if (this == READ) {
            return NONE;
        }
        return WRITE;
    }

    public boolean canRead(){
        return this.contain(Permission.READ);
    }

    public boolean canWrite(){
        return this.contain(Permission.WRITE);
    }

}