no message

skyouc

2025-08-07 d52bab61f91056e75e60ddeda76a72cd1bd63615

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
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; 
import org.springframework.context.annotation.Bean; 
import org.springframework.context.annotation.Configuration; 
import org.springframework.security.core.Authentication; 
import org.springframework.security.core.context.SecurityContextHolder; 
import java.util.Arrays; 
 
/** 
 * MybatisPlus配置 
 * 
 * @author vincent 
 * @since 2018-02-22 11:29:28 
 */ 
@Configuration 
public class MybatisPlusConfig { 
 
 
    @Bean 
    public MybatisPlusInterceptor mybatisPlusInterceptor() { 
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor(); 
 
        TenantLineHandler tenantLineHandler = new TenantLineHandler() { 
 
            @Override 
            public Expression getTenantId() { 
                return getLoginUserTenantId(); 
            } 
 
            @Override 
            public String getTenantIdColumn() { 
                return "host_id"; 
            } 
 
            @Override 
            public boolean ignoreTable(String tableName) { 
                return Arrays.asList( 
                        "sys_host", 
                        "sys_user_role", 
                        "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); 
            } 
 
        }; 
        TenantLineInnerInterceptor tenantLineInnerInterceptor = new TenantLineInnerInterceptor(tenantLineHandler); 
        interceptor.addInnerInterceptor(tenantLineInnerInterceptor); 
 
        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor(); 
        interceptor.addInnerInterceptor(paginationInnerInterceptor); 
 
 
        return interceptor; 
    } 
 
    /** 
     * 获取当前登录用户的租户id 
     * 
     * @return Integer 
     */ 
    public Expression getLoginUserTenantId() { 
        try { 
            Authentication authentication = SecurityContextHolder.getContext().getAuthentication(); 
            if (authentication != null) { 
                Object object = authentication.getPrincipal(); 
                if (object instanceof User) { 
                    return new LongValue(((User) object).getHostId()); 
                } 
                if (object instanceof UserLogin) { 
                    return new LongValue(((UserLogin) object).getHostId()); 
                } 
            } else { 
                return  new LongValue(1); 
            } 
        } catch (Exception e) { 
            System.out.println(e.getMessage()); 
        } 
//        return new NullValue(); 
        return new LongValue(1); 
    } 
 
    @Bean 
    public ConfigurationCustomizer mybatisConfigurationCustomizer() { 
        return configuration -> configuration.setObjectWrapperFactory(new MybatisMapWrapperFactory()); 
    } 
 
    @Bean 
    public DataChangeInterceptor sqlStatementInterceptor() { 
        return new DataChangeInterceptor(); 
    } 
 
}