org.rx.bean.ProceedEventArgs Maven / Gradle / Ivy

Go to download
package org.rx.bean;

import io.netty.util.concurrent.FastThreadLocal;
import lombok.Getter;
import lombok.NonNull;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import org.rx.core.EventArgs;
import org.rx.util.Snowflake;
import org.rx.util.function.Func;

import java.util.Set;

@Getter
@RequiredArgsConstructor
public class ProceedEventArgs extends EventArgs {
    private static final long serialVersionUID = -2969747570419733673L;
    static final FastThreadLocal TS_TRACE_ID = new FastThreadLocal() {
        @Override
        protected Long initialValue() throws Exception {
            return Snowflake.DEFAULT.nextId();
        }
    };

    private final Class declaringType;
    private final Object[] parameters;
    private final boolean isVoid;

    private Object returnValue;
    private long elapsedMillis = -1;
    @Setter
    private Throwable error;
    @Setter
    private LogStrategy logStrategy;
    @Setter
    private Set logTypeWhitelist;

    public Long getTraceId() {
        return TS_TRACE_ID.get();
    }

    public  T proceed(@NonNull Func proceed) throws Throwable {
        long start = System.currentTimeMillis();
        try {
            T retVal = proceed.invoke();
            returnValue = retVal;
            return retVal;
        } finally {
            elapsedMillis = System.currentTimeMillis() - start;
        }
    }
}