com.hmsonline.cassandra.triggers.LogEntry Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hms-cassandra-triggers Show documentation
Cassandra Triggers is a lightweight mechanism to implement trigger-like functionality for Cassandra.
There is a newer version: 1.0.1
Show newest version
package com.hmsonline.cassandra.triggers;

import java.nio.ByteBuffer;
import java.nio.charset.CharacterCodingException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import net.sf.json.JSONObject;
import net.sf.json.JSONSerializer;

import org.apache.cassandra.db.ColumnFamily;
import org.apache.cassandra.db.IColumn;
import org.apache.cassandra.thrift.ConsistencyLevel;
import org.apache.cassandra.utils.ByteBufferUtil;
import org.apache.cassandra.utils.UUIDGen;

public class LogEntry {
  
    private String commitLogRowKey = null;
    private String keyspace = null;
    private String columnFamily = null;
    private ConsistencyLevel consistencyLevel = null;
    private List operations = new ArrayList();
    private LogEntryStatus status = null;
    private ByteBuffer rowKey = null;
    private String uuid = null;
    private Map errors = new HashMap();
    private String host = null;
    public String getHost() {
        return host;
    }

    public void setHost(String host) {
        this.host = host;
    }

    private long timestamp = -1;

    public LogEntry() {
        this.setConsistencyLevel(ConsistencyLevel.ALL);        
    }

    public LogEntry(String keyspace, ColumnFamily columnFamily, ByteBuffer rowKey, ConsistencyLevel consistencyLevel,
            String host, long timestamp)
            throws Throwable {
        this.columnFamily = columnFamily.metadata().cfName;
        this.keyspace = keyspace;
        this.rowKey = rowKey;
        for (IColumn column : columnFamily.getSortedColumns()) {
            ColumnOperation operation = new ColumnOperation();
            operation.setName(column.name());
            operation.setDelete(columnFamily.isMarkedForDelete());
            operations.add(operation);
        }
        this.uuid = UUIDGen.getUUID(ByteBuffer.wrap(UUIDGen.getTimeUUIDBytes())).toString();
        this.status = LogEntryStatus.PREPARING;
        this.consistencyLevel = consistencyLevel;
        this.timestamp = timestamp;
        this.host = host;
    }

    public long getTimestamp() {
        return timestamp;
    }

    public void setTimestamp(long timestamp) {
        this.timestamp = timestamp;
    }

    public String getKeyspace() {
        return keyspace;
    }

    public void setKeyspace(String keyspace) {
        this.keyspace = keyspace;
    }

    public String getColumnFamily() {
        return columnFamily;
    }

    public void setColumnFamily(String columnFamily) {
        this.columnFamily = columnFamily;
    }

    public List getOperations() {
        return operations;
    }

    public void setOperations(List operations) {
        this.operations = operations;
    }

    public void addOperation(ColumnOperation operation) {
        operations.add(operation);
    }

    public LogEntryStatus getStatus() {
        return status;
    }

    public void setStatus(LogEntryStatus status) {
        this.status = status;
    }

    public ByteBuffer getRowKey() {
        return rowKey;
    }

    public void setRowKey(ByteBuffer rowKey) {
        this.rowKey = rowKey;
    }

    public String getUuid() {
        return uuid;
    }

    public void setUuid(String uuid) {
        this.uuid = uuid;
    }

    public ConsistencyLevel getConsistencyLevel() {
        return consistencyLevel;
    }

    public void setConsistencyLevel(ConsistencyLevel consistencyLevel) {
        this.consistencyLevel = consistencyLevel;
    }

    public Map getErrors() {
      return errors;
    }

    public void setErrors(Map errors) {
      this.errors = errors;
    }
    public boolean hasErrors(){
        return (this.getErrors() != null && this.getErrors().size() > 0);
    }
    
    public Map toMap() throws CharacterCodingException {
      HashMap result = new HashMap();
      result.put(LogEntryColumns.ROW.toString(), ByteBufferUtil.string(this.rowKey));
      result.put(LogEntryColumns.HOST.toString(), this.host);
      result.put(LogEntryColumns.KS.toString(), this.keyspace);
      result.put(LogEntryColumns.CF.toString(), this.columnFamily);
      result.put(LogEntryColumns.STATUS.toString(), "" + this.status);
      result.put(LogEntryColumns.TIMESTAMP.toString(), "" + this.timestamp);
      if(this.errors != null) {
        result.putAll(this.errors);
      }
      return result;
    }
    
    public static LogEntry fromJson(String json) throws CharacterCodingException {
      if(json == null || "".equals(json.trim())) {
        return null;
      }
      LogEntry result = new LogEntry();
      JSONObject jsonObj = (JSONObject) JSONSerializer.toJSON(json);
      result.setRowKey(ByteBufferUtil.bytes(jsonObj.getString(LogEntryColumns.ROW.toString())));
      result.setKeyspace(jsonObj.getString(LogEntryColumns.KS.toString()));
      result.setColumnFamily(jsonObj.getString(LogEntryColumns.CF.toString()));
      result.setHost(jsonObj.getString(LogEntryColumns.HOST.toString()));
      result.setStatus(LogEntryStatus.valueOf((jsonObj.getString(LogEntryColumns.STATUS.toString()))));
      result.setTimestamp(Long.parseLong((jsonObj.getString(LogEntryColumns.TIMESTAMP.toString()))));
      return result;
    }

    /**
     * @return the commitLogRowKey
     */
    public String getCommitLogRowKey() {
      return commitLogRowKey;
    }

    /**
     * @param commitLogRowKey the commitLogRowKey to set
     */
    public void setCommitLogRowKey(String commitLogRowKey) {
      this.commitLogRowKey = commitLogRowKey;
    }
}