001/*
002 *  Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com).
003 *  <p>
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *  <p>
008 *  http://www.apache.org/licenses/LICENSE-2.0
009 *  <p>
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016package com.mybatisflex.core.logicdelete;
017
018import com.mybatisflex.core.logicdelete.impl.DefaultLogicDeleteProcessor;
019
020import java.util.function.Supplier;
021
022/**
023 * 逻辑删除管理器。
024 */
025public class LogicDeleteManager {
026
027    private LogicDeleteManager() {
028    }
029
030    private static LogicDeleteProcessor processor = new DefaultLogicDeleteProcessor();
031    private static final ThreadLocal<Boolean> skipFlags = new ThreadLocal<>();
032
033    /**
034     * 获取逻辑删除处理器。
035     *
036     * @return 逻辑删除处理器
037     */
038    public static LogicDeleteProcessor getProcessor() {
039        return processor;
040    }
041
042    /**
043     * 设置逻辑删除处理器。
044     *
045     * @param processor 逻辑删除处理器
046     */
047    public static void setProcessor(LogicDeleteProcessor processor) {
048        LogicDeleteManager.processor = processor;
049    }
050
051    /**
052     * 跳过逻辑删除字段处理，直接进行数据库物理操作。
053     */
054    public static <T> T execWithoutLogicDelete(Supplier<T> supplier) {
055        try {
056            skipLogicDelete();
057            return supplier.get();
058        } finally {
059            restoreLogicDelete();
060        }
061    }
062
063    /**
064     * 跳过逻辑删除字段处理。
065     */
066    public static void skipLogicDelete() {
067        skipFlags.set(Boolean.TRUE);
068    }
069
070    /**
071     * 恢复逻辑删除字段处理。
072     */
073    public static void restoreLogicDelete() {
074        skipFlags.remove();
075    }
076
077    /**
078     * 获取逻辑删除列，返回 {@code null} 表示跳过逻辑删除。
079     *
080     * @param logicDeleteColumn 逻辑删除列
081     * @return 逻辑删除列
082     */
083    public static String getLogicDeleteColumn(String logicDeleteColumn) {
084        if (logicDeleteColumn == null) {
085            return null;
086        }
087        Boolean skipFlag = skipFlags.get();
088        if (skipFlag == null) {
089            return logicDeleteColumn;
090        }
091        return skipFlag ? null : logicDeleteColumn;
092    }
093
094}