/*
* DBeaver - Universal Database Manager
* Copyright (C) 2010-2017 Serge Rider (serge@jkiss.org)
*
* 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.jkiss.dbeaver.ext.oracle.model;
import org.jkiss.code.NotNull;
import org.jkiss.code.Nullable;
import org.jkiss.dbeaver.Log;
import org.jkiss.dbeaver.DBException;
import org.jkiss.dbeaver.model.DBPRefreshableObject;
import org.jkiss.dbeaver.model.DBPSaveableObject;
import org.jkiss.dbeaver.model.access.DBAUser;
import org.jkiss.dbeaver.model.exec.jdbc.JDBCResultSet;
import org.jkiss.dbeaver.model.exec.jdbc.JDBCSession;
import org.jkiss.dbeaver.model.exec.jdbc.JDBCPreparedStatement;
import org.jkiss.dbeaver.model.exec.jdbc.JDBCStatement;
import org.jkiss.dbeaver.model.impl.jdbc.cache.JDBCObjectCache;
import org.jkiss.dbeaver.model.meta.Association;
import org.jkiss.dbeaver.model.runtime.DBRProgressMonitor;
import org.jkiss.dbeaver.model.struct.DBSObject;
import java.sql.SQLException;
import java.util.Collection;
/**
* OracleGrantee
*/
public abstract class OracleGrantee extends OracleGlobalObject implements DBAUser, DBPSaveableObject, DBPRefreshableObject
{
private static final Log log = Log.getLog(OracleGrantee.class);
final RolePrivCache rolePrivCache = new RolePrivCache();
private final SystemPrivCache systemPrivCache = new SystemPrivCache();
private final ObjectPrivCache objectPrivCache = new ObjectPrivCache();
public OracleGrantee(OracleDataSource dataSource) {
super(dataSource, true);
}
@Association
public Collection<OraclePrivRole> getRolePrivs(DBRProgressMonitor monitor) throws DBException
{
return rolePrivCache.getAllObjects(monitor, this);
}
@Association
public Collection<OraclePrivSystem> getSystemPrivs(DBRProgressMonitor monitor) throws DBException
{
return systemPrivCache.getAllObjects(monitor, this);
}
@Association
public Collection<OraclePrivObject> getObjectPrivs(DBRProgressMonitor monitor) throws DBException
{
return objectPrivCache.getAllObjects(monitor, this);
}
@Nullable
@Override
public DBSObject refreshObject(@NotNull DBRProgressMonitor monitor) throws DBException {
rolePrivCache.clearCache();
systemPrivCache.clearCache();
objectPrivCache.clearCache();
return this;
}
static class RolePrivCache extends JDBCObjectCache<OracleGrantee, OraclePrivRole> {
@Override
protected JDBCStatement prepareObjectsStatement(@NotNull JDBCSession session, @NotNull OracleGrantee owner) throws SQLException
{
final JDBCPreparedStatement dbStat = session.prepareStatement(
"SELECT * FROM DBA_ROLE_PRIVS WHERE GRANTEE=? ORDER BY GRANTED_ROLE");
dbStat.setString(1, owner.getName());
return dbStat;
}
@Override
protected OraclePrivRole fetchObject(@NotNull JDBCSession session, @NotNull OracleGrantee owner, @NotNull JDBCResultSet resultSet) throws SQLException, DBException
{
return new OraclePrivRole(owner, resultSet);
}
}
static class SystemPrivCache extends JDBCObjectCache<OracleGrantee, OraclePrivSystem> {
@Override
protected JDBCStatement prepareObjectsStatement(@NotNull JDBCSession session, @NotNull OracleGrantee owner) throws SQLException
{
final JDBCPreparedStatement dbStat = session.prepareStatement(
"SELECT * FROM DBA_SYS_PRIVS WHERE GRANTEE=? ORDER BY PRIVILEGE");
dbStat.setString(1, owner.getName());
return dbStat;
}
@Override
protected OraclePrivSystem fetchObject(@NotNull JDBCSession session, @NotNull OracleGrantee owner, @NotNull JDBCResultSet resultSet) throws SQLException, DBException
{
return new OraclePrivSystem(owner, resultSet);
}
}
static class ObjectPrivCache extends JDBCObjectCache<OracleGrantee, OraclePrivObject> {
@Override
protected JDBCStatement prepareObjectsStatement(@NotNull JDBCSession session, @NotNull OracleGrantee owner) throws SQLException
{
final JDBCPreparedStatement dbStat = session.prepareStatement(
"SELECT p.*,o.OBJECT_TYPE\n" +
"FROM DBA_TAB_PRIVS p, DBA_OBJECTS o\n" +
"WHERE p.GRANTEE=? " +
"AND o.OWNER=p.OWNER AND o.OBJECT_NAME=p.TABLE_NAME AND o.OBJECT_TYPE<>'PACKAGE BODY'");
dbStat.setString(1, owner.getName());
return dbStat;
}
@Override
protected OraclePrivObject fetchObject(@NotNull JDBCSession session, @NotNull OracleGrantee owner, @NotNull JDBCResultSet resultSet) throws SQLException, DBException
{
return new OraclePrivObject(owner, resultSet);
}
}
}