001    /**
002     * Licensed to the Apache Software Foundation (ASF) under one or more
003     * contributor license agreements.  See the NOTICE file distributed with
004     * this work for additional information regarding copyright ownership.
005     * The ASF licenses this file to You under the Apache License, Version 2.0
006     * (the "License"); you may not use this file except in compliance with
007     * the License.  You may obtain a copy of the License at
008     *
009     *      http://www.apache.org/licenses/LICENSE-2.0
010     *
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS,
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     * See the License for the specific language governing permissions and
015     * limitations under the License.
016     */
017    
018    package org.apache.activemq.store.amq;
019    
020    import java.util.ArrayList;
021    import java.util.Iterator;
022    import java.util.List;
023    
024    import org.apache.activemq.command.JournalTopicAck;
025    import org.apache.activemq.command.Message;
026    import org.apache.activemq.command.MessageAck;
027    import org.apache.activemq.kaha.impl.async.Location;
028    
029    /**
030     */
031    /**
032     * Operations
033     * 
034     * 
035     */
036    public class AMQTx {
037    
038        private final Location location;
039        private List<AMQTxOperation> operations = new ArrayList<AMQTxOperation>();
040    
041        public AMQTx(Location location) {
042            this.location = location;
043        }
044    
045        public void add(AMQMessageStore store, Message msg, Location location) {
046            operations.add(new AMQTxOperation(AMQTxOperation.ADD_OPERATION_TYPE, store.getDestination(), msg,
047                                              location));
048        }
049    
050        public void add(AMQMessageStore store, MessageAck ack) {
051            operations.add(new AMQTxOperation(AMQTxOperation.REMOVE_OPERATION_TYPE, store.getDestination(), ack,
052                                              null));
053        }
054    
055        public void add(AMQTopicMessageStore store, JournalTopicAck ack) {
056            operations.add(new AMQTxOperation(AMQTxOperation.ACK_OPERATION_TYPE, store.getDestination(), ack,
057                                              null));
058        }
059    
060        public Message[] getMessages() {
061            List<Object> list = new ArrayList<Object>();
062            for (Iterator<AMQTxOperation> iter = operations.iterator(); iter.hasNext();) {
063                AMQTxOperation op = iter.next();
064                if (op.getOperationType() == AMQTxOperation.ADD_OPERATION_TYPE) {
065                    list.add(op.getData());
066                }
067            }
068            Message rc[] = new Message[list.size()];
069            list.toArray(rc);
070            return rc;
071        }
072    
073        public MessageAck[] getAcks() {
074            List<Object> list = new ArrayList<Object>();
075            for (Iterator<AMQTxOperation> iter = operations.iterator(); iter.hasNext();) {
076                AMQTxOperation op = iter.next();
077                if (op.getOperationType() == AMQTxOperation.REMOVE_OPERATION_TYPE) {
078                    list.add(op.getData());
079                }
080            }
081            MessageAck rc[] = new MessageAck[list.size()];
082            list.toArray(rc);
083            return rc;
084        }
085    
086        /**
087         * @return the location
088         */
089        public Location getLocation() {
090            return this.location;
091        }
092    
093        public List<AMQTxOperation> getOperations() {
094            return operations;
095        }
096    
097        public void setOperations(List<AMQTxOperation> operations) {
098            this.operations = operations;
099        }
100    }