org.mvel2.jsr223.MvelScriptEngine Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tbel Show documentation

TBEL is a powerful expression language for ThingsBoard platform user-defined functions. Original implementation is based on MVEL.

There is a newer version: 1.2.5

Show newest version

package org.mvel2.jsr223;

import java.io.IOException;
import java.io.Reader;
import java.io.Serializable;

import javax.script.*;

import org.mvel2.MVEL;

public class MvelScriptEngine extends AbstractScriptEngine implements ScriptEngine, Compilable {

    private volatile MvelScriptEngineFactory factory;

    @Override
    public Object eval(String script, ScriptContext context) throws ScriptException {
        Serializable expression = compiledScript(script);
        return evaluate(expression, context);
    }

    @Override
    public Object eval(Reader reader, ScriptContext context) throws ScriptException {
        return this.eval(readFully(reader), context);
    }

    @Override
    public Bindings createBindings() {
        return new MvelBindings();
    }

    @Override
    public ScriptEngineFactory getFactory() {
        if (this.factory == null) {
            synchronized (this) {
                if (this.factory == null) {
                    this.factory = new MvelScriptEngineFactory();
                }
            }
        }

        return this.factory;
    }

    private static String readFully(Reader reader) throws ScriptException {
        char[] arr = new char[8192];
        StringBuilder buf = new StringBuilder();

        int numChars;
        try {
            while ((numChars = reader.read(arr, 0, arr.length)) > 0) {
                buf.append(arr, 0, numChars);
            }
        } catch (IOException var5) {
            throw new ScriptException(var5);
        }

        return buf.toString();
    }

    @Override
    public CompiledScript compile(String script) throws ScriptException {
        return new MvelCompiledScript(this, compiledScript(script));
    }

    @Override
    public CompiledScript compile(Reader reader) throws ScriptException {
        return this.compile(readFully(reader));
    }

    public Serializable compiledScript(String script) throws ScriptException {
        try {
            Serializable expression = MVEL.compileExpression(script);
            return expression;
        } catch (Exception e) {
            throw new ScriptException(e);
        }
    }

    public Object evaluate(Serializable expression, ScriptContext context) throws ScriptException {
        try {
            return MVEL.executeExpression(expression, context.getBindings(ScriptContext.ENGINE_SCOPE));
        } catch (Exception e) {
            throw new ScriptException(e);
        }
    }
}