spring + groovy 实现动态代码注入执行

昨天线上的代码调用一个远程的服务无缘无故不成功，又没加那么多日志，不好定位问题，好想在线上执行一下代码，打印点log看看
于是想着怎么动态执行点java代码，忽然想起以前玩过的groovy，于是搞起来

大概思路是这样，写一个控制器，接收一段代码，动态执行，然后返回执行结果，切记，做好权限控制，免得杯具

没想到实现起来异常简单

1、gradle.build加入groovy依赖

compile "org.codehaus.groovy:groovy:2.5.3"

2、写个工具类，方便拿到spring上下文对象，这个很常见

import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;

public class SpringContextUtils implements ApplicationContextAware {

    static ApplicationContext context;

    @Override
    public void setApplicationContext(ApplicationContext context)
            throws BeansException {
        SpringContextUtils.context = context;
    }

    public static ApplicationContext getContext() {
        return context;
    }

    public static void autowireBean(Object bean) {
        context.getAutowireCapableBeanFactory().autowireBean(bean);
    }

    public static  T getBean(Class clazz) {
        return context.getBean(clazz);
    }

}

3、写个控制器

import com.ruizton.util.SpringContextUtils;
import groovy.lang.GroovyClassLoader;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.lang.reflect.Method;

@RestController
public class GroovyController {

    @RequestMapping("/runScript")
    public Object runScript(String script) throws Exception {
        if (script != null) {
         // 这里其实就是groovy的api动态的加载生成一个Class，然后反射生成对象，然后执行run方法，最后返回结果
         // 最精华的地方就是SpringContextUtils.autowireBean，可以实现自动注入bean，
            Class clazz = new GroovyClassLoader().parseClass(script);
            Method run = clazz.getMethod("run");
            Object o = clazz.newInstance();
            SpringContextUtils.autowireBean(o);
            Object ret = run.invoke(o);
            return ret;
        } else {
            return "no script";
        }
    }

}

上面是java的全部内容了，接下来就是调用了

首先准备写个test.groovy代码

import org.springframework.beans.factory.annotation.Autowired

class Foo {

    @Autowired
    FooService fooService;

    Object run() {
        // do something
        def f = fooService.findById(38);
        if (f != null) {
            return f.name
        }
        return null
    }

}

不写类直接写代码块也是可以的
test.groovy

def sum = 1 + 2
return "sum = " + sum

上面代码也会生成一个Class对象，里面的代码默认在run方法下面，所以控制器哪里都是调用的run方法

再写个python调用一个接口，其实就是读文件然后调用一下java的接口，要是不怕死，也可以在后台做一个可视化界面，加个执行按钮直接调用

test.py

#!/usr/bin/python

import requests

def read(filename):
    f = open(filename,'r')
    file = f.read()
    f.close()
    return file


res = requests.post('http://127.0.0.1/runScript.html', data = {
    'script': read('test.groovy')
})

print res.text

$ chmod +x test.py
$ ./test.py

大功告成。尽情玩耍吧