業務場景

我們都會碰到這樣的情況，某某使用者臨時又增加新的校驗規則需求，但是需求又很碎很急，沒法等到下一個版本上線（想打人有木有），這種時候如果為了上線該改動需要重啟服務，修改程式碼，除了增加開發工作量以外還增大了服務運維風險。我們當然可以使用類似weblogic之類支援熱部署的伺服器，但是這對大部分公司顯然不適用。
如果這個時候能夠將程式碼邏輯配置到資料庫中，讓運維人員編寫簡單的邏輯即可滿足需求而不需要大動干戈豈不是極好？所以這裡我引入了luaj指令碼這一概念。

技術介紹

lua是一種輕量級、支援互動式式程式設計的指令碼語言，在redhat、centos中都有自帶。
luaj即為LuaJavaBridge，提供與Java互相嵌入的支援。

互動呼叫

判斷規則仍然利用lua語言實現，存入資料庫中，java首先從資料庫中讀出判斷指令碼然後利用luaj ScriptEngineManager執行指令碼，指令碼中接受從java傳入的json字串引數，呼叫第三方lua解析為json物件，進行邏輯判斷後返回結果（json字串）。在luaj中只能使用.lua檔案，無法呼叫.so的C庫。

業務場景是根據不同的產品型別(測試列子一對一）獲取對應的校驗規則指令碼，並執行指令碼。

一、maven依賴

<!-- luaj -->
<dependency>
     <groupId>org.luaj</groupId
 
>
     <artifactId>luaj-jse</artifactId>
     <version>3.0.1</version>
</dependency>

二、建立規則表

規則表 t_test_rule

CREATE TABLE `t_test_rule` (
  `shell` varchar(5000) NOT NULL COMMENT 'lua指令碼',
  `set_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
 
 COMMENT '建立時間',
  `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主鍵',
  `type` int(1) DEFAULT '0' COMMENT '產品型別',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8;

三、編寫校驗指令碼

(demo)

package.path="F:/?.lua" --引入第三方lua，用於處理json,在luaj模式下只能使用.lua檔案而無法使用編譯好的.so庫
local dkjson = require("dkjson")
local str = "{\"code\":0,\"msg\":\"success!\"}"
local retObj,pos,err = dkjson.decode(str , 1, nil)--解析json字串
if(myParams==nil)
then
  retObj.code=1
  retObj.msg="入參丟失！"
  local retStr = dkjson.encode(retObj)
  return retStr
end


local jsonParams,jpos,jerr = dkjson.decode(otaParams , 1, nil)
if(jsonParams==nil)
then
  retObj.code=1
  retObj.msg=err
  local retStr = dkjson.encode(retObj)
  return retStr
end

if(jsonParams.version~=nil) then
 local sversion = string.gsub(version,"%p","")--lua字串過濾，去除.
 local num = tonumber(sversion)
 if(num <= 231) 
  then 
   retObj.code=1 
   retObj.msg="版本低於2.3.1"
 end
end

local retStr = dkjson.encode(retObj)
return retStr

java呼叫，這裡使用的是jfinal框架

public static boolean checkTheRule(Map<String,String> map, String params) {
        boolean flag = true;
        String type = map.get("type") + "";
        //規則查詢
        List<Record> shells = Db.find("SELECT t.shell from t_test_rule t where type = ?",type);

        if (shells != null) {
            for (Record tmp : shells) {
                String shell = tmp.getStr("shell");
                ScriptEngineManager mgr = new ScriptEngineManager();
                ScriptEngine e = mgr.getEngineByName("luaj");

                e.put("myParams", params);//引數傳入，這裡也可以傳入java類，成為在lua中呼叫執行的方法，比如mysql連線封裝方法
                try {
                    String resultMsg = e.eval(shell)+"";//執行並獲取返回結果
                    if(resultMsg==null || resultMsg==""){
                        log.info("lua指令碼執行返回空 ");
                        flag = false;
                        break;
                    }
                    RuleResult retObj = JSON.parseObject(resultMsg, RuleResult.class);
                    if(retObj!=null){
                        if(retObj.getCode()==0){
                            flag = true;
                        }else{
                            flag = false;
                            break;
                        }
                        log.info(retObj.getMsg());
                    }else{
                        log.info("json轉換失敗");
                        flag = false;
                        break;
                    }
                } catch (ScriptException e1) {
                    log.info("error:"+e1);
                    flag = false;
                    break;
                }
            }
        } else {
            log.info("沒有找到"+type+"對應的規則！");
            flag = true;
        }
        return flag;
    }

測試入口類

    @Test
    public void testPrint(){
        initConfig();
        Map map = new Hashtable();
        map.put("type", "38");
        map.put("version", "Ld2017");
        String params =  "{\"version\":\"2.3.0\"}";
        try {
            boolean flag = checkTheRule(map ,params);
            System.out.println(flag);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

四、執行test結果

後記

luaj中的lua指令碼如果要使用第三方方法，可以使用以下兩種方法：

• 引入第三方lua檔案
• 直接將封裝的Java方法傳入引擎中
  

第三方lua引用，用於解析JSON

package.path="xxx/?.lua"
local dkjson = require("dkjson")

Java方法設定，用於mysql讀取

/** 將java資料庫查詢方法丟入lua中使用 **/
e.put("readTable", new LuaMysqlConnector());

import java.util.List;

import org.luaj.vm2.LuaValue;
import org.luaj.vm2.lib.TwoArgFunction;

import com.jfinal.plugin.activerecord.Db;
import com.jfinal.plugin.activerecord.Record;

/**
 * 用於luaj在lua指令碼中執行mysql查詢
 * @author zh
 *
 */
public class LuaMysqlConnector extends TwoArgFunction{

    @Override
    /*
     * 入參lstatm 執行sql語句 retString 返回欄位字串;分割
     * @see org.luaj.vm2.lib.TwoArgFunction#call(org.luaj.vm2.LuaValue, org.luaj.vm2.LuaValue)
     */
    public LuaValue call(LuaValue lstatm, LuaValue retString) {
        /** 健壯性校驗 **/
        if (lstatm == null || lstatm.tojstring().equals("")) {
            return LuaValue.valueOf("丟失查詢sql");
        }
        if (retString == null
                || retString.tojstring().equals("")) {
            return LuaValue.valueOf("丟失返回欄位列表");
        }

        String[] retLabels = retString.tojstring().split(";");
        StringBuffer sb = new StringBuffer("[");
        List<Record> rets = Db.find(lstatm.tojstring());
        for (int r = 0; r < rets.size(); r++) {
            // 組裝為json格式
            sb.append("{");
            for (int i = 0; i < retLabels.length; i++) {
                sb.append("\"");
                sb.append(retLabels[i]);
                sb.append("\":\"");
                sb.append(rets.get(r).getStr(retLabels[i]));
                sb.append("\"");
                if (i != (retLabels.length - 1)) {
                    sb.append(",");
                }
            }
            if (r == (rets.size() - 1)) {
                sb.append("}");
            } else {
                sb.append("},");
            }

        }
        sb.append("]");
        return LuaValue.valueOf(sb.toString());
    }

}