TransformDemo.java

/*
 * Copyright 2003 The Apache Software Foundation
 *
 *  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 net.sf.cglib.transform.impl;

import net.sf.cglib.transform.*;
import net.sf.cglib.core.*;

import java.util.*;

/**
 *
 * @author  baliuka
 */
public class TransformDemo {
    
    public static void register(Class cls){
      System.out.println("register " + cls);
    }
    
   
    public static void start(){
     
        MA ma = new MA();
        makePersistent(ma);
        ma.setCharP('A');
        ma.getCharP();
        ma.setDoubleP(554);
        ma.setDoubleP(1.2);
        ma.getFloatP();
        ma.setName("testName");
        ma.publicField = "set value";
        ma.publicField = ma.publicField + " append value";
        ma.setBaseTest("base test field");
        ma.getBaseTest();
    
    }
    
    public static void makePersistent(Object obj){
        System.out.println( "makePersistent " + obj.getClass() + " " +  Arrays.asList(obj.getClass().getInterfaces()) ); 
        InterceptFieldEnabled t = (InterceptFieldEnabled)obj;
        t.setInterceptFieldCallback( new StateManager());
        FieldProvider provider = (FieldProvider)obj;
        System.out.println("Field Names " + Arrays.asList(provider.getFieldNames()) );
        System.out.println("Field Types " + Arrays.asList(provider.getFieldTypes()) );
        PersistenceCapable pc = (PersistenceCapable)obj;
        pc.setPersistenceManager("Manager");
    
    }
    
    public static void main( String args [] )throws Exception{
    
     ClassTransformerFactory transformation =  
     
        new ClassTransformerFactory (){
        
           public ClassTransformer newInstance(){
            try{       
              InterceptFieldTransformer t1 = new InterceptFieldTransformer( new Filter() );
        
        
              AddStaticInitTransformer t2 = new   AddStaticInitTransformer(
                TransformDemo.class.getMethod("register",new Class[]{Class.class}) 
              );                                   
        
        
              AddDelegateTransformer t3 = new AddDelegateTransformer(
                          new Class[]{PersistenceCapable.class},
                          PersistenceCapableImpl.class
                    );    
                          
              FieldProviderTransformer t4 = new FieldProviderTransformer();                 
        
              return new ClassTransformerChain( new ClassTransformer[]{t4,t1,t2,t3} );
              
            }catch(Exception e){
              throw new CodeGenerationException(e);
            } 
           }  
    };
        
        TransformingClassLoader loader = new TransformingClassLoader(
          TransformDemo.class.getClassLoader(),
          new ClassFilter(){
            public boolean accept(String name){
                System.out.println("load : "  + name);
                boolean f =
                  Base.class.getName().equals(name) ||  
                  MA.class.getName().equals(name) || 
                     TransformDemo.class.getName().equals(name);
                if(f){
                 System.out.println("transforming " + name);
                }
                return f;
            } 
         },
         transformation
        );
        
        
        loader.loadClass(TransformDemo.class.getName()).getMethod("start",new Class[]{}).invoke(null, (Object[])null);
    
    }
    
    
   public static class Filter implements InterceptFieldFilter{
       
        
        public boolean acceptRead(org.objectweb.asm.Type owner, String name) {
                  return true;
        }
        
        public boolean acceptWrite(org.objectweb.asm.Type owner, String name) {
                  return true;
        }
        
    };
    

  public   static class  StateManager implements InterceptFieldCallback{
        
       
        public boolean readBoolean(Object _this, String name, boolean oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }        
       
        public byte readByte(Object _this, String name, byte oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public char readChar(Object _this, String name, char oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public double readDouble(Object _this, String name, double oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public float readFloat(Object _this, String name, float oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public int readInt(Object _this, String name, int oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public long readLong(Object _this, String name, long oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public Object readObject(Object _this, String name, Object oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public short readShort(Object _this, String name, short oldValue) {
            System.out.println("read " +  name + " = " + oldValue);
            return oldValue;
        }
        
        public boolean writeBoolean(Object _this, String name, boolean oldValue, boolean newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public byte writeByte(Object _this, String name, byte oldValue, byte newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public char writeChar(Object _this, String name, char oldValue, char newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public double writeDouble(Object _this, String name, double oldValue, double newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public float writeFloat(Object _this, String name, float oldValue, float newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public int writeInt(Object _this, String name, int oldValue, int newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public long writeLong(Object _this, String name, long oldValue, long newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public Object writeObject(Object _this, String name, Object oldValue, Object newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
        public short writeShort(Object _this, String name, short oldValue, short newValue) {
            System.out.println( "write " + name + " = " + newValue);
            return newValue;
        }
        
       }

    
    
}