摘要：據(jù)說(shuō)已經(jīng)原生支持參數(shù)名讀取了。本文以為例進(jìn)行說(shuō)明通過(guò)字節(jié)碼操作工具我們可以實(shí)現(xiàn)運(yùn)行時(shí)參數(shù)名的讀寫。簡(jiǎn)單說(shuō)說(shuō)原理字節(jié)碼為每個(gè)方法保存了一份方法本地變量列表。

據(jù)說(shuō)Java8已經(jīng)原生支持參數(shù)名讀取了。具體不是很清楚。本文以java7為例進(jìn)行說(shuō)明.
通過(guò)ASM字節(jié)碼操作工具我們可以實(shí)現(xiàn)運(yùn)行時(shí)參數(shù)名的讀寫。
簡(jiǎn)單說(shuō)說(shuō)原理：java字節(jié)碼為每個(gè)方法保存了一份方法本地變量列表。可以通過(guò)ASM獲取這個(gè)列表。但是可能會(huì)獲得列表順序與期望的不一致。我們獲取的本地變量了列表使用不同的編譯器編譯得到的順序可能不同。一般而言，通過(guò)javac編譯出來(lái)的列表順序是按照本地變量使用的順序。而我們期望的是聲明的順序。如下面這個(gè)方法:

public String handle(String a,String b){
    String c;
    c="str";
    return a+b;
}

得到的localVariable列表順序:我們期望的是this,a,b,c.而實(shí)際上讀取到的會(huì)是this ,c ,a,b
這個(gè)this是非靜態(tài)方法中的第一個(gè)本地變量。所以我們會(huì)在下面的代碼中對(duì)這個(gè)順序進(jìn)行重排序以解決該問(wèn)題。

不多說(shuō)，直接看代碼:

package net.xby1993.springmvc.controller;

import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.objectweb.asm.ClassReader;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.LocalVariableNode;
import org.objectweb.asm.tree.MethodNode;

public class MethodParamNamesScanner {

    /**
     * 獲取方法參數(shù)名列表
     * 
     * @param clazz
     * @param m
     * @return
     * @throws IOException
     */
    public static List getMethodParamNames(Class clazz, Method m) throws IOException {
        try (InputStream in = clazz.getResourceAsStream("/" + clazz.getName().replace(".", "/") + ".class")) {
            return getMethodParamNames(in,m);
        }

    }
    public static List getMethodParamNames(InputStream in, Method m) throws IOException {
        try (InputStream ins=in) {
            return getParamNames(ins,
                    new EnclosingMetadata(m.getName(),Type.getMethodDescriptor(m), m.getParameterTypes().length));
        }

    }
    /**
     * 獲取構(gòu)造器參數(shù)名列表
     * 
     * @param clazz
     * @param constructor
     * @return
     */
    public static List getConstructorParamNames(Class clazz, Constructor constructor) {
        try (InputStream in = clazz.getResourceAsStream("/" + clazz.getName().replace(".", "/") + ".class")) {
            return getConstructorParamNames(in, constructor);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } 
        return new ArrayList();
    }
    public static List getConstructorParamNames(InputStream ins, Constructor constructor) {
        try (InputStream in = ins) {
            return getParamNames(in, new EnclosingMetadata(constructor.getName,Type.getConstructorDescriptor(constructor),
                    constructor.getParameterTypes().length));
        } catch (IOException e1) {
            // TODO Auto-generated catch block
            e1.printStackTrace();
        }
        return new ArrayList();
    }
    /**
     * 獲取參數(shù)名列表輔助方法
     * 
     * @param in
     * @param m
     * @return
     * @throws IOException
     */
    private static List getParamNames(InputStream in, EnclosingMetadata m) throws IOException {
        ClassReader cr = new ClassReader(in);
        ClassNode cn = new ClassNode();
        cr.accept(cn, ClassReader.EXPAND_FRAMES);// 建議EXPAND_FRAMES
        // ASM樹接口形式訪問(wèn)
        List methods = cn.methods;
        List list = new ArrayList();
        for (int i = 0; i < methods.size(); ++i) {
            List varNames = new ArrayList();
            MethodNode method = methods.get(i);
            // 驗(yàn)證方法簽名
            if (method.desc.equals(m.desc)&&method.name.equals(m.name)) {
//                System.out.println("desc->"+method.desc+":"+m.desc);
                List local_variables = method.localVariables;
                for (int l = 0; l < local_variables.size(); l++) {
                    String varName = local_variables.get(l).name;
                    // index-記錄了正確的方法本地變量索引。(方法本地變量順序可能會(huì)被打亂。而index記錄了原始的順序)
                    int index = local_variables.get(l).index;
                    if (!"this".equals(varName)) // 非靜態(tài)方法,第一個(gè)參數(shù)是this
                        varNames.add(new LocalVariable(index, varName));
                }
                LocalVariable[] tmpArr = varNames.toArray(new LocalVariable[varNames.size()]);
                // 根據(jù)index來(lái)重排序，以確保正確的順序
                Arrays.sort(tmpArr);
                for (int j = 0; j < m.size; j++) {
                    list.add(tmpArr[j].name);
                }
                break;

            }

        }
        return list;
    }

    /**
     * 方法本地變量索引和參數(shù)名封裝
     * @author xby Administrator
     */
    static class LocalVariable implements Comparable {
        public int index;
        public String name;

        public LocalVariable(int index, String name) {
            this.index = index;
            this.name = name;
        }

        public int compareTo(LocalVariable o) {
            return this.index - o.index;
        }
    }

    /**
     * 封裝方法描述和參數(shù)個(gè)數(shù)
     * 
     * @author xby Administrator
     */
    static class EnclosingMetadata {
              //method name
              public String name;
        // method description
        public String desc;
        // params size
        public int size;

        public EnclosingMetadata(String name,String desc, int size) {
                      this.name=name;
            this.desc = desc;
            this.size = size;
        }
    }

    public static void main(String[] args) throws IOException {
        for (Method m : AdminController.class.getDeclaredMethods()) {
            List list = getMethodParamNames(AdminController.class, m);
            System.out.println(m.getName() + ":");
            for (String str : list) {
                System.out.println(str);
            }
            System.out.println("------------------------");
        }
    }
}