package org.jeecg.modules.mdc.util;

import org.apache.commons.jexl3.*;
import org.springframework.stereotype.Component;
import java.math.BigDecimal;
import java.util.Map;

/**
 * @Author: Lius
 * @CreateTime: 2025-08-19
 * @Description: 自定义公式计算类
 */
@Component
public class JexlFormulaCalculator {

    // 通过JexlBuilder创建JexlEngine实例
    private static final JexlEngine jexl = new JexlBuilder().create();

    /**
     * 计算自定义公式，除数为0时返回0
     * @param formula 公式字符串，如 "processLong/measureTime*1.0"
     * @param variables 公式中的变量键值对
     * @return 计算结果（BigDecimal类型，除数为0时返回0）
     */
    public BigDecimal calculate(String formula, Map<String, Object> variables) {
        try {
            // 解析公式
            JexlExpression expression = jexl.createExpression(formula);

            // 创建上下文并注入变量
            MapContext context = new MapContext();
            for (Map.Entry<String, Object> entry : variables.entrySet()) {
                String varName = entry.getKey();
                Object value = entry.getValue();
                // 确保变量为数字类型
                if (value instanceof Number) {
                    context.set(varName, value);
                } else {
                    throw new IllegalArgumentException("变量" + varName + "必须是数字类型");
                }
            }

            // 执行计算
            Object result = expression.evaluate(context);

            // 转换为BigDecimal返回
            if (result instanceof Number) {
                return new BigDecimal(result.toString());
            } else {
                throw new RuntimeException("公式计算结果不是数字");
            }
        } catch (JexlException e) {
            // 关键：不依赖内部类，直接通过异常消息判断除数为0
            if (e.getMessage() != null && e.getMessage().contains("division by zero")) {
                return BigDecimal.ZERO; // 除数为0返回0
            }
            // 其他JEXL异常正常抛出
            throw new RuntimeException("公式计算失败: " + formula, e);
        } catch (Exception e) {
            throw new RuntimeException("公式计算失败: " + formula, e);
        }
    }
}