com.jn.sqlhelper.dialect.SqlRequestContextHolder Maven / Gradle / Ivy

Go to download
package com.jn.sqlhelper.dialect;

import com.jn.langx.util.Objs;
import com.jn.langx.util.function.Consumer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


public class SqlRequestContextHolder {
    private static final Logger logger = LoggerFactory.getLogger(SqlRequestContextHolder.class);
    protected static final ThreadLocal variables = new ThreadLocal();

    private static final SqlRequestContextHolder INSTANCE = new SqlRequestContextHolder();

    public static SqlRequestContextHolder getInstance() {
        return INSTANCE;
    }

    public SqlRequestContext get() {
        return variables.get();
    }

    public boolean isOrderByRequest() {
        SqlRequestContext ctx = get();
        return Objs.isNotNull(ctx) && ctx.isOrderByRequest();
    }
    public void setSqlRequest(final SqlRequest request) {
        setContextContent(new Consumer() {
            @Override
            public void accept(SqlRequestContext context) {
                context.setRequest(request);
                request.setContext(context);
            }
        });
    }
    private  void setContextContent(Consumer consumer) {
        SqlRequestContext context = get();
        try {
            if (Objs.isNull(context)) {
                context = SqlRequestContext.class.newInstance();
                if (Objs.isNull(context)) {
                    variables.remove();
                }
            }
            if (Objs.isNotNull(context)) {
                variables.set(context);
                consumer.accept(context);
            }
        } catch (Throwable ex) {
            logger.error(ex.getMessage(), ex);
        }
    }

    public void clear(){
        SqlRequestContext requestContext = get();
        if(Objs.isNotNull(requestContext)){
            requestContext.clear();
        }
        variables.remove();
    }
}