diff options
Diffstat (limited to 'java/systests/src/main/java/org/apache/qpid/server/store/QuotaMessageStore.java')
-rw-r--r-- | java/systests/src/main/java/org/apache/qpid/server/store/QuotaMessageStore.java | 180 |
1 files changed, 180 insertions, 0 deletions
diff --git a/java/systests/src/main/java/org/apache/qpid/server/store/QuotaMessageStore.java b/java/systests/src/main/java/org/apache/qpid/server/store/QuotaMessageStore.java new file mode 100644 index 0000000000..07965cfa95 --- /dev/null +++ b/java/systests/src/main/java/org/apache/qpid/server/store/QuotaMessageStore.java @@ -0,0 +1,180 @@ +/* + * + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you 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 org.apache.qpid.server.store; + +import java.util.concurrent.atomic.AtomicBoolean; +import java.util.concurrent.atomic.AtomicLong; + +import org.apache.commons.configuration.Configuration; +import org.apache.qpid.AMQStoreException; +import org.apache.qpid.server.message.EnqueableMessage; +import org.apache.qpid.server.message.MessageContentSource; + +public class QuotaMessageStore extends NullMessageStore +{ + private final AtomicLong _messageId = new AtomicLong(1); + private final AtomicBoolean _closed = new AtomicBoolean(false); + + private long _totalStoreSize;; + private boolean _limitBusted; + private long _persistentSizeLowThreshold; + private long _persistentSizeHighThreshold; + + private final StateManager _stateManager; + private final EventManager _eventManager = new EventManager(); + + public QuotaMessageStore() + { + _stateManager = new StateManager(_eventManager); + } + + @Override + public void configureConfigStore(String name, ConfigurationRecoveryHandler recoveryHandler, Configuration config) + throws Exception + { + _persistentSizeHighThreshold = config.getLong(MessageStoreConstants.OVERFULL_SIZE_PROPERTY, Long.MAX_VALUE); + _persistentSizeLowThreshold = config.getLong(MessageStoreConstants.UNDERFULL_SIZE_PROPERTY, + _persistentSizeHighThreshold); + if (_persistentSizeLowThreshold > _persistentSizeHighThreshold || _persistentSizeLowThreshold < 0l) + { + _persistentSizeLowThreshold = _persistentSizeHighThreshold; + } + _stateManager.attainState(State.INITIALISING); + } + + @Override + public void configureMessageStore(String name, MessageStoreRecoveryHandler recoveryHandler, + TransactionLogRecoveryHandler tlogRecoveryHandler, Configuration config) throws Exception + { + _stateManager.attainState(State.INITIALISED); + } + + @Override + public void activate() throws Exception + { + _stateManager.attainState(State.ACTIVATING); + _stateManager.attainState(State.ACTIVE); + } + + @SuppressWarnings("unchecked") + @Override + public StoredMessage<StorableMessageMetaData> addMessage(StorableMessageMetaData metaData) + { + final long id = _messageId.getAndIncrement(); + return new StoredMemoryMessage(id, metaData); + } + + @Override + public Transaction newTransaction() + { + return new Transaction() + { + private AtomicLong _storeSizeIncrease = new AtomicLong(); + + @Override + public StoreFuture commitTranAsync() throws AMQStoreException + { + QuotaMessageStore.this.storedSizeChange(_storeSizeIncrease.intValue()); + return StoreFuture.IMMEDIATE_FUTURE; + } + + @Override + public void enqueueMessage(TransactionLogResource queue, EnqueableMessage message) throws AMQStoreException + { + _storeSizeIncrease.addAndGet(((MessageContentSource)message).getSize()); + } + + @Override + public void dequeueMessage(TransactionLogResource queue, EnqueableMessage message) throws AMQStoreException + { + _storeSizeIncrease.addAndGet(-((MessageContentSource)message).getSize()); + } + + @Override + public void commitTran() throws AMQStoreException + { + QuotaMessageStore.this.storedSizeChange(_storeSizeIncrease.intValue()); + } + + @Override + public void abortTran() throws AMQStoreException + { + } + + @Override + public void removeXid(long format, byte[] globalId, byte[] branchId) + { + } + + @Override + public void recordXid(long format, byte[] globalId, byte[] branchId, Record[] enqueues, Record[] dequeues) + { + } + }; + } + + @Override + public boolean isPersistent() + { + return true; + } + + @Override + public void close() throws Exception + { + _stateManager.attainState(State.CLOSING); + _closed.getAndSet(true); + _stateManager.attainState(State.CLOSED); + } + + @Override + public void addEventListener(EventListener eventListener, Event... events) + { + _eventManager.addEventListener(eventListener, events); + } + + private void storedSizeChange(final int delta) + { + if(_persistentSizeHighThreshold > 0) + { + synchronized (this) + { + long newSize = _totalStoreSize += delta; + if(!_limitBusted && newSize > _persistentSizeHighThreshold) + { + _limitBusted = true; + _eventManager.notifyEvent(Event.PERSISTENT_MESSAGE_SIZE_OVERFULL); + } + else if(_limitBusted && newSize < _persistentSizeHighThreshold) + { + _limitBusted = false; + _eventManager.notifyEvent(Event.PERSISTENT_MESSAGE_SIZE_UNDERFULL); + } + } + } + } + + @Override + public String getStoreType() + { + return "QUOTA"; + } +} |