com.alibaba.qlexpress4.runtime.instruction.CheckTimeOutInstruction Maven / Gradle / Ivy

Go to download
package com.alibaba.qlexpress4.runtime.instruction;

import com.alibaba.qlexpress4.QLOptions;
import com.alibaba.qlexpress4.exception.ErrorReporter;
import com.alibaba.qlexpress4.exception.QLErrorCodes;
import com.alibaba.qlexpress4.runtime.QContext;
import com.alibaba.qlexpress4.runtime.QResult;
import com.alibaba.qlexpress4.utils.PrintlnUtils;

import java.util.function.Consumer;

/**
 * Operation: check if program timeout
 * Input: 0
 * Output: 0
 * 
 * Author: DQinYuan
 */
public class CheckTimeOutInstruction extends QLInstruction {
    public CheckTimeOutInstruction(ErrorReporter errorReporter) {
        super(errorReporter);
    }

    @Override
    public QResult execute(QContext qContext, QLOptions qlOptions) {
        if (qlOptions.getTimeoutMillis() <= 0) {
            return QResult.NEXT_INSTRUCTION;
        }
        if (System.currentTimeMillis() - qContext.scriptStartTimeStamp() > qlOptions.getTimeoutMillis()) {
            // timeout
            throw errorReporter.reportFormat(
                    QLErrorCodes.SCRIPT_TIME_OUT.name(),
                    QLErrorCodes.SCRIPT_TIME_OUT.getErrorMsg(),
                    qlOptions.getTimeoutMillis()
            );
        }
        return QResult.NEXT_INSTRUCTION;
    }

    @Override
    public int stackInput() {
        return 0;
    }

    @Override
    public int stackOutput() {
        return 0;
    }

    @Override
    public void println(int index, int depth, Consumer debug) {
        PrintlnUtils.printlnByCurDepth(depth, index + ": CheckTimeout", debug);
    }
}