/**
* Copyright (c) 2011-2012 Optimax Software Ltd.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* * Neither the name of Optimax Software, ElasticInbox, nor the names
* of its contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.elasticinbox.core.cassandra;
import static me.prettyprint.hector.api.factory.HFactory.createMutator;
import java.io.IOException;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import me.prettyprint.cassandra.serializers.StringSerializer;
import me.prettyprint.hector.api.Keyspace;
import me.prettyprint.hector.api.mutation.Mutator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.elasticinbox.config.Configurator;
import com.elasticinbox.core.AccountDAO;
import com.elasticinbox.core.MessageDAO;
import com.elasticinbox.core.blob.store.BlobStorage;
import com.elasticinbox.core.blob.store.BlobStorageMediator;
import com.elasticinbox.core.cassandra.persistence.AccountPersistence;
import com.elasticinbox.core.cassandra.persistence.LabelCounterPersistence;
import com.elasticinbox.core.cassandra.persistence.LabelIndexPersistence;
import com.elasticinbox.core.cassandra.persistence.MessagePersistence;
import com.elasticinbox.core.cassandra.utils.BatchConstants;
import com.elasticinbox.core.model.Mailbox;
import com.elasticinbox.core.model.Message;
import com.elasticinbox.core.model.ReservedLabels;
public final class CassandraAccountDAO implements AccountDAO
{
private final Keyspace keyspace;
private final static StringSerializer strSe = StringSerializer.get();
private final BlobStorage blobStorage;
@SuppressWarnings("unused")
private final static Logger logger = LoggerFactory
.getLogger(CassandraAccountDAO.class);
public CassandraAccountDAO(Keyspace keyspace) {
this.keyspace = keyspace;
this.blobStorage = new BlobStorageMediator(null, null);
}
@Override
public void add(final Mailbox mailbox) throws IOException, IllegalArgumentException
{
// currently nothing happens here
//Map<String, Object> attributes = new HashMap<String, Object>();
// TODO: make quota configurable on the mailbox level
//attributes.put("quota:bytes", DEFAULT_QUOTA_BYTES);
//attributes.put("quota:messages", DEFAULT_QUOTA_MESSAGES);
//attributes.put("version", CURRENT_MAILBOX_VERSION);
//AccountPersistence.set(mailbox.getId(), attributes);
}
@Override
public void delete(final Mailbox mailbox) throws IOException
{
// purge all previously deleted objects
// TODO: we should not instantiate here
MessageDAO messageDAO = new CassandraMessageDAO(keyspace);
messageDAO.purge(mailbox, new Date());
// delete all objects from object store
try {
List<UUID> messageIds = null;
UUID start = null;
// loop until we delete all items
do {
// get all message ids
messageIds = LabelIndexPersistence.get(mailbox.getId(),
ReservedLabels.ALL_MAILS.getId(), start,
BatchConstants.BATCH_READS, true);
// get all message headers
Map<UUID, Message> messages =
MessagePersistence.fetch(mailbox.getId(), messageIds, false);
// delete message sources from object store
for(UUID messageId : messages.keySet()) {
blobStorage.delete(messages.get(messageId).getLocation());
}
// set start element for the next loop
start = messageIds.isEmpty() ? null : messageIds.get(messageIds.size()-1);
}
while (messageIds.size() >= BatchConstants.BATCH_READS);
} catch (Exception e) {
throw new IOException(e);
}
// begin batch operation
Mutator<String> m = createMutator(keyspace, strSe);
// delete all MessageMetadata
MessagePersistence.deleteAllMessages(m, mailbox.getId());
// delete all indexes from IndexLabel
LabelIndexPersistence.deleteIndexes(m, mailbox.getId());
// delete all counters
LabelCounterPersistence.deleteAll(m, mailbox.getId());
// delete Account data
AccountPersistence.delete(m, mailbox.getId());
// commit batch operation
m.execute();
}
/**
* Get quota (maximum) bytes for the given mailbox
*
* @param mailbox
* @return
*/
public Long getQuotaBytes(final Mailbox mailbox) {
// TODO: add account quota attribute check
return Configurator.getDefaultQuotaBytes();
}
/**
* Get quota (maximum) messages for the given mailbox
*
* @param mailbox
* @return
*/
public Long getQuotaCount(final Mailbox mailbox) {
// TODO: add account quota attribute check
return Configurator.getDefaultQuotaCount();
}
}