#

Junjie

2025-02-14 ac4341ea6b66ae02427d39d35f41d42d78b2eb2e

 zy-asrs-wms/src/main/java/com/zy/asrs/wms/common/config/MybatisPlusConfig.java

@@ -1,10 +1,14 @@
package com.zy.asrs.wms.common.config;

import com.baomidou.mybatisplus.autoconfigure.ConfigurationCustomizer;
import com.baomidou.mybatisplus.extension.MybatisMapWrapperFactory;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
import com.zy.asrs.wms.common.interceptor.DataChangeInterceptor;
import com.zy.asrs.wms.system.entity.User;
import com.zy.asrs.wms.system.entity.UserLogin;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.LongValue;
import net.sf.jsqlparser.expression.NullValue;
@@ -12,7 +16,6 @@
import org.springframework.context.annotation.Configuration;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;

import java.util.Arrays;

/**
@@ -45,7 +48,14 @@
                return Arrays.asList(
                        "sys_host",
                        "sys_user_role",
                        "sys_role_menu"
                        "sys_role_menu",
                        "sys_language",
                        "man_loc_sts",
                        "man_task_sts",
                        "man_task_type",
                        "man_order_settle",
                        "view_man_loc_detl_field",
                        "sys_cache_statistics"
                ).contains(tableName);
            }

@@ -55,6 +65,7 @@

        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
        interceptor.addInnerInterceptor(paginationInnerInterceptor);


        return interceptor;
    }
@@ -72,6 +83,9 @@
                if (object instanceof User) {
                    return new LongValue(((User) object).getHostId());
                }
                if (object instanceof UserLogin) {
                    return new LongValue(((UserLogin) object).getHostId());
                }
            }
        } catch (Exception e) {
            System.out.println(e.getMessage());
@@ -79,4 +93,14 @@
        return new NullValue();
    }

    @Bean
    public ConfigurationCustomizer mybatisConfigurationCustomizer() {
        return configuration -> configuration.setObjectWrapperFactory(new MybatisMapWrapperFactory());
    }

    @Bean
    public DataChangeInterceptor sqlStatementInterceptor() {
        return new DataChangeInterceptor();
    }

}