|
|
|
@ -27,11 +27,13 @@ import org.wso2.carbon.policy.mgt.common.PolicyManagementException;
|
|
|
|
|
import org.wso2.carbon.policy.mgt.core.cache.PolicyCacheManager;
|
|
|
|
|
import org.wso2.carbon.policy.mgt.core.mgt.PolicyManager;
|
|
|
|
|
import org.wso2.carbon.policy.mgt.core.mgt.impl.PolicyManagerImpl;
|
|
|
|
|
import org.wso2.carbon.policy.mgt.core.util.PolicyManagementConstants;
|
|
|
|
|
import org.wso2.carbon.policy.mgt.core.util.PolicyManagerUtil;
|
|
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
|
import javax.cache.Cache;
|
|
|
|
|
import java.util.HashMap;
|
|
|
|
|
import java.util.Iterator;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
import java.util.Set;
|
|
|
|
|
|
|
|
|
|
public class PolicyCacheManagerImpl implements PolicyCacheManager {
|
|
|
|
|
|
|
|
|
@ -41,6 +43,14 @@ public class PolicyCacheManagerImpl implements PolicyCacheManager {
|
|
|
|
|
|
|
|
|
|
private static PolicyCacheManagerImpl policyCacheManager;
|
|
|
|
|
|
|
|
|
|
// private static Cache<Integer, Policy> getPolicyCache() {
|
|
|
|
|
// return PolicyManagerUtil.getPolicyCache(PolicyManagementConstants.DM_CACHE);
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
private static Cache<Integer, List<Policy>> getPolicyListCache() {
|
|
|
|
|
return PolicyManagerUtil.getPolicyListCache(PolicyManagementConstants.DM_CACHE_LIST);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private PolicyCacheManagerImpl() {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -57,43 +67,87 @@ public class PolicyCacheManagerImpl implements PolicyCacheManager {
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void addAllPolicies(List<Policy> policies) {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
if (map.isEmpty()) {
|
|
|
|
|
for (Policy policy : policies) {
|
|
|
|
|
map.put(policy.getId(), policy);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
// HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// if (cache.isEmpty()) {
|
|
|
|
|
|
|
|
|
|
// for (Policy policy : policies) {
|
|
|
|
|
// cache.put(policy.getId(), policy);
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
lCache.put(1, policies);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void updateAllPolicies(List<Policy> policies) {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
map.clear();
|
|
|
|
|
if (map.isEmpty()) {
|
|
|
|
|
for (Policy policy : policies) {
|
|
|
|
|
map.put(policy.getId(), policy);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
//// HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// cache.removeAll();
|
|
|
|
|
//// map.clear();
|
|
|
|
|
//// if (map.isEmpty()) {
|
|
|
|
|
// for (Policy policy : policies) {
|
|
|
|
|
//// map.put(policy.getId(), policy);
|
|
|
|
|
// cache.put(policy.getId(), policy);
|
|
|
|
|
// }
|
|
|
|
|
//// }
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
lCache.removeAll();
|
|
|
|
|
lCache.put(1, policies);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public List<Policy> getAllPolicies() throws PolicyManagementException {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
if (map.isEmpty()) {
|
|
|
|
|
// // HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// Iterator iterator = cache.iterator();
|
|
|
|
|
//// iterator.hasNext()
|
|
|
|
|
// if (!iterator.hasNext()) {
|
|
|
|
|
// PolicyManager policyManager = new PolicyManagerImpl();
|
|
|
|
|
// this.addAllPolicies(policyManager.getPolicies());
|
|
|
|
|
// }
|
|
|
|
|
// if (log.isDebugEnabled()) {
|
|
|
|
|
// //log.debug("No of policies stored in the cache .. : " + map.size());
|
|
|
|
|
//
|
|
|
|
|
// //Set<Integer> keySet = map.keySet();
|
|
|
|
|
// Iterator iterator2 = cache.iterator();
|
|
|
|
|
// while (iterator2.hasNext()) {
|
|
|
|
|
// org.wso2.carbon.caching.impl.CacheEntry thisEntry = (org.wso2.carbon.caching.impl.CacheEntry)
|
|
|
|
|
// iterator2.next();
|
|
|
|
|
// log.debug("Policy id in maps .. : " + thisEntry.getKey() + " policy name : " + ((Policy) thisEntry
|
|
|
|
|
// .getValue()).
|
|
|
|
|
// getPolicyName() + " Activated : " + ((Policy) thisEntry.getValue()).isActive());
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// List<Policy> policies = new ArrayList<>();
|
|
|
|
|
// while (iterator.hasNext()){
|
|
|
|
|
// CacheEntry thisEntry = (CacheEntry) iterator.next();
|
|
|
|
|
// policies.add((Policy) thisEntry.getValue());
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// return policies;
|
|
|
|
|
//// return new ArrayList<>(map.values());
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
if (!lCache.containsKey(1)) {
|
|
|
|
|
PolicyManager policyManager = new PolicyManagerImpl();
|
|
|
|
|
this.addAllPolicies(policyManager.getPolicies());
|
|
|
|
|
}
|
|
|
|
|
if (log.isDebugEnabled()) {
|
|
|
|
|
log.debug("No of policies stored in the cache .. : " + map.size());
|
|
|
|
|
|
|
|
|
|
Set<Integer> keySet = map.keySet();
|
|
|
|
|
for (Integer x : keySet) {
|
|
|
|
|
log.debug("Policy id in maps .. : " + map.get(x).getId() + " policy name : " + map.get(x).
|
|
|
|
|
getPolicyName() + " Activated : " + map.get(x).isActive());
|
|
|
|
|
if (log.isDebugEnabled()) {
|
|
|
|
|
List<Policy> cachedPolicy = lCache.get(1);
|
|
|
|
|
for (Policy policy : cachedPolicy) {
|
|
|
|
|
log.debug("Policy id in cache .. : " + policy.getId() + " policy name : " + policy.
|
|
|
|
|
getPolicyName() + " Activated : " + policy.isActive());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return new ArrayList<>(map.values());
|
|
|
|
|
|
|
|
|
|
return lCache.get(1);
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
@ -105,59 +159,136 @@ public class PolicyCacheManagerImpl implements PolicyCacheManager {
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void removeAllPolicies() {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
map.clear();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// cache.removeAll();
|
|
|
|
|
//HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
//map.clear();
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
lCache.removeAll();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void addPolicy(Policy policy) {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
if (!map.containsKey(policy.getId())) {
|
|
|
|
|
map.put(policy.getId(), policy);
|
|
|
|
|
} else {
|
|
|
|
|
log.warn("Policy id (" + policy.getId() + ") already exist in the map. hence not attempted to store.");
|
|
|
|
|
//HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// if (!cache.containsKey(policy.getId())) {
|
|
|
|
|
// cache.put(policy.getId(), policy);
|
|
|
|
|
// } else {
|
|
|
|
|
// log.warn("Policy id (" + policy.getId() + ") already exist in the map. hence not attempted to store.");
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
if (lCache.containsKey(1)) {
|
|
|
|
|
List<Policy> cachedPolicy = lCache.get(1);
|
|
|
|
|
|
|
|
|
|
for (Policy pol : cachedPolicy) {
|
|
|
|
|
if (pol.getId() == policy.getId()) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
cachedPolicy.add(policy);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void updatePolicy(Policy policy) {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
if (map.containsKey(policy.getId())) {
|
|
|
|
|
map.remove(policy.getId());
|
|
|
|
|
map.put(policy.getId(), policy);
|
|
|
|
|
// HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// if (cache.containsKey(policy.getId())) {
|
|
|
|
|
// cache.remove(policy.getId());
|
|
|
|
|
// cache.put(policy.getId(), policy);
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
if (lCache.containsKey(1)) {
|
|
|
|
|
List<Policy> cachedPolicy = lCache.get(1);
|
|
|
|
|
Iterator iterator = cachedPolicy.iterator();
|
|
|
|
|
while (iterator.hasNext()) {
|
|
|
|
|
Policy pol = (Policy) iterator.next();
|
|
|
|
|
if (pol.getId() == policy.getId()) {
|
|
|
|
|
iterator.remove();
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
cachedPolicy.add(policy);
|
|
|
|
|
lCache.replace(1, cachedPolicy);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void updatePolicy(int policyId) throws PolicyManagementException {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
if (map.containsKey(policyId)) {
|
|
|
|
|
this.removePolicy(policyId);
|
|
|
|
|
}
|
|
|
|
|
// HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// if (cache.containsKey(policyId)) {
|
|
|
|
|
// this.removePolicy(policyId);
|
|
|
|
|
// }
|
|
|
|
|
// PolicyManager policyManager = new PolicyManagerImpl();
|
|
|
|
|
// Policy policy = policyManager.getPolicy(policyId);
|
|
|
|
|
// cache.put(policyId, policy);
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
if (lCache.containsKey(1)) {
|
|
|
|
|
PolicyManager policyManager = new PolicyManagerImpl();
|
|
|
|
|
Policy policy = policyManager.getPolicy(policyId);
|
|
|
|
|
map.put(policyId, policy);
|
|
|
|
|
this.updatePolicy(policy);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void removePolicy(int policyId) {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
if (map.containsKey(policyId)) {
|
|
|
|
|
map.remove(policyId);
|
|
|
|
|
} else {
|
|
|
|
|
log.warn("Policy id (" + policyId + ") does not exist in the cache. Hence not removed.");
|
|
|
|
|
// HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// if (cache.containsKey(policyId)) {
|
|
|
|
|
// cache.remove(policyId);
|
|
|
|
|
// } else {
|
|
|
|
|
// log.warn("Policy id (" + policyId + ") does not exist in the cache. Hence not removed.");
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
if (lCache.containsKey(1)) {
|
|
|
|
|
List<Policy> cachedPolicy = lCache.get(1);
|
|
|
|
|
Iterator iterator = cachedPolicy.iterator();
|
|
|
|
|
while (iterator.hasNext()) {
|
|
|
|
|
Policy pol = (Policy) iterator.next();
|
|
|
|
|
if (pol.getId() == policyId) {
|
|
|
|
|
iterator.remove();
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
lCache.replace(1, cachedPolicy);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public Policy getPolicy(int policyId) throws PolicyManagementException {
|
|
|
|
|
HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
if (!map.containsKey(policyId)) {
|
|
|
|
|
//HashMap<Integer, Policy> map = this.getTenantRelatedMap();
|
|
|
|
|
// Cache<Integer, Policy> cache = getPolicyCache();
|
|
|
|
|
// if (!cache.containsKey(policyId)) {
|
|
|
|
|
// this.removeAllPolicies();
|
|
|
|
|
// this.getAllPolicies();
|
|
|
|
|
// }
|
|
|
|
|
// return cache.get(policyId);
|
|
|
|
|
|
|
|
|
|
Cache<Integer, List<Policy>> lCache = getPolicyListCache();
|
|
|
|
|
if (!lCache.containsKey(1)) {
|
|
|
|
|
this.removeAllPolicies();
|
|
|
|
|
this.getAllPolicies();
|
|
|
|
|
}
|
|
|
|
|
return map.get(policyId);
|
|
|
|
|
|
|
|
|
|
Policy policy = null;
|
|
|
|
|
List<Policy> cachedPolicy = lCache.get(1);
|
|
|
|
|
Iterator iterator = cachedPolicy.iterator();
|
|
|
|
|
while (iterator.hasNext()) {
|
|
|
|
|
Policy pol = (Policy) iterator.next();
|
|
|
|
|
if (pol.getId() == policyId) {
|
|
|
|
|
policy = pol;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return policy;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
@ -175,10 +306,10 @@ public class PolicyCacheManagerImpl implements PolicyCacheManager {
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private HashMap<Integer, Policy> getTenantRelatedMap(){
|
|
|
|
|
private HashMap<Integer, Policy> getTenantRelatedMap() {
|
|
|
|
|
|
|
|
|
|
int tenantId = PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId();
|
|
|
|
|
if(!tenantedPolicyMap.containsKey(tenantId)){
|
|
|
|
|
if (!tenantedPolicyMap.containsKey(tenantId)) {
|
|
|
|
|
HashMap<Integer, Policy> policyMap = new HashMap<>();
|
|
|
|
|
tenantedPolicyMap.put(tenantId, policyMap);
|
|
|
|
|
}
|
|
|
|
|