CNSAttributesCassandraPersistence.java example

Explorer
cmb-master
/**
 * Copyright 2012 Comcast Corporation
 * 
 * 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.comcast.cns.persistence;

import java.util.HashMap;
import java.util.Map;

import org.apache.log4j.Logger;
import org.json.JSONObject;

import com.comcast.cmb.common.persistence.AbstractDurablePersistence;
import com.comcast.cmb.common.persistence.AbstractDurablePersistence.CMB_SERIALIZER;
import com.comcast.cmb.common.persistence.AbstractDurablePersistence.CmbColumnSlice;
import com.comcast.cmb.common.persistence.DurablePersistenceFactory;
import com.comcast.cmb.common.persistence.PersistenceFactory;
import com.comcast.cns.controller.CNSCache;
import com.comcast.cns.model.CNSSubscription;
import com.comcast.cns.model.CNSSubscriptionAttributes;
import com.comcast.cns.model.CNSSubscriptionDeliveryPolicy;
import com.comcast.cns.model.CNSTopicAttributes;
import com.comcast.cns.model.CNSTopicDeliveryPolicy;

/**
 * Provide Cassandra Persistence for topic attributes
 * @author bwolf, vvenkatraman, jorge, tina
 *
 */
public class CNSAttributesCassandraPersistence implements ICNSAttributesPersistence {
	
	private static final String columnFamilyTopicAttributes = "CNSTopicAttributes";
	private static final String columnFamilySubscriptionAttributes = "CNSSubscriptionAttributes";
	private static final String columnFamilyTopicStats = "CNSTopicStats";
	private static Logger logger = Logger.getLogger(CNSAttributesCassandraPersistence.class);
	private static final AbstractDurablePersistence cassandraHandler = DurablePersistenceFactory.getInstance();
	
	public CNSAttributesCassandraPersistence() {
	}

	@Override
	public void setTopicAttributes(CNSTopicAttributes topicAttributes, String topicArn) throws Exception {
		
		cassandraHandler.insertRow(AbstractDurablePersistence.CNS_KEYSPACE, topicArn, columnFamilyTopicAttributes, getColumnValues(topicAttributes), CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER, null);
		
		if (topicAttributes.getDisplayName() != null) {
			PersistenceFactory.getTopicPersistence().updateTopicDisplayName(topicArn, topicAttributes.getDisplayName());
		}
		
		CNSCache.removeTopicAttributes(topicArn);
	}

	private Map<String, String> getColumnValues(CNSTopicAttributes topicAttributes) {
		
		Map<String, String> columnVals = new HashMap<String, String>();
		
		if (topicAttributes.getUserId() != null) {
			columnVals.put("userId", topicAttributes.getUserId());
		}
		
		if (topicAttributes.getTopicArn() != null) {
			columnVals.put("topicArn", topicAttributes.getTopicArn());
		}
		
		if (topicAttributes.getSubscriptionsPending() > -1) {
			columnVals.put("subscriptionPending", topicAttributes.getSubscriptionsPending()+"");
		}
		
		if (topicAttributes.getSubscriptionsDeleted() > -1) {
			columnVals.put("subscriptionDeleted", topicAttributes.getSubscriptionsDeleted()+"");
		}
		
		if (topicAttributes.getSubscriptionsConfirmed() > -1) {
			columnVals.put("subscriptionConfirmed", topicAttributes.getSubscriptionsConfirmed()+"");
		}
		
		// currently only accept delivery policy parameter, fill up with defaults if elements are missing and return as both delivery policy and effective delivery policy
		
		/*if (topicAttributes.getEffectiveDeliveryPolicy() != null) {
			columnVals.put("effectiveDeliveryPolicy", topicAttributes.getEffectiveDeliveryPolicy().toString());
		}*/
		
		if (topicAttributes.getDeliveryPolicy() != null) {
			columnVals.put("deliveryPolicy", topicAttributes.getDeliveryPolicy().toString());
		}

		if (topicAttributes.getPolicy() != null) {
			columnVals.put("policy", topicAttributes.getPolicy().toString());
		}
		
		return columnVals;
	}

	@Override
	public CNSTopicAttributes getTopicAttributes(String topicArn) throws Exception {
		
		CNSTopicAttributes topicAttributes = new CNSTopicAttributes();
		topicAttributes.setTopicArn(topicArn);
		
		CmbColumnSlice<String, String> slice = cassandraHandler.readColumnSlice(AbstractDurablePersistence.CNS_KEYSPACE, columnFamilyTopicAttributes, topicArn, null, null, 10, CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER);
		
		if (slice != null) {
			
			if (slice.getColumnByName("policy") != null) {
				topicAttributes.setPolicy(slice.getColumnByName("policy").getValue());
			}
			
			CNSTopicDeliveryPolicy deliveryPolicy = null;
			
			if (slice.getColumnByName("deliveryPolicy") != null) {
				deliveryPolicy = new CNSTopicDeliveryPolicy(new JSONObject(slice.getColumnByName("deliveryPolicy").getValue()));
			} else {
				deliveryPolicy = new CNSTopicDeliveryPolicy();
			}
			
			topicAttributes.setEffectiveDeliveryPolicy(deliveryPolicy);		
			topicAttributes.setDeliveryPolicy(deliveryPolicy);

			if (slice.getColumnByName("userId") != null) {
				topicAttributes.setUserId(slice.getColumnByName("userId").getValue());
			}
			
			topicAttributes.setDisplayName(PersistenceFactory.getTopicPersistence().getTopic(topicArn).getDisplayName());
		}
		
		long subscriptionConfirmedCount = cassandraHandler.getCounter(AbstractDurablePersistence.CNS_KEYSPACE, columnFamilyTopicStats, topicArn, "subscriptionConfirmed", CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER);
		topicAttributes.setSubscriptionsConfirmed(subscriptionConfirmedCount);
		
		long subscriptionPendingCount = cassandraHandler.getCounter(AbstractDurablePersistence.CNS_KEYSPACE, columnFamilyTopicStats, topicArn, "subscriptionPending", CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER);
		topicAttributes.setSubscriptionsPending(subscriptionPendingCount);
		
		long subscriptionDeletedCount = cassandraHandler.getCounter(AbstractDurablePersistence.CNS_KEYSPACE, columnFamilyTopicStats, topicArn, "subscriptionDeleted", CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER);
		topicAttributes.setSubscriptionsDeleted(subscriptionDeletedCount);

		return topicAttributes;
	}

	@Override
	public void setSubscriptionAttributes(CNSSubscriptionAttributes subscriptionAtributes, String subscriptionArn) throws Exception {

		cassandraHandler.insertRow(AbstractDurablePersistence.CNS_KEYSPACE, subscriptionArn, columnFamilySubscriptionAttributes, getColumnValues(subscriptionAtributes), CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER, null);
		String topicArn = com.comcast.cns.util.Util.getCnsTopicArn(subscriptionArn);
		CNSCache.removeTopicAttributes(topicArn);
	}

	private Map<String, String> getColumnValues(CNSSubscriptionAttributes subscriptionAttributes) {
		
		Map<String, String> colVals = new HashMap<String, String>();
		
		if (subscriptionAttributes.getDeliveryPolicy() != null) {
			colVals.put("deliveryPolicy", subscriptionAttributes.getDeliveryPolicy().toString());
		}
		
		
		// currently only accept delivery policy parameter, fill up with defaults if elements are missing and return as both delivery policy and effective delivery policy
		
		/*if (subscriptionAtributes.getEffectiveDeliveryPolicy() != null) {
			colVals.put("effectiveDeliveryPolicy", subscriptionAtributes.getEffectiveDeliveryPolicy().toString());
		}*/
		
		if (subscriptionAttributes.getSubscriptionArn() != null) {
			colVals.put("subscriptionArn", subscriptionAttributes.getSubscriptionArn());
		}
		
		if (subscriptionAttributes.getTopicArn() != null) {
			colVals.put("topicArn", subscriptionAttributes.getTopicArn());
		}
		
		if (subscriptionAttributes.getUserId() != null) {
			colVals.put("userId", subscriptionAttributes.getUserId());
		}
		
		return colVals;
	}

	@Override
	public CNSSubscriptionAttributes getSubscriptionAttributes(String subscriptionArn) throws Exception {
		
		CNSSubscriptionAttributes subscriptionAttributes = null;
		CmbColumnSlice<String, String> slice = cassandraHandler.readColumnSlice(AbstractDurablePersistence.CNS_KEYSPACE, columnFamilySubscriptionAttributes, subscriptionArn, null, null, 10, CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER, CMB_SERIALIZER.STRING_SERIALIZER);
		
		if (slice != null) {
			
			subscriptionAttributes = new CNSSubscriptionAttributes();
			
			if (slice.getColumnByName("confirmationWasAuthenticated") != null) {
				subscriptionAttributes.setConfirmationWasAuthenticated(Boolean.getBoolean(slice.getColumnByName("confirmationWasAuthenticated").getValue()));
			}
			
			if (slice.getColumnByName("deliveryPolicy") != null) {
				subscriptionAttributes.setDeliveryPolicy(new CNSSubscriptionDeliveryPolicy(new JSONObject(slice.getColumnByName("deliveryPolicy").getValue())));
			}
			
			// if "ignore subscription override" is checked, get effective delivery policy from topic delivery policy, otherwise 
			// get effective delivery policy from subscription delivery policy
			
			CNSSubscription subscription = PersistenceFactory.getSubscriptionPersistence().getSubscription(subscriptionArn);
			
			if (subscription == null) {
				throw new SubscriberNotFoundException("Subscription not found. arn=" + subscriptionArn);
			}
			
			CNSTopicAttributes topicAttributes = getTopicAttributes(subscription.getTopicArn());
			
			if (topicAttributes != null) {
				
				CNSTopicDeliveryPolicy topicEffectiveDeliveryPolicy = topicAttributes.getEffectiveDeliveryPolicy();
				
				if (topicEffectiveDeliveryPolicy != null) {
					
					if (topicEffectiveDeliveryPolicy.isDisableSubscriptionOverrides() || subscriptionAttributes.getDeliveryPolicy() == null) {
						CNSSubscriptionDeliveryPolicy effectiveDeliveryPolicy = new CNSSubscriptionDeliveryPolicy();
						effectiveDeliveryPolicy.setHealthyRetryPolicy(topicEffectiveDeliveryPolicy.getDefaultHealthyRetryPolicy());
						effectiveDeliveryPolicy.setSicklyRetryPolicy(topicEffectiveDeliveryPolicy.getDefaultSicklyRetryPolicy());
						effectiveDeliveryPolicy.setThrottlePolicy(topicEffectiveDeliveryPolicy.getDefaultThrottlePolicy());
						subscriptionAttributes.setEffectiveDeliveryPolicy(effectiveDeliveryPolicy);
					} else {
						subscriptionAttributes.setEffectiveDeliveryPolicy(subscriptionAttributes.getDeliveryPolicy());
					}
				}
			}
			
			if (slice.getColumnByName("topicArn") != null) {
				subscriptionAttributes.setTopicArn(slice.getColumnByName("topicArn").getValue());
			}

			if (slice.getColumnByName("userId") != null) {
				subscriptionAttributes.setUserId(slice.getColumnByName("userId").getValue());
			}
			
			subscriptionAttributes.setSubscriptionArn(subscriptionArn);
		}
		
		return subscriptionAttributes;
	}
}