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> T getBean(Class<T> 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方法

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

#!/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

大功告成。尽情玩耍吧