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 package org.apache.activemq.store.kahadaptor; 018 019 import java.io.DataInput; 020 import java.io.DataOutput; 021 import java.io.IOException; 022 import java.util.ArrayList; 023 import java.util.List; 024 import org.apache.activemq.kaha.Marshaller; 025 import org.apache.activemq.kaha.impl.async.Location; 026 import org.apache.activemq.store.amq.AMQTx; 027 import org.apache.activemq.store.amq.AMQTxOperation; 028 import org.apache.activemq.wireformat.WireFormat; 029 030 /** 031 * Marshall an AMQTx 032 * 033 * @version $Revision: 1.10 $ 034 */ 035 public class AMQTxMarshaller implements Marshaller<AMQTx> { 036 037 private WireFormat wireFormat; 038 039 public AMQTxMarshaller(WireFormat wireFormat) { 040 this.wireFormat = wireFormat; 041 } 042 043 public AMQTx readPayload(DataInput dataIn) throws IOException { 044 Location location = new Location(); 045 location.readExternal(dataIn); 046 AMQTx result = new AMQTx(location); 047 int size = dataIn.readInt(); 048 for (int i = 0; i < size; i++) { 049 AMQTxOperation op = new AMQTxOperation(); 050 op.readExternal(wireFormat, dataIn); 051 result.getOperations().add(op); 052 } 053 return result; 054 } 055 056 public void writePayload(AMQTx amqtx, DataOutput dataOut) throws IOException { 057 amqtx.getLocation().writeExternal(dataOut); 058 List<AMQTxOperation> list = amqtx.getOperations(); 059 List<AMQTxOperation> ops = new ArrayList<AMQTxOperation>(); 060 061 for (AMQTxOperation op : list) { 062 if (op.getOperationType() == op.ADD_OPERATION_TYPE) { 063 ops.add(op); 064 } 065 } 066 dataOut.writeInt(ops.size()); 067 for (AMQTxOperation op : ops) { 068 op.writeExternal(wireFormat, dataOut); 069 } 070 } 071 }