| /** |
| * |
| * 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.activemq.nms.openwire.tool.marshalers; |
| |
| import java.io.File; |
| import java.io.FileWriter; |
| import java.io.PrintWriter; |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.Comparator; |
| import java.util.Iterator; |
| import java.util.List; |
| |
| import org.apache.activemq.openwire.tool.MultiSourceGenerator; |
| import org.codehaus.jam.JAnnotation; |
| import org.codehaus.jam.JAnnotationValue; |
| import org.codehaus.jam.JClass; |
| import org.codehaus.jam.JProperty; |
| |
| /** |
| * |
| * @version $Revision: 381410 $ |
| */ |
| public class AmqCSharpMarshallersGenerator extends MultiSourceGenerator { |
| |
| protected String targetDir="./src/main"; |
| protected List<JClass> concreteClasses = new ArrayList<JClass>(); |
| protected File factoryFile; |
| protected String factoryFileName = "MarshallerFactory"; |
| protected String indent = " "; |
| |
| /** |
| * Overrides the base class init since we don't mark any marshaller classes as |
| * being manually maintained. |
| */ |
| protected void initialiseManuallyMaintainedClasses() { |
| } |
| |
| /** |
| * Returns all the valid properties available on the current class. Overrides the |
| * method in {@link MultiSourceGenerator} to add filtering on the Openwire Version |
| * number so that we can rerun this tool for older versions and produce an exact |
| * match to what was previously generated. |
| * |
| * @return List of Properties valid for the current {@link JClass} and Openwire version. |
| */ |
| public List<JProperty> getProperties() { |
| List<JProperty> answer = new ArrayList<JProperty>(); |
| JProperty[] properties = jclass.getDeclaredProperties(); |
| for (int i = 0; i < properties.length; i++) { |
| JProperty property = properties[i]; |
| if (isValidProperty(property)) { |
| |
| JAnnotation annotation = property.getAnnotation("openwire:property"); |
| JAnnotationValue version = annotation.getValue("version"); |
| |
| if( version.asInt() <= this.getOpenwireVersion() ) { |
| answer.add(property); |
| } |
| } |
| } |
| return answer; |
| } |
| |
| public Object run() { |
| filePostFix = getFilePostFix(); |
| if (destDir == null) { |
| destDir = new File(targetDir+"/OpenWire/V"+getOpenwireVersion()); |
| } |
| Object answer = super.run(); |
| processFactory(); |
| return answer; |
| } |
| |
| protected void processClass(JClass jclass) { |
| super.processClass(jclass); |
| |
| if (!jclass.isAbstract()) { |
| concreteClasses.add(jclass); |
| } |
| } |
| |
| protected String getClassName(JClass jclass) { |
| return super.getClassName(jclass) + "Marshaller"; |
| } |
| |
| protected String getBaseClassName(JClass jclass) { |
| String answer = "BaseDataStreamMarshaller"; |
| JClass superclass = jclass.getSuperclass(); |
| if (superclass != null) { |
| String superName = superclass.getSimpleName(); |
| if (!superName.equals("Object") && !superName.equals("JNDIBaseStorable") && !superName.equals("DataStructureSupport")) { |
| answer = superName + "Marshaller"; |
| } |
| } |
| return answer; |
| } |
| |
| protected String getFilePostFix() { |
| return ".cs"; |
| } |
| |
| protected void generateLicence(PrintWriter out) { |
| out.println("/*"); |
| out.println(" * Licensed to the Apache Software Foundation (ASF) under one or more"); |
| out.println(" * contributor license agreements. See the NOTICE file distributed with"); |
| out.println(" * this work for additional information regarding copyright ownership."); |
| out.println(" * The ASF licenses this file to You under the Apache License, Version 2.0"); |
| out.println(" * (the \"License\"); you may not use this file except in compliance with"); |
| out.println(" * the License. You may obtain a copy of the License at"); |
| out.println(" *"); |
| out.println(" * http://www.apache.org/licenses/LICENSE-2.0"); |
| out.println(" *"); |
| out.println(" * Unless required by applicable law or agreed to in writing, software"); |
| out.println(" * distributed under the License is distributed on an \"AS IS\" BASIS,"); |
| out.println(" * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied."); |
| out.println(" * See the License for the specific language governing permissions and"); |
| out.println(" * limitations under the License."); |
| out.println(" */"); |
| } |
| |
| protected void generateFile(PrintWriter out) throws Exception { |
| generateLicence(out); |
| |
| out.println(""); |
| out.println("/*"); |
| out.println(" *"); |
| out.println(" * Marshaler code for OpenWire format for "+super.getClassName(jclass) ); |
| out.println(" *"); |
| out.println(" * NOTE!: This file is auto generated - do not modify!"); |
| out.println(" * if you need to make a change, please see the Java Classes"); |
| out.println(" * in the nms-activemq-openwire-generator module"); |
| out.println(" *"); |
| out.println(" */"); |
| out.println(""); |
| out.println("using System;"); |
| out.println("using System.IO;"); |
| out.println(""); |
| out.println("using Apache.NMS.ActiveMQ.Commands;"); |
| out.println(""); |
| out.println("namespace Apache.NMS.ActiveMQ.OpenWire.V"+getOpenwireVersion()); |
| out.println("{"); |
| out.println(" /// <summary>"); |
| out.println(" /// Marshalling code for Open Wire Format for "+super.getClassName(jclass)); |
| out.println(" /// </summary>"); |
| if( isAbstractClass() ) { |
| out.println(" abstract class "+getClassName()+" : "+getBaseClass()); |
| } else { |
| out.println(" class "+getClassName()+" : "+getBaseClass()); |
| } |
| out.println(" {"); |
| |
| if (!isAbstractClass()) { |
| out.println(" /// <summery>"); |
| out.println(" /// Creates an instance of the Object that this marshaller handles."); |
| out.println(" /// </summery>"); |
| out.println(" public override DataStructure CreateObject() "); |
| out.println(" {"); |
| out.println(" return new " + jclass.getSimpleName() + "();"); |
| out.println(" }"); |
| out.println(""); |
| out.println(" /// <summery>"); |
| out.println(" /// Returns the type code for the Object that this Marshaller handles.."); |
| out.println(" /// </summery>"); |
| out.println(" public override byte GetDataStructureType() "); |
| out.println(" {"); |
| out.println(" return " + jclass.getSimpleName() + ".ID_" + jclass.getSimpleName().toUpperCase() + ";"); |
| out.println(" }"); |
| } |
| |
| out.println(""); |
| out.println(" // "); |
| out.println(" // Un-marshal an object instance from the data input stream"); |
| out.println(" // "); |
| out.println(" public override void TightUnmarshal(OpenWireFormat wireFormat, Object o, BinaryReader dataIn, BooleanStream bs) "); |
| out.println(" {"); |
| out.println(" base.TightUnmarshal(wireFormat, o, dataIn, bs);"); |
| |
| if (!getProperties().isEmpty() || isMarshallerAware()) { |
| out.println(""); |
| out.println(" " + jclass.getSimpleName() + " info = (" + jclass.getSimpleName() + ")o;"); |
| } |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.BeforeUnmarshall(wireFormat);"); |
| out.println(""); |
| } |
| |
| generateTightUnmarshalBody(out); |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.AfterUnmarshall(wireFormat);"); |
| } |
| |
| out.println(" }"); |
| out.println(""); |
| out.println(" //"); |
| out.println(" // Write the booleans that this object uses to a BooleanStream"); |
| out.println(" //"); |
| out.println(" public override int TightMarshal1(OpenWireFormat wireFormat, Object o, BooleanStream bs)"); |
| out.println(" {"); |
| |
| if( checkNeedsInfoPointerTM1() ) { |
| out.println(" " + jclass.getSimpleName() + " info = (" + jclass.getSimpleName() + ")o;"); |
| } |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.BeforeMarshall(wireFormat);"); |
| } |
| |
| out.println(""); |
| out.println(" int rc = base.TightMarshal1(wireFormat, o, bs);"); |
| |
| int baseSize = generateTightMarshal1Body(out); |
| |
| out.println(""); |
| out.println(" return rc + " + baseSize + ";"); |
| out.println(" }"); |
| out.println(""); |
| out.println(" // "); |
| out.println(" // Write a object instance to data output stream"); |
| out.println(" //"); |
| out.println(" public override void TightMarshal2(OpenWireFormat wireFormat, Object o, BinaryWriter dataOut, BooleanStream bs)"); |
| out.println(" {"); |
| out.println(" base.TightMarshal2(wireFormat, o, dataOut, bs);"); |
| |
| if( checkNeedsInfoPointerTM2() ) { |
| out.println(""); |
| out.println(" " + jclass.getSimpleName() + " info = (" + jclass.getSimpleName() + ")o;"); |
| } |
| |
| generateTightMarshal2Body(out); |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.AfterMarshall(wireFormat);"); |
| } |
| |
| out.println(" }"); |
| out.println(""); |
| out.println(" // "); |
| out.println(" // Un-marshal an object instance from the data input stream"); |
| out.println(" // "); |
| out.println(" public override void LooseUnmarshal(OpenWireFormat wireFormat, Object o, BinaryReader dataIn) "); |
| out.println(" {"); |
| out.println(" base.LooseUnmarshal(wireFormat, o, dataIn);"); |
| |
| if (!getProperties().isEmpty() || isMarshallerAware()) { |
| out.println(""); |
| out.println(" " + jclass.getSimpleName() + " info = (" + jclass.getSimpleName() + ")o;"); |
| } |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.BeforeUnmarshall(wireFormat);"); |
| out.println(""); |
| } |
| |
| generateLooseUnmarshalBody(out); |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.AfterUnmarshall(wireFormat);"); |
| } |
| |
| out.println(" }"); |
| out.println(""); |
| out.println(" // "); |
| out.println(" // Write a object instance to data output stream"); |
| out.println(" //"); |
| out.println(" public override void LooseMarshal(OpenWireFormat wireFormat, Object o, BinaryWriter dataOut)"); |
| out.println(" {"); |
| |
| if (!getProperties().isEmpty() || isMarshallerAware()) { |
| out.println(""); |
| out.println(" " + jclass.getSimpleName() + " info = (" + jclass.getSimpleName() + ")o;"); |
| } |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.BeforeMarshall(wireFormat);"); |
| } |
| |
| out.println(""); |
| out.println(" base.LooseMarshal(wireFormat, o, dataOut);"); |
| |
| generateLooseMarshalBody(out); |
| |
| if (isMarshallerAware()) { |
| out.println(""); |
| out.println(" info.AfterMarshall(wireFormat);"); |
| } |
| |
| out.println(" }"); |
| out.println(" }"); |
| out.println("}"); |
| } |
| |
| protected void processFactory() { |
| if (factoryFile == null) { |
| factoryFile = new File(destDir, factoryFileName + filePostFix); |
| } |
| PrintWriter out = null; |
| try { |
| out = new PrintWriter(new FileWriter(factoryFile)); |
| generateFactory(out); |
| } catch (Exception e) { |
| throw new RuntimeException(e); |
| } finally { |
| if (out != null) { |
| out.close(); |
| } |
| } |
| } |
| |
| public void generateFactory(PrintWriter out) { |
| generateLicence(out); |
| out.println("/*"); |
| out.println(" *"); |
| out.println(" * MarshallerFactory code for OpenWire Protocol Version "+getOpenwireVersion() ); |
| out.println(" *"); |
| out.println(" * NOTE!: This file is auto generated - do not modify!"); |
| out.println(" * if you need to make a change, please see the Java Classes"); |
| out.println(" * in the nms-activemq-openwire-generator module"); |
| out.println(" *"); |
| out.println(" */"); |
| out.println(""); |
| out.println("namespace Apache.NMS.ActiveMQ.OpenWire.V" + getOpenwireVersion() + ""); |
| out.println("{"); |
| out.println(" /// <summary>"); |
| out.println(" /// Used to create marshallers for a specific version of the OpenWire protocol."); |
| out.println(" /// Each non-abstract DataStructure object has a registered Marshaller that is"); |
| out.println(" /// created and added to the OpenWireFormat objects format collection."); |
| out.println(" /// </summary>"); |
| out.println(" public class MarshallerFactory : IMarshallerFactory"); |
| out.println(" {"); |
| out.println(" /// <summery>"); |
| out.println(" /// Adds the Marshallers for this version of the OpenWire protocol to the "); |
| out.println(" /// Collection of Marshallers stored in the OpenWireFormat class."); |
| out.println(" /// </summery>"); |
| out.println(" public void configure(OpenWireFormat format) "); |
| out.println(" {"); |
| out.println(" format.clearMarshallers();"); |
| |
| List<JClass> list = new ArrayList<JClass>(getConcreteClasses()); |
| Collections.sort(list, new Comparator<JClass>() { |
| public int compare(JClass o1, JClass o2) { |
| return o1.getSimpleName().compareTo(o2.getSimpleName()); |
| } |
| }); |
| |
| for( JClass jclass : list ) { |
| out.println(" format.addMarshaller(new " + jclass.getSimpleName() + "Marshaller());"); |
| } |
| |
| out.println(" }"); |
| out.println(" }"); |
| out.println("}"); |
| } |
| |
| public List<JClass> getConcreteClasses() { |
| return concreteClasses; |
| } |
| |
| public void setConcreteClasses(List<JClass> concreteClasses) { |
| this.concreteClasses = concreteClasses; |
| } |
| |
| public String getTargetDir() { |
| return targetDir; |
| } |
| |
| public void setTargetDir(String targetDir) { |
| this.targetDir = targetDir; |
| } |
| |
| // //////////////////////////////////////////////////////////////////////////////////// |
| // This section is for the tight wire format encoding generator |
| // //////////////////////////////////////////////////////////////////////////////////// |
| |
| /** |
| * Checks if the tightMarshal1 method needs an casted version of its |
| * dataStructure argument and then returns true or false to indicate this |
| * to the caller. |
| * @returns true if the tightMarshal1 method needs an info pointer. |
| */ |
| protected boolean checkNeedsInfoPointerTM1() { |
| |
| if( isMarshallerAware() ){ |
| return true; |
| } |
| |
| for ( JProperty property : getProperties() ) { |
| JClass propertyType = property.getType(); |
| String type = propertyType.getSimpleName(); |
| |
| if( !( type.equals("byte") ) && |
| !( type.equals("char") ) && |
| !( type.equals("short") ) && |
| !( type.equals("int") ) ) { |
| |
| return true; |
| } |
| |
| } |
| |
| return false; |
| } |
| |
| /** |
| * Checks if the tightMarshal2 method needs an casted version of its |
| * dataStructure argument and then returns true or false to indicate this |
| * to the caller. |
| * @returns true if the tightMarshal2 method needs an info pointer. |
| */ |
| protected boolean checkNeedsInfoPointerTM2() { |
| |
| if( isMarshallerAware() ){ |
| return true; |
| } |
| |
| for ( JProperty property : getProperties() ) { |
| JClass propertyType = property.getType(); |
| String type = propertyType.getSimpleName(); |
| |
| if( !type.equals("boolean") ) { |
| |
| return true; |
| } |
| |
| } |
| |
| return false; |
| } |
| |
| protected void generateTightUnmarshalBody(PrintWriter out) { |
| for (JProperty property : getProperties()) { |
| JAnnotation annotation = property.getAnnotation("openwire:property"); |
| JAnnotationValue size = annotation.getValue("size"); |
| JClass propertyType = property.getType(); |
| String propertyTypeName = propertyType.getSimpleName(); |
| |
| if (propertyType.isArrayType() && !propertyTypeName.equals("byte[]")) { |
| generateTightUnmarshalBodyForArrayProperty(out, property, size); |
| } else { |
| generateTightUnmarshalBodyForProperty(out, property, size); |
| } |
| } |
| } |
| |
| protected void generateTightUnmarshalBodyForProperty(PrintWriter out, JProperty property, JAnnotationValue size) { |
| |
| String propertyName = property.getSimpleName(); |
| String type = property.getType().getSimpleName(); |
| |
| if (type.equals("boolean")) { |
| out.println(" info." + propertyName + " = bs.ReadBoolean();"); |
| } else if (type.equals("byte")) { |
| out.println(" info." + propertyName + " = dataIn.ReadByte();"); |
| } else if (type.equals("char")) { |
| out.println(" info." + propertyName + " = dataIn.ReadChar();"); |
| } else if (type.equals("short")) { |
| out.println(" info." + propertyName + " = dataIn.ReadInt16();"); |
| } else if (type.equals("int")) { |
| out.println(" info." + propertyName + " = dataIn.ReadInt32();"); |
| } else if (type.equals("long")) { |
| out.println(" info." + propertyName + " = TightUnmarshalLong(wireFormat, dataIn, bs);"); |
| } else if (type.equals("String")) { |
| out.println(" info." + propertyName + " = TightUnmarshalString(dataIn, bs);"); |
| } else if (type.equals("byte[]") || type.equals("ByteSequence")) { |
| if (size != null) { |
| out.println(" info." + propertyName + " = ReadBytes(dataIn, " + size.asInt() + ");"); |
| } else { |
| out.println(" info." + propertyName + " = ReadBytes(dataIn, bs.ReadBoolean());"); |
| } |
| } else if (isThrowable(property.getType())) { |
| out.println(" info." + propertyName + " = TightUnmarshalBrokerError(wireFormat, dataIn, bs);"); |
| } else if (isCachedProperty(property)) { |
| out.println(" info." + propertyName + " = (" + type + ") TightUnmarshalCachedObject(wireFormat, dataIn, bs);"); |
| } else { |
| out.println(" info." + propertyName + " = (" + type + ") TightUnmarshalNestedObject(wireFormat, dataIn, bs);"); |
| } |
| } |
| |
| protected void generateTightUnmarshalBodyForArrayProperty(PrintWriter out, JProperty property, JAnnotationValue size) { |
| JClass propertyType = property.getType(); |
| String arrayType = propertyType.getArrayComponentType().getSimpleName(); |
| String propertyName = property.getSimpleName(); |
| out.println(); |
| if (size != null) { |
| out.println(" {"); |
| out.println(" " + arrayType + "[] value = new " + arrayType + "[" + size.asInt() + "];"); |
| out.println(" " + "for( int i=0; i < " + size.asInt() + "; i++ ) {"); |
| out.println(" value[i] = (" + arrayType + ") TightUnmarshalNestedObject(wireFormat,dataIn, bs);"); |
| out.println(" }"); |
| out.println(" info." + propertyName + " = value;"); |
| out.println(" }"); |
| } else { |
| out.println(" if (bs.ReadBoolean()) {"); |
| out.println(" short size = dataIn.ReadInt16();"); |
| out.println(" " + arrayType + "[] value = new " + arrayType + "[size];"); |
| out.println(" for( int i=0; i < size; i++ ) {"); |
| out.println(" value[i] = (" + arrayType + ") TightUnmarshalNestedObject(wireFormat,dataIn, bs);"); |
| out.println(" }"); |
| out.println(" info." + propertyName + " = value;"); |
| out.println(" }"); |
| out.println(" else {"); |
| out.println(" info." + propertyName + " = null;"); |
| out.println(" }"); |
| } |
| } |
| |
| protected int generateTightMarshal1Body(PrintWriter out) { |
| int baseSize = 0; |
| for (JProperty property : getProperties()) { |
| JAnnotation annotation = property.getAnnotation("openwire:property"); |
| JAnnotationValue size = annotation.getValue("size"); |
| JClass propertyType = property.getType(); |
| String type = propertyType.getSimpleName(); |
| String getter = "info." + property.getSimpleName(); |
| |
| if (type.equals("boolean")) { |
| out.println(" bs.WriteBoolean(" + getter + ");"); |
| } else if (type.equals("byte")) { |
| baseSize += 1; |
| } else if (type.equals("char")) { |
| baseSize += 2; |
| } else if (type.equals("short")) { |
| baseSize += 2; |
| } else if (type.equals("int")) { |
| baseSize += 4; |
| } else if (type.equals("long")) { |
| out.println(" rc += TightMarshalLong1(wireFormat, " + getter + ", bs);"); |
| } else if (type.equals("String")) { |
| out.print(""); |
| out.println(" rc += TightMarshalString1(" + getter + ", bs);"); |
| } else if (type.equals("byte[]") || type.equals("ByteSequence")) { |
| if (size == null) { |
| out.println(" bs.WriteBoolean(" + getter + "!=null);"); |
| out.println(" rc += " + getter + "==null ? 0 : " + getter + ".Length+4;"); |
| } else { |
| baseSize += size.asInt(); |
| } |
| } else if (propertyType.isArrayType()) { |
| if (size != null) { |
| out.println(" rc += TightMarshalObjectArrayConstSize1(wireFormat, " + getter + ", bs, " + size.asInt() + ");"); |
| } else { |
| out.println(" rc += TightMarshalObjectArray1(wireFormat, " + getter + ", bs);"); |
| } |
| } else if (isThrowable(propertyType)) { |
| out.println(" rc += TightMarshalBrokerError1(wireFormat, " + getter + ", bs);"); |
| } else { |
| if (isCachedProperty(property)) { |
| out.println(" rc += TightMarshalCachedObject1(wireFormat, (DataStructure)" + getter + ", bs);"); |
| } else { |
| out.println(" rc += TightMarshalNestedObject1(wireFormat, (DataStructure)" + getter + ", bs);"); |
| } |
| } |
| } |
| return baseSize; |
| } |
| |
| protected void generateTightMarshal2Body(PrintWriter out) { |
| for (JProperty property : getProperties()) { |
| JAnnotation annotation = property.getAnnotation("openwire:property"); |
| JAnnotationValue size = annotation.getValue("size"); |
| JClass propertyType = property.getType(); |
| String type = propertyType.getSimpleName(); |
| String getter = "info." + property.getSimpleName(); |
| |
| if (type.equals("boolean")) { |
| out.println(" bs.ReadBoolean();"); |
| } else if (type.equals("byte")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("char")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("short")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("int")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("long")) { |
| out.println(" TightMarshalLong2(wireFormat, " + getter + ", dataOut, bs);"); |
| } else if (type.equals("String")) { |
| out.println(" TightMarshalString2(" + getter + ", dataOut, bs);"); |
| } else if (type.equals("byte[]") || type.equals("ByteSequence")) { |
| if (size != null) { |
| out.println(" dataOut.Write(" + getter + ", 0, " + size.asInt() + ");"); |
| } else { |
| out.println(" if(bs.ReadBoolean()) {"); |
| out.println(" dataOut.Write(" + getter + ".Length);"); |
| out.println(" dataOut.Write(" + getter + ");"); |
| out.println(" }"); |
| } |
| } else if (propertyType.isArrayType()) { |
| if (size != null) { |
| out.println(" TightMarshalObjectArrayConstSize2(wireFormat, " + getter + ", dataOut, bs, " + size.asInt() + ");"); |
| } else { |
| out.println(" TightMarshalObjectArray2(wireFormat, " + getter + ", dataOut, bs);"); |
| } |
| } else if (isThrowable(propertyType)) { |
| out.println(" TightMarshalBrokerError2(wireFormat, " + getter + ", dataOut, bs);"); |
| } else { |
| if (isCachedProperty(property)) { |
| out.println(" TightMarshalCachedObject2(wireFormat, (DataStructure)" + getter + ", dataOut, bs);"); |
| } else { |
| out.println(" TightMarshalNestedObject2(wireFormat, (DataStructure)" + getter + ", dataOut, bs);"); |
| } |
| } |
| } |
| } |
| |
| // //////////////////////////////////////////////////////////////////////////////////// |
| // This section is for the loose wire format encoding generator |
| // //////////////////////////////////////////////////////////////////////////////////// |
| |
| protected void generateLooseUnmarshalBody(PrintWriter out) { |
| for (JProperty property : getProperties()) { |
| JAnnotation annotation = property.getAnnotation("openwire:property"); |
| JAnnotationValue size = annotation.getValue("size"); |
| JClass propertyType = property.getType(); |
| String propertyTypeName = propertyType.getSimpleName(); |
| |
| if (propertyType.isArrayType() && !propertyTypeName.equals("byte[]")) { |
| generateLooseUnmarshalBodyForArrayProperty(out, property, size); |
| } else { |
| generateLooseUnmarshalBodyForProperty(out, property, size); |
| } |
| } |
| } |
| |
| protected void generateLooseUnmarshalBodyForProperty(PrintWriter out, JProperty property, JAnnotationValue size) { |
| |
| String propertyName = property.getSimpleName(); |
| String type = property.getType().getSimpleName(); |
| |
| if (type.equals("boolean")) { |
| out.println(" info." + propertyName + " = dataIn.ReadBoolean();"); |
| } else if (type.equals("byte")) { |
| out.println(" info." + propertyName + " = dataIn.ReadByte();"); |
| } else if (type.equals("char")) { |
| out.println(" info." + propertyName + " = dataIn.ReadChar();"); |
| } else if (type.equals("short")) { |
| out.println(" info." + propertyName + " = dataIn.ReadInt16();"); |
| } else if (type.equals("int")) { |
| out.println(" info." + propertyName + " = dataIn.ReadInt32();"); |
| } else if (type.equals("long")) { |
| out.println(" info." + propertyName + " = LooseUnmarshalLong(wireFormat, dataIn);"); |
| } else if (type.equals("String")) { |
| out.println(" info." + propertyName + " = LooseUnmarshalString(dataIn);"); |
| } else if (type.equals("byte[]") || type.equals("ByteSequence")) { |
| if (size != null) { |
| out.println(" info." + propertyName + " = ReadBytes(dataIn, " + size.asInt() + ");"); |
| } else { |
| out.println(" info." + propertyName + " = ReadBytes(dataIn, dataIn.ReadBoolean());"); |
| } |
| } else if (isThrowable(property.getType())) { |
| out.println(" info." + propertyName + " = LooseUnmarshalBrokerError(wireFormat, dataIn);"); |
| } else if (isCachedProperty(property)) { |
| out.println(" info." + propertyName + " = (" + type + ") LooseUnmarshalCachedObject(wireFormat, dataIn);"); |
| } else { |
| out.println(" info." + propertyName + " = (" + type + ") LooseUnmarshalNestedObject(wireFormat, dataIn);"); |
| } |
| } |
| |
| protected void generateLooseUnmarshalBodyForArrayProperty(PrintWriter out, JProperty property, JAnnotationValue size) { |
| JClass propertyType = property.getType(); |
| String arrayType = propertyType.getArrayComponentType().getSimpleName(); |
| String propertyName = property.getSimpleName(); |
| out.println(); |
| if (size != null) { |
| out.println(" {"); |
| out.println(" " + arrayType + "[] value = new " + arrayType + "[" + size.asInt() + "];"); |
| out.println(" " + "for( int i=0; i < " + size.asInt() + "; i++ ) {"); |
| out.println(" value[i] = (" + arrayType + ") LooseUnmarshalNestedObject(wireFormat,dataIn);"); |
| out.println(" }"); |
| out.println(" info." + propertyName + " = value;"); |
| out.println(" }"); |
| } else { |
| out.println(" if (dataIn.ReadBoolean()) {"); |
| out.println(" short size = dataIn.ReadInt16();"); |
| out.println(" " + arrayType + "[] value = new " + arrayType + "[size];"); |
| out.println(" for( int i=0; i < size; i++ ) {"); |
| out.println(" value[i] = (" + arrayType + ") LooseUnmarshalNestedObject(wireFormat,dataIn);"); |
| out.println(" }"); |
| out.println(" info." + propertyName + " = value;"); |
| out.println(" }"); |
| out.println(" else {"); |
| out.println(" info." + propertyName + " = null;"); |
| out.println(" }"); |
| } |
| } |
| |
| protected void generateLooseMarshalBody(PrintWriter out) { |
| for (JProperty property : getProperties()) { |
| JAnnotation annotation = property.getAnnotation("openwire:property"); |
| JAnnotationValue size = annotation.getValue("size"); |
| JClass propertyType = property.getType(); |
| String type = propertyType.getSimpleName(); |
| String getter = "info." + property.getSimpleName(); |
| |
| if (type.equals("boolean")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("byte")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("char")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("short")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("int")) { |
| out.println(" dataOut.Write(" + getter + ");"); |
| } else if (type.equals("long")) { |
| out.println(" LooseMarshalLong(wireFormat, " + getter + ", dataOut);"); |
| } else if (type.equals("String")) { |
| out.println(" LooseMarshalString(" + getter + ", dataOut);"); |
| } else if (type.equals("byte[]") || type.equals("ByteSequence")) { |
| if (size != null) { |
| out.println(" dataOut.Write(" + getter + ", 0, " + size.asInt() + ");"); |
| } else { |
| out.println(" dataOut.Write(" + getter + "!=null);"); |
| out.println(" if(" + getter + "!=null) {"); |
| out.println(" dataOut.Write(" + getter + ".Length);"); |
| out.println(" dataOut.Write(" + getter + ");"); |
| out.println(" }"); |
| } |
| } else if (propertyType.isArrayType()) { |
| if (size != null) { |
| out.println(" LooseMarshalObjectArrayConstSize(wireFormat, " + getter + ", dataOut, " + size.asInt() + ");"); |
| } else { |
| out.println(" LooseMarshalObjectArray(wireFormat, " + getter + ", dataOut);"); |
| } |
| } else if (isThrowable(propertyType)) { |
| out.println(" LooseMarshalBrokerError(wireFormat, " + getter + ", dataOut);"); |
| } else { |
| if (isCachedProperty(property)) { |
| out.println(" LooseMarshalCachedObject(wireFormat, (DataStructure)" + getter + ", dataOut);"); |
| } else { |
| out.println(" LooseMarshalNestedObject(wireFormat, (DataStructure)" + getter + ", dataOut);"); |
| } |
| } |
| } |
| } |
| |
| } |
