gemlite.core.internal.asm.serialize.DataSerializeHelper.java Source code

Introduction

Here is the source code for gemlite.core.internal.asm.serialize.DataSerializeHelper.java

Source

/*                                                                         
 * Copyright 2010-2013 the original author or authors.                     
 *                                                                         
 * Licensed 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 gemlite.core.internal.asm.serialize;

import gemlite.core.internal.asm.serialize.dataserialize.DataSFieldProcessor;

import java.io.IOException;
import java.util.List;

import org.objectweb.asm.Opcodes;
import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.FieldNode;
import org.objectweb.asm.tree.InsnList;
import org.objectweb.asm.tree.InsnNode;
import org.objectweb.asm.tree.MethodNode;

@SuppressWarnings({ "unchecked", "rawtypes" })
public class DataSerializeHelper implements Opcodes {
    public final static String DATAS_NAME = "com/gemstone/gemfire/DataSerializable";
    private DataSFieldProcessor fp = new DataSFieldProcessor();

    private final static DataSerializeHelper instance = new DataSerializeHelper();

    private DataSerializeHelper() {

    }

    public final static DataSerializeHelper getInstance() {
        return instance;
    }

    public boolean process(ClassNode cn) throws IOException {
        boolean result = addDataSerializeInterface(cn);
        return result;
    }

    /***
     * @param cn
     * @return true:add a interface
     */

    private boolean addDataSerializeInterface(ClassNode cn) {
        List interfaces = cn.interfaces;
        for (int i = 0; i < interfaces.size(); i++) {
            String interfaceName = (String) interfaces.get(i);
            if (DATAS_NAME.equals(interfaceName)) {
                //        if (DomainMojoHelper.log().isDebugEnabled())
                //          DomainMojoHelper.log().debug(cn.name + " has implemented DataSerialize interface.");
                return false;
            }
        }
        //    if (DomainMojoHelper.log().isDebugEnabled())
        //      DomainMojoHelper.log().debug(cn.name + " add DataSerialize interface code.");
        interfaces.add(DATAS_NAME);

        implementInterface(cn);
        return true;
    }

    private void implementInterface(ClassNode cn) {
        MethodNode toMethod = new MethodNode(ACC_PUBLIC, "toData", "(Ljava/io/DataOutput;)V", null,
                new String[] { "java/io/IOException" });
        InsnList instToMethod = toMethod.instructions;

        MethodNode fromMethod = new MethodNode(ACC_PUBLIC, "fromData", "(Ljava/io/DataInput;)V", null,
                new String[] { "java/io/IOException", "java/lang/ClassNotFoundException" });
        InsnList instFromMethod = fromMethod.instructions;
        for (int i = 0; i < cn.fields.size(); i++) {
            FieldNode fn = (FieldNode) cn.fields.get(i);

            fp.toMethod(cn.name, fn, instToMethod);
            fp.fromMethod(cn.name, fn, instFromMethod);
        }
        instToMethod.add(new InsnNode(RETURN));
        cn.methods.add(toMethod);

        instFromMethod.add(new InsnNode(RETURN));
        cn.methods.add(fromMethod);

        //    if (DomainMojoHelper.log().isDebugEnabled())
        //      DomainMojoHelper.log().debug(cn.name + " add toData and fromData method.");
    }

    public void endProcess() {

    }

}