Groovy调用示例

项目背景介绍：

由于某电商网站的数据量比较大，价格体系变更频繁，因此采用Groovy语言对价格统一展示，设计一个价格引擎具体如下：

一 、控制层中调用Groovy脚本

                ObjectMapper mapper = new ObjectMapper();  
Map<String,Object> params = JsonUtil.jsonToMap( getInParam().get("params").toString() );
TAceRule aceRule = new TAceRule();
aceRule.setName( getInParam().get("ace_rule_name").toString() );//获取groovy脚本名称
MyBatisDAO mybatisDao = (MyBatisDAO) this.getDaoFactory().get( "myBatisDAO" );
aceRule = mybatisDao.getAceRule(aceRule);
if(aceRule == null){
getOutParam().put("errorDetail", "没有配置通用规则");
return RETURN.REQ_PARAM_ERROR;
}
// 获取groovy脚本引擎
Calculate cal = CalculateManager.instant.calculate(StringUtils.hasText(aceRule.getEngine())?aceRule.getEngine():"groovy");//获取计算引擎，默认为groovy
CalculateContext context = CalculateManager.instant.context();//实例化CalculateContext  类
context.setContext("dao", getDaoFactory());  //设置Dao,Groovy引用的时候直接用dao来调用
context.setContext("session",getSession());//设置session，
context.setContext("cookie", getCookies());//设置cookie
context.setContext("inparams", params );//设置参数
cal.calculate(aceRule.getContext(), context);//参数 aceRule.getContext() 为数据库中的groovy脚本文件。
setOutParam(new HashMap<String, Object>(context.getContext()));
getOutParam().put("errorDetail", "计算完成");
return RETURN.SUCCESS;

二、CalculateManager 接口类

public enum CalculateManager {
instant;
public Calculate calculate(String script ){
//默认获取groovy
return new ScriptCalculate(script);
}

public CalculateContext context(){
return new CalculateContext();
}
}

三、获取Groovy驱动

import java.math.BigDecimal;
import java.util.HashMap;
import java.util.Map;
import javax.script.Bindings;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import org.leafframework.mvc.service.ace.Calculate;
import org.leafframework.mvc.service.ace.CalculateContext;
import org.leafframework.util.MapUtil;
import org.springframework.util.StringUtils;

public class ScriptCalculate implements Calculate{
private ScriptEngine engine ;
@Override
public BigDecimal calculate(String calculateScript,
CalculateContext context) {
//创建绑定
Bindings bindings= engine.createBindings();
for(Map.Entry<String, ?> entry:context.getContext().entrySet()){
bindings.put(entry.getKey(), entry.getValue());
}
Object result = null;
BigDecimal ret = null;
try {
result = engine.eval(calculateScript, bindings);
MapUtil<String, Object> retMap=MapUtil.of();
if(result != null){
if(result instanceof Number){
ret= new BigDecimal(((Number)result).doubleValue());
retMap.put(evalPrice, ret);
}else if(result instanceof String){
ret= new BigDecimal((String)result);
retMap.put(evalPrice, ret);
}else if(result instanceof Bindings){
retMap.put(evalResult, new HashMap<String,Object>((Bindings)result));
}else{
retMap.put(evalResult, result);
}
}
context.setContext(retMap.map());

} catch (Throwable e) {
// TODO Auto-generated catch block
e.printStackTrace();
context.setContext(MapUtil.<String,Object>of().put("error", e.getMessage()).map());
}
return ret;
}
public ScriptCalculate(String scriptname){
ScriptEngineManager manager = new ScriptEngineManager();
engine = manager.getEngineByName(StringUtils.hasText(scriptname)?scriptname:"javascript");
if(engine == null){ //默认使用js引擎
engine = manager.getEngineByName("javascript");
}
}
}

三、CalculateContext  类，将待传递到Groovy中的参数封装到Map中

public class CalculateContext {
private Map<String, Object> map = new HashMap<String, Object>();
public void setContext(String k, Object v){
map.put(k, v);
}
public Object getContext(String k){
return map.get(k);
}
public void setContext(Map<String, Object> map){
this.map=map;
}
public Map<String, Object> getContext(){
return map;
}
}

四、数据中groovy脚本文件

import groovy.json.JsonSlurper;
import org.leafframework.data.dao.orm.TGGoodsPrice;
import org.springframework.util.StringUtils;
import java.text.DecimalFormat;
def params = new JsonSlurper().parseText(inparams.params) ;
//data 数据格式：商品ID，数量;
String pricecode=inparams.priceCode;
String custtype=inparams.custType;
String custlvl=inparams.custLvl;
String data = params.data;
def    mybatis=dao.myBatisDAO;
def result=[:];
double orderAmont=0.0;
double feecardAmont=0.0;
double numrateAmont=0.0;
double userDiscAmont=0.0;
double realOrderAmont=0.0;
if((!StringUtils.isEmpty(data)))
{
String[] goodslist = data.split(";");
double diectPrice = 0.0;
for(String str:goodslist)
{
String[] goods = str.split(",");
String goodsId=goods[0];
int count = Integer.valueOf(goods[1]);
def priceList = mybatis.getGoodsPrice(["pricecode":pricecode?.trim(),"goodsId":goodsId?.trim()]);
if(priceList.size()==0)
{
priceList = mybatis.getGoodsPrice(["pricecode":"0001","goodsId":goodsId?.trim()]);
}
double nrAmont=0.0;
double feAmont=0.0;
double orAmont=0.0;
for(def goodprice:priceList)
{
double price   = goodprice.getValue();
int    code    = goodprice.getCode();
int    min     = 0;
int    max     = 0;
double disc    = 0.0; 
double feecard = 0.0; 
println count+" : "+price;
if( goodprice.getNumMin()!=null)
{
min = goodprice.getNumMin();
}
if( goodprice.getNumMax()!=null)
{
max = goodprice.getNumMax();
}
if( goodprice.getDiscount()!=null)
{
disc = goodprice.getDiscount();
}
if( goodprice.getFeecard()!=null)
{
feecard = goodprice.getFeecard();
}
//println price +":"+code +":  "+min+":"+max+"  :"+ disc+":"+feecard;
if( count >=min && count<=max )
{
println "1sssssss:"+disc; 
if( disc!=null &&( disc >0 && disc<1) )
{
nrAmont=(count*(price-feecard))*(1-disc);
}
feAmont = count*feecard;
orAmont = price*count;
println nrAmont+" : "+ feAmont+":"+orAmont; 
break;
}
if( code == 1)
{
println "2sssssss:"+disc; 
if( disc!=null && (disc >0 && disc<1 ))
{
nrAmont=(count*(price-feecard))*(1-disc);
}
feAmont =count*feecard;
orAmont =price*count;
}
}
orderAmont   =orderAmont+orAmont;
feecardAmont =feecardAmont+feAmont;
numrateAmont =numrateAmont+nrAmont;
}
def custdisc = mybatis.getCustDisc( custlvl );
double userrate=custdisc.get(0).getDiscount();
if( userrate >0 && userrate<1 )
{
userrate=custdisc.get(0).getDiscount();
userDiscAmont= (1-userrate)*(orderAmont-feecardAmont-numrateAmont);
println "user: "+userrate+" : "+ (orderAmont-feecardAmont-numrateAmont);
}
realOrderAmont=orderAmont-feecardAmont-numrateAmont-userDiscAmont;
}
DecimalFormat df = new DecimalFormat(".00");
result.feecardAmont    =Double.parseDouble(df.format( feecardAmont) );
result.numrateAmont    =Double.parseDouble(df.format( numrateAmont) );
result.userDiscAmont   =Double.parseDouble(df.format( userDiscAmont) );
result.orderAmont      =Double.parseDouble(df.format( orderAmont) );
result.realOrderAmont  =Double.parseDouble(df.format( realOrderAmont) );
result;