first commit

play-admin/src/main/java/com/starry/admin/common/mybatis/config/MybatisPlusConfig.java

@@ -0,0 +1,79 @@
+package com.starry.admin.common.mybatis.config;
+
+import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
+import com.baomidou.mybatisplus.annotation.DbType;
+import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.inner.OptimisticLockerInnerInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
+import com.starry.admin.common.mybatis.handler.MyTenantLineHandler;
+import org.mybatis.spring.annotation.MapperScan;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.jdbc.datasource.DataSourceTransactionManager;
+import org.springframework.transaction.annotation.EnableTransactionManagement;
+
+import javax.sql.DataSource;
+
+
+/**
+ * @author admin
+ */
+@Configuration
+@MapperScan("com.starry.**.mapper")
+@EnableTransactionManagement
+public class MybatisPlusConfig {
+
+    /**
+     * druid注入
+     *
+     * @return dataSource
+     */
+    @Bean(name = "dataSource")
+    @ConfigurationProperties(prefix = "spring.datasource.druid")
+    public DataSource dataSource() {
+        return DruidDataSourceBuilder.create().build();
+    }
+
+    /**
+     * 配置事物管理器
+     *
+     * @return DataSourceTransactionManager
+     */
+    @Bean(name = "transactionManager")
+    public DataSourceTransactionManager transactionManager() {
+        return new DataSourceTransactionManager(dataSource());
+    }
+
+    @Bean
+    public MybatisPlusInterceptor mybatisPlusInterceptor() {
+        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
+        // 租户插件
+        interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new MyTenantLineHandler()));
+        // 分页插件
+        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
+        // 乐观锁插件
+        interceptor.addInnerInterceptor(optimisticLockerInnerInterceptor());
+        // 阻断插件
+        interceptor.addInnerInterceptor(blockAttackInnerInterceptor());
+        return interceptor;
+    }
+
+    /**
+     * 乐观锁
+     */
+    @Bean
+    public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor() {
+        return new OptimisticLockerInnerInterceptor();
+    }
+
+    /**
+     * 防止全表更新与删除
+     */
+    @Bean
+    public BlockAttackInnerInterceptor blockAttackInnerInterceptor() {
+        return new BlockAttackInnerInterceptor();
+    }
+}

play-admin/src/main/java/com/starry/admin/common/mybatis/handler/MyMetaObjectHandler.java

@@ -0,0 +1,45 @@
+package com.starry.admin.common.mybatis.handler;
+
+import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
+import com.starry.admin.utils.SecurityUtils;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.ibatis.reflection.MetaObject;
+import org.springframework.stereotype.Component;
+
+import java.util.Date;
+
+/**
+ * @author admin
+ * 字段默认值处理类
+ * @since 2021/9/1
+ */
+@Slf4j
+@Component
+public class MyMetaObjectHandler implements MetaObjectHandler {
+
+    @Override
+    public void insertFill(MetaObject metaObject) {
+        log.info("start insert fill ....");
+        this.setFieldValByName("createdTime", new Date(), metaObject);
+        this.setFieldValByName("deleted", false, metaObject);
+        this.setFieldValByName("version", 1L, metaObject);
+        Object createUser = this.getFieldValByName("createdBy", metaObject);
+        if (createUser == null) {
+            if (SecurityUtils.isLogin()) {
+                this.setFieldValByName("createdBy", SecurityUtils.getUserId(), metaObject);
+            }
+        }
+    }
+
+    @Override
+    public void updateFill(MetaObject metaObject) {
+        log.info("start update fill ....");
+        this.setFieldValByName("updatedTime", new Date(), metaObject);
+        Object createUser = this.getFieldValByName("updatedBy", metaObject);
+        if (createUser == null) {
+            if (SecurityUtils.isLogin()) {
+                this.setFieldValByName("updatedBy", SecurityUtils.getUserId(), metaObject);
+            }
+        }
+    }
+}

play-admin/src/main/java/com/starry/admin/common/mybatis/handler/MyTenantLineHandler.java

@@ -0,0 +1,54 @@
+package com.starry.admin.common.mybatis.handler;
+
+import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
+import com.starry.admin.utils.SecurityUtils;
+import com.starry.common.utils.StringUtils;
+import lombok.extern.slf4j.Slf4j;
+import net.sf.jsqlparser.expression.Expression;
+import net.sf.jsqlparser.expression.LongValue;
+import net.sf.jsqlparser.expression.NullValue;
+import org.springframework.stereotype.Component;
+
+import java.util.Arrays;
+
+/**
+ * @author admin
+ * 多租户处理器
+ * @since 2023/3/7
+ */
+@Slf4j
+@Component
+public class MyTenantLineHandler implements TenantLineHandler {
+
+    /**
+     * 排除过滤的表
+     */
+    private static final String[] TABLE_FILTER = {"sys_menu", "sys_tenant_package", "sys_tenant", "sys_dict", "sys_dict_data"};
+
+    /**
+     * 排除过滤的表前缀
+     */
+    private static final String[] TABLE_PRE = {"qrtz", "gen"};
+
+    @Override
+    public Expression getTenantId() {
+        // 取出当前请求的服务商ID，通过解析器注入到SQL中。
+        Long tenantId = SecurityUtils.getTenantId();
+        if (tenantId == null) {
+            return new NullValue();
+        }
+        return new LongValue(tenantId);
+    }
+
+    /**
+     * 跳过不需要加多租户的表
+     */
+    @Override
+    public boolean ignoreTable(String tableName) {
+        String prefix = StringUtils.substringBefore(tableName, "_");
+        if (Arrays.asList(TABLE_FILTER).contains(tableName) || Arrays.asList(TABLE_PRE).contains(prefix)) {
+            return true;
+        }
+        return false;
+    }
+}