io.logspace.jvm.agent.shaded.mchange.v2.codegen.bean.SerializableExtension Maven / Gradle / Ivy
The newest version!
/*
* Distributed as part of c3p0 v.0.9.1.1
*
* Copyright (C) 2005 Machinery For Change, Inc.
*
* Author: Steve Waldman
*
* This library is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 2.1, as
* published by the Free Software Foundation.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this software; see the file LICENSE. If not, write to the
* Free Software Foundation, Inc., 59 Temple Place, Suite 330,
* Boston, MA 02111-1307, USA.
*/
package io.logspace.jvm.agent.shaded.mchange.v2.codegen.bean;
import java.util.*;
import java.io.IOException;
import io.logspace.jvm.agent.shaded.mchange.v2.codegen.IndentedWriter;
/**
* Note: this class pays no attention to whether users have marked any property variables as transient.
* In fact, it will work most efficiently if users mark ALL variables as transient... to define transient
* properties for this class, use the constructor which allows a user-specified set of transients.
*/
public class SerializableExtension implements GeneratorExtension
{
Set transientProperties;
Map transientPropertyInitializers;
/**
* @param transientProperties a set of Strings, the names of all properties that should be considered transient and not serialized
* @param transientPropertyInitializers an optional Map of a subset of the transient property names to non-default initialization
* expressions, which should be unterminated expressions, and which will be used verbatim in
* the generated code.
*/
public SerializableExtension(Set transientProperties, Map transientPropertyInitializers)
{
this.transientProperties = transientProperties;
this.transientPropertyInitializers = transientPropertyInitializers;
}
public SerializableExtension()
{ this ( Collections.EMPTY_SET, null ); }
public Collection extraGeneralImports()
{ return Collections.EMPTY_SET; }
public Collection extraSpecificImports()
{
Set set = new HashSet();
set.add( "java.io.IOException" );
set.add( "java.io.Serializable" );
set.add( "java.io.ObjectOutputStream" );
set.add( "java.io.ObjectInputStream" );
return set;
}
public Collection extraInterfaceNames()
{
Set set = new HashSet();
set.add( "Serializable" );
return set;
}
public void generate(ClassInfo info, Class superclassType, Property[] props, Class[] propTypes, IndentedWriter iw)
throws IOException
{
iw.println("private static final long serialVersionUID = 1;");
iw.println("private static final short VERSION = 0x0001;");
iw.println();
iw.println("private void writeObject( ObjectOutputStream oos ) throws IOException");
iw.println("{");
iw.upIndent();
iw.println( "oos.writeShort( VERSION );" );
for( int i = 0, len = props.length; i < len; ++i )
{
Property prop = props[i];
if (! transientProperties.contains( prop.getName() ) )
{
Class propType = propTypes[i];
if (propType != null && propType.isPrimitive()) //primitives should always resolve, object types may not, and be null
{
if (propType == byte.class)
iw.println("oos.writeByte(" + prop.getName() + ");");
else if (propType == char.class)
iw.println("oos.writeChar(" + prop.getName() + ");");
else if (propType == short.class)
iw.println("oos.writeShort(" + prop.getName() + ");");
else if (propType == int.class)
iw.println("oos.writeInt(" + prop.getName() + ");");
else if (propType == boolean.class)
iw.println("oos.writeBoolean(" + prop.getName() + ");");
else if (propType == long.class)
iw.println("oos.writeLong(" + prop.getName() + ");");
else if (propType == float.class)
iw.println("oos.writeFloat(" + prop.getName() + ");");
else if (propType == double.class)
iw.println("oos.writeDouble(" + prop.getName() + ");");
}
else
writeStoreObject( prop, propType, iw );
}
}
generateExtraSerWriteStatements( info, superclassType, props, propTypes, iw);
iw.downIndent();
iw.println("}");
iw.println();
iw.println("private void readObject( ObjectInputStream ois ) throws IOException, ClassNotFoundException");
iw.println("{");
iw.upIndent();
iw.println("short version = ois.readShort();");
iw.println("switch (version)");
iw.println("{");
iw.upIndent();
iw.println("case VERSION:");
iw.upIndent();
for( int i = 0, len = props.length; i < len; ++i )
{
Property prop = props[i];
if (! transientProperties.contains( prop.getName() ) )
{
Class propType = propTypes[i];
if (propType != null && propType.isPrimitive()) //if a propType is unresolvable, it ain't a primitive
{
if (propType == byte.class)
iw.println("this." + prop.getName() + " = ois.readByte();");
else if (propType == char.class)
iw.println("this." + prop.getName() + " = ois.readChar();");
else if (propType == short.class)
iw.println("this." + prop.getName() + " = ois.readShort();");
else if (propType == int.class)
iw.println("this." + prop.getName() + " = ois.readInt();");
else if (propType == boolean.class)
iw.println("this." + prop.getName() + " = ois.readBoolean();");
else if (propType == long.class)
iw.println("this." + prop.getName() + " = ois.readLong();");
else if (propType == float.class)
iw.println("this." + prop.getName() + " = ois.readFloat();");
else if (propType == double.class)
iw.println("this." + prop.getName() + " = ois.readDouble();");
}
else
writeUnstoreObject( prop, propType, iw );
}
else
{
String initializer = (String) transientPropertyInitializers.get( prop.getName() );
if (initializer != null)
iw.println("this." + prop.getName() + " = " + initializer +';');
}
}
generateExtraSerInitializers( info, superclassType, props, propTypes, iw);
iw.println("break;");
iw.downIndent();
iw.println("default:");
iw.upIndent();
iw.println("throw new IOException(\"Unsupported Serialized Version: \" + version);");
iw.downIndent();
iw.downIndent();
iw.println("}");
iw.downIndent();
iw.println("}");
}
protected void writeStoreObject( Property prop, Class propType, IndentedWriter iw ) throws IOException
{
iw.println("oos.writeObject( " + prop.getName() + " );");
}
protected void writeUnstoreObject( Property prop, Class propType, IndentedWriter iw ) throws IOException
{
iw.println("this." + prop.getName() + " = (" + prop.getSimpleTypeName() + ") ois.readObject();");
}
protected void generateExtraSerWriteStatements(ClassInfo info, Class superclassType, Property[] props, Class[] propTypes, IndentedWriter iw)
throws IOException
{}
protected void generateExtraSerInitializers(ClassInfo info, Class superclassType, Property[] props, Class[] propTypes, IndentedWriter iw)
throws IOException
{}
}