KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > test > util > ClassMover


1 /*
2   * JBoss, Home of Professional Open Source
3   * Copyright 2005, JBoss Inc., and individual contributors as indicated
4   * by the @authors tag. See the copyright.txt in the distribution for a
5   * full listing of individual contributors.
6   *
7   * This is free software; you can redistribute it and/or modify it
8   * under the terms of the GNU Lesser General Public License as
9   * published by the Free Software Foundation; either version 2.1 of
10   * the License, or (at your option) any later version.
11   *
12   * This software is distributed in the hope that it will be useful,
13   * but WITHOUT ANY WARRANTY; without even the implied warranty of
14   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15   * Lesser General Public License for more details.
16   *
17   * You should have received a copy of the GNU Lesser General Public
18   * License along with this software; if not, write to the Free
19   * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20   * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21   */
22 package org.jboss.test.util;
23
24 import java.net.URL JavaDoc;
25 import java.io.File JavaDoc;
26
27 /** A utility class that moves a class file to a .bak file to effectively
28  * remove it from the classpath using #move(String) and restores it using
29  * #restore(File).
30  *
31  * @author Scott.Stark@jboss.org
32  * @version $Revision: 37406 $
33  */
34 public class ClassMover
35 {
36    public static File JavaDoc move(String JavaDoc className)
37       throws ClassNotFoundException JavaDoc
38    {
39       File JavaDoc classFileBak = null;
40       String JavaDoc resName = className.replace('.', '/');
41       URL JavaDoc classRes = ClassMover.class.getResource("/" + resName + ".class");
42       if (classRes != null)
43       {
44          System.out.println("Found "+className+" impl at: " + classRes);
45          File JavaDoc classFile = new File JavaDoc(classRes.getFile());
46          classFileBak = new File JavaDoc(classFile.getAbsolutePath() + ".bak");
47          classFileBak.delete();
48          boolean moved = classFile.renameTo(classFileBak);
49          System.out.println("Moved to .bak: " + moved);
50       }
51       else
52       {
53          throw new ClassNotFoundException JavaDoc("No class file found: "+className);
54       }
55       return classFileBak;
56    }
57
58    public static void restore(File JavaDoc classFileBak)
59    {
60       if( classFileBak.exists() )
61       {
62          String JavaDoc name = classFileBak.getAbsolutePath();
63          String JavaDoc origName = name.substring(0, name.length() - 4);
64          File JavaDoc classFile = new File JavaDoc(origName);
65          boolean restored = classFileBak.renameTo(classFile);
66          System.out.println("Restored from .bak: " + restored);
67       }
68    }
69 }
70
Popular Tags