[support qlexpress]

core/pom.xml

@@ -22,8 +22,8 @@
                 <groupId>org.apache.maven.plugins</groupId>
                 <artifactId>maven-compiler-plugin</artifactId>
                 <configuration>
-                    <source>1.6</source>
-                    <target>1.6</target>
+                    <source>8</source>
+                    <target>8</target>
                     <encoding>UTF-8</encoding>
                     <showDeprecation>true</showDeprecation>
                 </configuration>
@@ -222,6 +222,10 @@
             <groupId>ognl</groupId>
             <artifactId>ognl</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.alibaba</groupId>
+            <artifactId>QLExpress4</artifactId>
+        </dependency>
         <dependency>
             <groupId>junit</groupId>
             <artifactId>junit</artifactId>

core/src/main/java/com/taobao/arthas/core/GlobalOptions.java

@@ -11,6 +11,8 @@ public class GlobalOptions {
             + "not allowed to set object properties. "
             + "Want to set object properties, execute `options strict false`";
 
+    public static final String QLEXPRESS_CONFIG = "qlexpress-config";
+
     /**
      * 是否支持系统类<br/>
      * 这个开关打开之后将能代理到来自JVM的部分类，由于有非常强的安全风险可能会引起系统崩溃<br/>
@@ -136,4 +138,26 @@ public class GlobalOptions {
             description = STRICT_MESSAGE
     )
     public static volatile boolean strict = true;
+
+
+    /**
+     * 是否切换使用表达式ognl/qlexpress开关
+     */
+    @Option(level = 1,
+            name = "express-type",
+            summary = "Option to use ognl/qlExpress",
+            description = "Option to use ognl/qlExpress in commands, default ognl"
+    )
+    public static volatile String ExpressType = "ognl";
+
+
+    /**
+     * qlexpress使用参数
+     */
+    @Option(level = 1,
+            name = QLEXPRESS_CONFIG,
+            summary = "config init when use qlExpress, example: {\"\": \"\"} ",
+            description = ""
+    )
+    public static volatile String QLExpressConfig = "";
 }

core/src/main/java/com/taobao/arthas/core/command/express/ExpressFactory.java

@@ -1,32 +1,57 @@
 package com.taobao.arthas.core.command.express;
 
+import com.alibaba.fastjson.JSON;
+import com.taobao.arthas.core.GlobalOptions;
+import com.taobao.arthas.core.command.model.ExpressTypeEnum;
+import com.taobao.arthas.core.command.model.QLExpressConfigModel;
+
 /**
  * ExpressFactory
  * @author ralf0131 2017-01-04 14:40.
  * @author hengyunabc 2018-10-08
  */
 public class ExpressFactory {
-
-    private static final ThreadLocal<Express> expressRef = new ThreadLocal<Express>() {
-        @Override
-        protected Express initialValue() {
-            return new OgnlExpress();
-        }
-    };
+    private static final ThreadLocal<Express> expressRef = ThreadLocal.withInitial(() -> new OgnlExpress());
+    private static final ThreadLocal<Express> expressRefQLExpress = ThreadLocal.withInitial(() -> new QLExpress());
 
     /**
      * get ThreadLocal Express Object
      * @param object
      * @return
      */
     public static Express threadLocalExpress(Object object) {
+        if (GlobalOptions.ExpressType == ExpressTypeEnum.QLEXPRESS.getExpressType()) {
+            return expressRefQLExpress.get().reset().bind(object);
+        }
         return expressRef.get().reset().bind(object);
     }
 
     public static Express unpooledExpress(ClassLoader classloader) {
+        if (classloader == null) {
+            classloader = ClassLoader.getSystemClassLoader();
+        }
+        if (GlobalOptions.ExpressType == ExpressTypeEnum.QLEXPRESS.getExpressType()) {
+            return new QLExpress(classloader);
+        }
+        return new OgnlExpress(new ClassLoaderClassResolver(classloader));
+    }
+
+    public static Express unpooledExpressByOGNL(ClassLoader classloader) {
         if (classloader == null) {
             classloader = ClassLoader.getSystemClassLoader();
         }
         return new OgnlExpress(new ClassLoaderClassResolver(classloader));
     }
+
+    public static boolean checkQLExpressConfig(String configValue) {
+        try {
+            if ("".equals(configValue)) {
+                return true;
+            }
+            JSON.parseObject(configValue, QLExpressConfigModel.class);
+            return true;
+        }catch (Throwable t){
+            return false;
+        }
+    }
 }

core/src/main/java/com/taobao/arthas/core/command/express/QLExpress.java

@@ -0,0 +1,108 @@
+package com.taobao.arthas.core.command.express;
+
+import com.alibaba.arthas.deps.org.slf4j.Logger;
+import com.alibaba.arthas.deps.org.slf4j.LoggerFactory;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.qlexpress4.Express4Runner;
+import com.alibaba.qlexpress4.InitOptions;
+import com.alibaba.qlexpress4.QLOptions;
+import com.taobao.arthas.core.GlobalOptions;
+import com.taobao.arthas.core.command.model.QLExpressConfigModel;
+
+
+/**
+ * @Author TaoKan
+ * @Date 2024/9/17 6:01 PM
+ */
+public class QLExpress implements Express {
+    private static final Logger logger = LoggerFactory.getLogger(QLExpress.class);
+    private Express4Runner expressRunner;
+    private QLGlobalContext qlGlobalContext;
+    private Object runResult;
+
+    private QLExpressConfigModel qlExpressConfigModel;
+
+    private QLOptions qlOptions;
+
+    private InitOptions initOptions;
+
+    public QLExpress() {
+        initQLExpress();
+        initConfig(null);
+        initContext();
+    }
+
+    public QLExpress(ClassLoader classloader) {
+        initQLExpress();
+        initConfig(classloader);
+        initContext();
+    }
+
+    private void initConfig(ClassLoader classloader) {
+        try {
+            QLExpressConfigModel qlExpressConfigModel = JSON.parseObject(GlobalOptions.QLExpressConfig, QLExpressConfigModel.class);
+            this.qlExpressConfigModel = qlExpressConfigModel;
+            QLOptions.Builder qlOptionsBuilder = QLOptions.builder();
+            qlOptionsBuilder.cache(qlExpressConfigModel.isCache());
+            qlOptionsBuilder.avoidNullPointer(qlExpressConfigModel.isAvoidNullPointer());
+            qlOptionsBuilder.maxArrLength(qlExpressConfigModel.getMaxArrLength());
+            qlOptionsBuilder.polluteUserContext(qlExpressConfigModel.isPolluteUserContext());
+            qlOptionsBuilder.precise(qlExpressConfigModel.isPrecise());
+            qlOptionsBuilder.timeoutMillis(qlExpressConfigModel.getTimeoutMillis());
+            qlOptions = qlOptionsBuilder.build();
+
+            InitOptions.Builder initOptionsBuilder = InitOptions.builder();
+            initOptionsBuilder.allowPrivateAccess(qlExpressConfigModel.isAllowPrivateAccess());
+            initOptionsBuilder.debug(qlExpressConfigModel.isDebug());
+            initOptionsBuilder.useCacheClear(qlExpressConfigModel.isUseCacheClear());
+            initOptions = initOptionsBuilder.build();
+            //4.0设置InitOptions
+        }catch (Throwable t){
+            //异常不设置options
+            logger.error("Error Init Options For QLExpress:", t);
+        }
+    }
+
+    private void initQLExpress() {
+        expressRunner = QLExpressRunner.getInstance();
+    }
+
+    private void initContext() {
+        qlGlobalContext = new QLGlobalContext();
+    }
+
+    @Override
+    public Object get(String express) throws ExpressException {
+        try {
+            Object result = expressRunner.execute(express, qlGlobalContext, qlOptions);
+            return result;
+        } catch (Exception e) {
+            logger.error("Error during evaluating the expression with QLExpress:", e);
+            throw new ExpressException(express, e);
+        }
+    }
+
+    @Override
+    public boolean is(String express) throws ExpressException {
+        final Object ret = get(express);
+        return ret instanceof Boolean && (Boolean) ret;
+    }
+
+    @Override
+    public Express bind(Object object) {
+        qlGlobalContext.bindObj(object);
+        return this;
+    }
+
+    @Override
+    public Express bind(String name, Object value) {
+        qlGlobalContext.put(name, value);
+        return this;
+    }
+
+    @Override
+    public Express reset() {
+        qlGlobalContext.clear();
+        return this;
+    }
+}

core/src/main/java/com/taobao/arthas/core/command/express/QLExpressRunner.java

@@ -0,0 +1,34 @@
+package com.taobao.arthas.core.command.express;
+
+
+import com.alibaba.qlexpress4.Express4Runner;
+import com.alibaba.qlexpress4.InitOptions;
+
+/**
+ * @Author TaoKan
+ * @Date 2024/9/22 12:20 PM
+ */
+public class QLExpressRunner {
+    private volatile static QLExpressRunner instance = null;
+    private Express4Runner expressRunner;
+
+    private QLExpressRunner(){
+        expressRunner = new Express4Runner(InitOptions.DEFAULT_OPTIONS);
+    }
+
+    //对外提供静态方法获取对象
+    public static Express4Runner getInstance(){
+        //第一次判断,如果instance不为null,不进入抢锁阶段,直接返回实例
+        if(instance == null){
+            synchronized (QLExpressRunner.class){
+                //抢到锁之后再次进行判断是否为null
+                if(instance == null){
+                    instance = new QLExpressRunner();
+                }
+            }
+        }
+        return instance.expressRunner;
+    }
+
+
+}

core/src/main/java/com/taobao/arthas/core/command/express/QLGlobalContext.java

@@ -0,0 +1,49 @@
+package com.taobao.arthas.core.command.express;
+
+import com.alibaba.qlexpress4.exception.PureErrReporter;
+import com.alibaba.qlexpress4.runtime.ReflectLoader;
+import com.alibaba.qlexpress4.runtime.Value;
+import com.alibaba.qlexpress4.runtime.context.ExpressContext;
+import com.alibaba.qlexpress4.runtime.data.MapItemValue;
+
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * @Author TaoKan
+ * @Date 2024/9/22 12:39 PM
+ */
+public class QLGlobalContext implements ExpressContext {
+    private Map<String, Object> context;
+    private Object object;
+    private ReflectLoader reflectLoader;
+
+    public QLGlobalContext(Map<String, Object> context, Object bindObject, ReflectLoader reflectLoader) {
+        this.context = context;
+        this.object = bindObject;
+        this.reflectLoader = reflectLoader;
+    }
+
+    public QLGlobalContext() {
+        this.context = new ConcurrentHashMap<>();
+    }
+
+    public void put(String name, Object value){
+        context.put(name, value);
+    }
+
+    public void clear() {
+        context.clear();
+    }
+
+    public void bindObj(Object object) {
+        this.object = object;
+    }
+    @Override
+    public Value get(Map<String, Object> attachments, String variableName) {
+        if ((this.reflectLoader != null) && (this.object != null) && !variableName.startsWith("#")) {
+            return this.reflectLoader.loadField(this.object, variableName, true, PureErrReporter.INSTANCE);
+        }
+        return new MapItemValue(this.context, variableName);
+    }
+}

core/src/main/java/com/taobao/arthas/core/command/klass100/OgnlCommand.java

@@ -31,13 +31,13 @@
 @Name("ognl")
 @Summary("Execute ognl expression.")
 @Description(Constants.EXAMPLE
-                + "  ognl '@[email protected](\"hello \\u4e2d\\u6587\")' \n"
-                + "  ognl -x 2 '@Singleton@getInstance()' \n"
-                + "  ognl '@Demo@staticFiled' \n"
-                + "  ognl '#value1=@System@getProperty(\"java.home\"), #value2=@System@getProperty(\"java.runtime.name\"), {#value1, #value2}'\n"
-                + "  ognl -c 5d113a51 '@com.taobao.arthas.core.GlobalOptions@isDump' \n"
-                + Constants.WIKI + Constants.WIKI_HOME + "ognl\n"
-                + "  https://commons.apache.org/proper/commons-ognl/language-guide.html")
+        + "  ognl '@[email protected](\"hello \\u4e2d\\u6587\")' \n"
+        + "  ognl -x 2 '@Singleton@getInstance()' \n"
+        + "  ognl '@Demo@staticFiled' \n"
+        + "  ognl '#value1=@System@getProperty(\"java.home\"), #value2=@System@getProperty(\"java.runtime.name\"), {#value1, #value2}'\n"
+        + "  ognl -c 5d113a51 '@com.taobao.arthas.core.GlobalOptions@isDump' \n"
+        + Constants.WIKI + Constants.WIKI_HOME + "ognl\n"
+        + "  https://commons.apache.org/proper/commons-ognl/language-guide.html")
 public class OgnlCommand extends AnnotatedCommand {
     private static final Logger logger = LoggerFactory.getLogger(OgnlCommand.class);
 
@@ -100,7 +100,7 @@ public void process(CommandProcess process) {
             classLoader = ClassLoader.getSystemClassLoader();
         }
 
-        Express unpooledExpress = ExpressFactory.unpooledExpress(classLoader);
+        Express unpooledExpress = ExpressFactory.unpooledExpressByOGNL(classLoader);
         try {
             Object value = unpooledExpress.get(express);
             OgnlModel ognlModel = new OgnlModel()

core/src/main/java/com/taobao/arthas/core/command/model/ExpressTypeEnum.java

@@ -0,0 +1,21 @@
+package com.taobao.arthas.core.command.model;
+
+/**
+ * @Author TaoKan
+ * @Date 2024/9/22 7:32 AM
+ */
+public enum ExpressTypeEnum
+{
+    OGNL("ognl"),
+    QLEXPRESS("QLExpress");
+
+    private String expressType;
+
+    ExpressTypeEnum(String expressType) {
+        this.expressType = expressType;
+    }
+
+    public String getExpressType() {
+        return expressType;
+    }
+}