MethodArea.java

package org.xukai.jvm.engine;

import org.xukai.jvm.clz.ClassFile;
import org.xukai.jvm.constant.MethodRefInfo;
import org.xukai.jvm.loader.ClassFileLoader;
import org.xukai.jvm.method.Method;

import java.util.HashMap;
import java.util.Map;


public class MethodArea {
	
	public static final MethodArea instance = new MethodArea();
	
	/**
	 * 注意：我们做了极大的简化， ClassLoader 只有一个， 实际JVM中的ClassLoader,是一个双亲委托的模型
	 */
	
	private ClassFileLoader clzLoader = null;
	
	Map<String,ClassFile> map = new HashMap<String,ClassFile>();
	
	private MethodArea(){		
	}
	
	public static MethodArea getInstance(){
		return instance;
	}
	
	public void setClassFileLoader(ClassFileLoader clzLoader){
		this.clzLoader = clzLoader;
	}
	
	public Method getMainMethod(String className){
		
		ClassFile clzFile = this.findClassFile(className);
		
		return clzFile.getMainMethod();
	}
	
	
	public  ClassFile findClassFile(String className){
		
		if(map.get(className) != null){
			return map.get(className);
		}
		// 看来该class 文件还没有load过
		ClassFile clzFile = this.clzLoader.loadClass(className);
		
		map.put(className, clzFile);
		
		return clzFile;
		
	}
	
	
	public Method getMethod(String className, String methodName, String paramAndReturnType){
		ClassFile clzFile = this.findClassFile(className);
		return clzFile.getMethod(methodName,paramAndReturnType);
	}
	
	
	public Method getMethod(MethodRefInfo methodRef){
		String methodName = methodRef.getMethodName();
		String paramAndReturnType = methodRef.getParamAndReturnType();
		for (String clzName : map.keySet()) {
			Method method = getMethod(clzName, methodName, paramAndReturnType);
			if (method != null) {
				return method;
			}
		}
		return null;
	}
}