Role.java

Go to the documentation of this file.

// Copyright 2014 Cloudera Inc.
//
// 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 com.cloudera.impala.catalog;

import java.util.List;
import java.util.Set;
import java.util.concurrent.atomic.AtomicInteger;

import com.cloudera.impala.thrift.TCatalogObjectType;
import com.cloudera.impala.thrift.TRole;
import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;

public class Role implements CatalogObject {
  private final TRole role_;
  // The last role ID assigned, starts at 0.
  private static AtomicInteger roleId_ = new AtomicInteger(0);
  private long catalogVersion_ = Catalog.INITIAL_CATALOG_VERSION;

  private final CatalogObjectCache<RolePrivilege> rolePrivileges_ =
      new CatalogObjectCache<RolePrivilege>();

  public Role(String roleName, Set<String> grantGroups) {
    role_ = new TRole();
    role_.setRole_name(roleName);
    role_.setRole_id(roleId_.incrementAndGet());
    role_.setGrant_groups(Lists.newArrayList(grantGroups));
  }

  private Role(TRole role) {
    role_ = role;
  }

  public boolean addPrivilege(RolePrivilege privilege) {
    return rolePrivileges_.add(privilege);
  }

  public List<RolePrivilege> getPrivileges() {
    return Lists.newArrayList(rolePrivileges_.getValues());
  }

  public Set<String> getPrivilegeNames() {
    return Sets.newHashSet(rolePrivileges_.keySet());
  }

  public RolePrivilege getPrivilege(String privilegeName) {
    return rolePrivileges_.get(privilegeName);
  }

  public RolePrivilege removePrivilege(String privilegeName) {
    return rolePrivileges_.remove(privilegeName);
  }

  public synchronized void addGrantGroup(String groupName) {
    if (role_.getGrant_groups().contains(groupName)) return;
    role_.addToGrant_groups(groupName);
  }

  public synchronized void removeGrantGroup(String groupName) {
    role_.getGrant_groups().remove(groupName);
    // Should never have duplicates in the list of groups.
    Preconditions.checkState(!role_.getGrant_groups().contains(groupName));
  }

  public TRole toThrift() {
    return role_;
  }

  public static Role fromThrift(TRole thriftRole) {
    return new Role(thriftRole);
  }

  public Set<String> getGrantGroups() {
    return Sets.newHashSet(role_.getGrant_groups());
  }
  @Override
  public TCatalogObjectType getCatalogObjectType() { return TCatalogObjectType.ROLE; }
  @Override
  public String getName() { return role_.getRole_name(); }
  public int getId() { return role_.getRole_id(); }
  @Override
  public synchronized long getCatalogVersion() { return catalogVersion_; }
  @Override
  public synchronized void setCatalogVersion(long newVersion) {
    catalogVersion_ = newVersion;
  }
  @Override
  public boolean isLoaded() { return true; }
}