net.sf.jabb.seqtx.azure.SequentialTransactionWrapper Maven / Gradle / Ivy

Go to download
/**
 * 
 */
package net.sf.jabb.seqtx.azure;

import java.io.Serializable;
import java.time.Instant;
import java.util.Date;

import org.apache.commons.lang3.SerializationUtils;
import org.apache.commons.lang3.Validate;
import org.apache.commons.lang3.builder.ToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle;

import com.microsoft.azure.storage.table.DynamicTableEntity;
import com.microsoft.azure.storage.table.EntityProperty;

import net.sf.jabb.seqtx.SequentialTransactionState;
import net.sf.jabb.seqtx.SimpleSequentialTransaction;

/**
 * @author James Hu
 *
 */
public class SequentialTransactionWrapper {
	static public final int MAX_BINARY_LENGTH = 64*1024;

	
	protected DynamicTableEntity entity;
	protected SimpleSequentialTransaction transaction;
	protected SequentialTransactionWrapper previous;
	protected SequentialTransactionWrapper next;
	
	protected String seriesId;
	protected String previousTransactionId;
	protected String nextTransactionId;

	
	public SequentialTransactionWrapper(){
		
	}
	
	/**
	 * Create a new instance with a DynamicTableEntity.
	 * The transaction field of the newly created instance will be null, but the seriesId, previousTransactionId, and nextTransactionId fields will be initialized.
	 * @param entity	the entity from Azure table storage
	 */
	public SequentialTransactionWrapper(DynamicTableEntity entity){
		this.entity = entity;
		
		this.seriesId = entity.getPartitionKey();
		
		EntityProperty p = null;
		
		p = entity.getProperties().get("Previous");
		this.previousTransactionId = p == null ? null : p.getValueAsString();

		p = entity.getProperties().get("Next");
		this.nextTransactionId = p == null ? null : p.getValueAsString();
	}
	
	/**
	 * Create a new instance with a SimpleSequentialTransaction.
 	 * The entity field of the newly created instance will be null, so as other fields except the transaction field.
	 * @param transaction	the SimpleSequentialTransaction object
	 */
	public SequentialTransactionWrapper(SimpleSequentialTransaction transaction){
		this.transaction = transaction;
	}
	
	@Override
	public String toString(){
		return new ToStringBuilder(this, ToStringStyle.SHORT_PREFIX_STYLE)
			.append("transactionId", this.getEntityTransactionId())
			.append("processorId", this.getTransactionNotNull().getProcessorId())
			.append("state", this.getTransactionNotNull().getState())
			.append("startPosition", this.getTransactionNotNull().getStartPosition())
			.append("endPosition", this.getTransactionNotNull().getEndPosition())
			.append("previous", this.getEntity().getProperties().get("Previous").getValueAsString())
			.append("next", this.getEntity().getProperties().get("Next").getValueAsString())
			.toString();
	}
	
	public void updateFromEntity(){
		if (transaction == null){
			transaction = new SimpleSequentialTransaction();
		}
		
		EntityProperty p = null;
		
		p = entity.getProperties().get("Attempts");
		transaction.setAttempts(p == null ? 0 : p.getValueAsInteger());
		
		p = entity.getProperties().get("Detail");
		transaction.setDetail(p == null ? null : (Serializable)SerializationUtils.deserialize(p.getValueAsByteArray()));

		p = entity.getProperties().get("EndPosition");
		transaction.setEndPosition(p == null ? null : p.getValueAsString());

		p = entity.getProperties().get("FinishTime");
		transaction.setFinishTime(p == null ? null : p.getValueAsDate().toInstant());
		
		p = entity.getProperties().get("ProcessorId");
		transaction.setProcessorId(p == null ? null : p.getValueAsString());
		
		p = entity.getProperties().get("StartPosition");
		transaction.setStartPosition(p == null ? null : p.getValueAsString());
		
		p = entity.getProperties().get("StartTime");
		transaction.setStartTime(p == null ? null : p.getValueAsDate().toInstant());
		
		p = entity.getProperties().get("State");
		transaction.setState(p == null ? null : SequentialTransactionState.valueOf(p.getValueAsString()));
		
		p = entity.getProperties().get("Timeout");
		transaction.setTimeout(p == null ? null : p.getValueAsDate().toInstant());
		
		transaction.setTransactionId(entity.getRowKey());
		this.seriesId = entity.getPartitionKey();
		
		p = entity.getProperties().get("Previous");
		this.previousTransactionId = p == null ? null : p.getValueAsString();

		p = entity.getProperties().get("Next");
		this.nextTransactionId = p == null ? null : p.getValueAsString();

	}
	
	public void updateToEntity(){
		if (entity == null){
			entity = new DynamicTableEntity();
		}
		
		Instant i = null;
		SequentialTransactionState s = null;
		
		entity.getProperties().put("Attempts", new EntityProperty(transaction.getAttempts()));
		
		if (transaction.getDetail() != null){
			byte[] serializedDetail = SerializationUtils.serialize(transaction.getDetail());
			Validate.isTrue(serializedDetail.length <= MAX_BINARY_LENGTH, 
					"Serialized transaction detail must not exceed %d bytes, that's the limitation of Azure table storage."
					, MAX_BINARY_LENGTH);
			entity.getProperties().put("Detail", new EntityProperty(serializedDetail));
		}else{
			entity.getProperties().remove("Detail");
		}
		
		entity.getProperties().put("EndPosition", new EntityProperty(transaction.getEndPosition()));
		
		i = transaction.getFinishTime();
		entity.getProperties().put("FinishTime", new EntityProperty(i == null ? null : Date.from(i)));
		
		
		entity.getProperties().put("ProcessorId", new EntityProperty(transaction.getProcessorId()));
		
		entity.getProperties().put("StartPosition", new EntityProperty(transaction.getStartPosition()));
		
		i = transaction.getStartTime();
		entity.getProperties().put("StartTime", new EntityProperty(i == null ? null : Date.from(i)));
		
		s = transaction.getState();
		entity.getProperties().put("State", new EntityProperty(s == null ? null : s.name()));
		
		i = transaction.getTimeout();
		entity.getProperties().put("Timeout", new EntityProperty(i == null ? null : Date.from(i)));
		
		entity.setRowKey(transaction.getTransactionId());
		
		entity.setPartitionKey(seriesId);
		entity.getProperties().put("Previous", new EntityProperty(this.previousTransactionId));
		entity.getProperties().put("Next", new EntityProperty(this.nextTransactionId));
	}
	
	public SimpleSequentialTransaction getTransactionNotNull(){
		if (transaction == null){
			updateFromEntity();
		}
		return transaction;
	}
	
	/**
	 * Mark this as the first transaction, also changes the entity field.
	 */
	public void setFirstTransaction(){
		this.previousTransactionId = "";
		if (entity != null){
			entity.getProperties().put("Previous", new EntityProperty(""));
		}
	}
	public boolean isFirstTransaction(){
		return previousTransactionId == null || previousTransactionId.length() == 0;
	}
	
	/**
	 * Mark this as the last transaction, also changes the entity field.
	 */
	public void setLastTransaction(){
		this.nextTransactionId = "";
		if (entity != null){
			entity.getProperties().put("Next", new EntityProperty(""));
		}
	}
	public boolean isLastTransaction(){
		return nextTransactionId == null || nextTransactionId.length() == 0;
	}

	public String entityKeysToString(){
		return entity == null ? null : (entity.getPartitionKey() + "/" + entity.getRowKey());
	}
	
	public String getEntityTransactionId(){
		return entity == null ? null : entity.getPartitionKey();
	}

	public DynamicTableEntity getEntity() {
		return entity;
	}
	public void setEntity(DynamicTableEntity entity) {
		this.entity = entity;
	}
	public SequentialTransactionWrapper getPrevious() {
		return previous;
	}
	public void setPrevious(SequentialTransactionWrapper previous) {
		this.previous = previous;
	}
	public SequentialTransactionWrapper getNext() {
		return next;
	}
	public void setNext(SequentialTransactionWrapper next) {
		this.next = next;
	}

	public SimpleSequentialTransaction getTransaction() {
		return transaction;
	}

	public void setTransaction(SimpleSequentialTransaction transaction) {
		this.transaction = transaction;
	}

	public String getSeriesId() {
		return seriesId;
	}

	public void setSeriesId(String seriesId) {
		this.seriesId = seriesId;
	}

	public String getPreviousTransactionId() {
		return previousTransactionId;
	}

	public void setPreviousTransactionId(String previousTransactionId) {
		this.previousTransactionId = previousTransactionId;
	}

	public String getNextTransactionId() {
		return nextTransactionId;
	}

	public void setNextTransactionId(String nextTransactionId) {
		this.nextTransactionId = nextTransactionId;
	}
}