com.clustercontrol.accesscontrol.util.ObjectPrivilegeUtil.java Source code

Introduction

Here is the source code for com.clustercontrol.accesscontrol.util.ObjectPrivilegeUtil.java

Source

/*
    
Copyright (C) 2012 NTT DATA Corporation
    
This program is free software; you can redistribute it and/or
Modify it under the terms of the GNU General Public License
as published by the Free Software Foundation, version 2.
    
This program is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied
warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE.  See the GNU General Public License for more details.
    
 */

package com.clustercontrol.accesscontrol.util;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import javax.persistence.EntityManagerFactory;
import javax.persistence.metamodel.EntityType;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.clustercontrol.accesscontrol.annotation.HinemosObjectPrivilege;
import com.clustercontrol.accesscontrol.bean.PrivilegeConstant.ObjectPrivilegeMode;
import com.clustercontrol.accesscontrol.model.ObjectPrivilegeInfo;
import com.clustercontrol.accesscontrol.model.ObjectPrivilegeTargetInfo;
import com.clustercontrol.bean.HinemosModuleConstant;
import com.clustercontrol.commons.util.HinemosEntityManager;
import com.clustercontrol.commons.util.JpaTransactionManager;
import com.clustercontrol.fault.JobMasterNotFound;
import com.clustercontrol.fault.ObjectPrivilege_InvalidRole;
import com.clustercontrol.jobmanagement.model.JobMstEntityPK;

/**
 * ???Util
 *
 */
public class ObjectPrivilegeUtil {
    private static Log m_log = LogFactory.getLog(ObjectPrivilegeUtil.class);

    /** ??Entity */
    private static Map<String, Class<?>> m_objectPrivilegeMap = new ConcurrentHashMap<String, Class<?>>();

    /** ????? */
    private static Class<?> getObjectPrivilegeClass(String objectType) {
        // ???
        createObjectPrivilegeMap();
        return m_objectPrivilegeMap.get(objectType);
    }

    /** ??? */
    private static void createObjectPrivilegeMap() {
        if (m_objectPrivilegeMap == null || m_objectPrivilegeMap.size() == 0) {
            EntityManagerFactory emf = new JpaTransactionManager().getEntityManager().getEntityManagerFactory();
            Set<EntityType<?>> entityTypes = emf.getMetamodel().getEntities();
            String str = "";
            for (EntityType<?> entityType : entityTypes) {
                Class<?> clazz = entityType.getBindableJavaType();
                if (ObjectPrivilegeTargetInfo.class.isAssignableFrom(clazz)) {
                    try {
                        HinemosObjectPrivilege hinemosObjectPrivilege = clazz
                                .getAnnotation(HinemosObjectPrivilege.class);
                        String objectType = hinemosObjectPrivilege.objectType();
                        if (hinemosObjectPrivilege.isModifyCheck()) {
                            str += "[" + objectType + "," + clazz + "] ";

                            if (m_objectPrivilegeMap.get(objectType) != null) {
                                m_log.warn("duplicate objectType=" + objectType);
                            }
                            m_objectPrivilegeMap.put(objectType, clazz);
                        }
                    } catch (Exception e) {
                        continue;
                    }
                }
            }
            m_log.info("objectMap=" + str);
        }
    }

    /** ????? */
    public static Object getObjectPrivilegeObject(String objectType, String objectId, ObjectPrivilegeMode mode)
            throws JobMasterNotFound, ObjectPrivilege_InvalidRole {
        HinemosEntityManager em = new JpaTransactionManager().getEntityManager();
        Class<?> objectPrivilegeClass = getObjectPrivilegeClass(objectType);
        m_log.debug("class=" + objectPrivilegeClass + ", objectType=" + objectType + ", objectId=" + objectId);
        if (HinemosModuleConstant.JOB.equals(objectType)) {
            // JobMstEntity????objectId!= PK
            return em.find(objectPrivilegeClass, new JobMstEntityPK(objectId, objectId), mode);
        } else {
            return em.find(objectPrivilegeClass, objectId, mode);
        }
    }

    /** ???????? */
    public static List<ObjectPrivilegeInfo> getObjectPrivilegeEntities(Class<?> objectPrivilegeClass,
            String objectId, String roleId) {
        HinemosEntityManager em = new JpaTransactionManager().getEntityManager();
        HinemosObjectPrivilege hinemosObjectPrivilege = objectPrivilegeClass
                .getAnnotation(HinemosObjectPrivilege.class);
        if (hinemosObjectPrivilege == null) {
            // HinemosObjectPrivilege????????null?
            return null;
        }
        String objectType = hinemosObjectPrivilege.objectType();
        return em.createNamedQuery("ObjectPrivilegeInfo.findByObjectIdTypeRoleId", ObjectPrivilegeInfo.class)
                .setParameter("objectType", objectType).setParameter("objectId", objectId)
                .setParameter("roleId", roleId).getResultList();
    }

    /** ??????? */
    public static void deleteObjectPrivilege(String objectType, String objectId) {
        HinemosEntityManager em = new JpaTransactionManager().getEntityManager();
        em.createNamedQuery("ObjectPrivilegeInfo.deleteByObjectTypeObjectId").setParameter("objectType", objectType)
                .setParameter("objectId", objectId).executeUpdate();
    }
}