org.yx.asm.ArgPojos Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sumk Show documentation
A quick developing framewort for internet company
There is a newer version: 4.0.2
/**
 * Copyright (C) 2016 - 2030 youtongluan.
 *
 * 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 org.yx.asm;

import static org.objectweb.asm.Opcodes.ACC_PUBLIC;
import static org.objectweb.asm.Opcodes.ACC_SUPER;
import static org.objectweb.asm.Opcodes.ALOAD;
import static org.objectweb.asm.Opcodes.ARETURN;
import static org.objectweb.asm.Opcodes.GETFIELD;
import static org.objectweb.asm.Opcodes.INVOKESPECIAL;
import static org.objectweb.asm.Opcodes.RETURN;

import java.lang.reflect.Method;

import org.objectweb.asm.ClassWriter;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;
import org.yx.log.Logs;
import org.yx.util.UUIDSeed;

public class ArgPojos {

	@SuppressWarnings("unchecked")
	public static Class create(MethodParamInfo p) throws Exception {
		final Method method = p.getMethod();
		String fullName = String.join("_", p.getDeclaringClass().getSimpleName(), method.getName(), UUIDSeed.seq());
		if (Logs.asm().isTraceEnabled()) {
			Logs.asm().trace("begin generate paramters pojo :{}", fullName);
		}

		ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS);
		cw.visit(AsmUtils.jvmVersion(), ACC_PUBLIC | ACC_SUPER, fullName, null, "java/lang/Object",
				new String[] { "org/yx/asm/ArgPojo" });
		Arg[] args = new Arg[p.getArgNames().length];
		for (int i = 0; i < args.length; i++) {
			String arg = p.getArgNames()[i];
			String desc = p.getDescs()[i];
			args[i] = new Arg(arg, desc);
			cw.visitField(ACC_PUBLIC, arg, desc, p.getSignatures()[i], null).visitEnd();
		}

		MethodVisitor mv = cw.visitMethod(ACC_PUBLIC, "", "()V", null, null);
		mv.visitCode();
		mv.visitVarInsn(ALOAD, 0);
		mv.visitMethodInsn(INVOKESPECIAL, "java/lang/Object", "", "()V", false);
		mv.visitInsn(RETURN);
		mv.visitMaxs(1, 1);
		mv.visitEnd();

		mv = cw.visitMethod(ACC_PUBLIC, "params", "()[Ljava/lang/Object;", null, null);
		mv.visitCode();
		buildArgArray(fullName, mv, args, method.getParameterTypes());
		mv.visitInsn(ARETURN);
		mv.visitMaxs(1, 0);
		mv.visitEnd();

		mv = cw.visitMethod(ACC_PUBLIC, "invoke", "(Ljava/lang/Object;)Ljava/lang/Object;", null,
				new String[] { "java/lang/Throwable" });
		mv.visitCode();
		Class ownerClz = method.getDeclaringClass();
		final String ownerName = ownerClz.getName().replace('.', '/');

		mv.visitLdcInsn(Type.getType(ownerClz));
		mv.visitVarInsn(ALOAD, 1);
		mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/lang/Class", "cast", "(Ljava/lang/Object;)Ljava/lang/Object;",
				false);
		mv.visitTypeInsn(Opcodes.CHECKCAST, ownerName);
		loadObjectFields(fullName, mv, args, method.getParameterTypes());
		mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, ownerName, method.getName(), p.getMethodDesc(), false);
		Class returnType = method.getReturnType();
		if (void.class == method.getReturnType()) {
			mv.visitInsn(Opcodes.ACONST_NULL);
			Logs.asm().debug("{} has no return", fullName);
		} else if (returnType.isPrimitive()) {
			WriterHelper.boxPrimitive(mv, returnType);
		}
		mv.visitInsn(ARETURN);
		mv.visitMaxs(1, 0);
		mv.visitEnd();

		cw.visitEnd();

		return (Class) AsmUtils.loadClass(fullName.replace('/', '.'), cw.toByteArray());
	}

	private static void buildArgArray(String fullName, MethodVisitor mv, Arg[] args, Class[] params) {
		int frameIndex = 1;
		for (int i = 0; i < params.length; i++) {
			mv.visitVarInsn(ALOAD, 0);
			mv.visitFieldInsn(GETFIELD, fullName, args[i].name, args[i].desc);
			frameIndex += WriterHelper.storeToLocalVariable(mv, params[i], frameIndex);
		}
		WriterHelper.buildParamArray(mv, params);
	}

	private static void loadObjectFields(String fullName, MethodVisitor mv, Arg[] args, Class[] params) {
		for (int i = 0; i < params.length; i++) {
			mv.visitVarInsn(ALOAD, 0);
			mv.visitFieldInsn(GETFIELD, fullName, args[i].name, args[i].desc);
		}
	}

	private static class Arg {
		String name;
		String desc;

		private Arg(String name, String desc) {
			this.name = name;
			this.desc = desc;
		}
	}
}