diff --git a/ebike-staff/src/main/java/com/cdzy/staff/config/MyBatisFlexConfiguration.java b/ebike-staff/src/main/java/com/cdzy/staff/config/MyBatisFlexConfiguration.java index 657fb3a..e0b9aa8 100644 --- a/ebike-staff/src/main/java/com/cdzy/staff/config/MyBatisFlexConfiguration.java +++ b/ebike-staff/src/main/java/com/cdzy/staff/config/MyBatisFlexConfiguration.java @@ -1,13 +1,17 @@ package com.cdzy.staff.config; import com.cdzy.common.enums.EbikeContents; +import com.cdzy.staff.component.EbikeTenantFactory; import com.mybatisflex.annotation.KeyType; import com.mybatisflex.core.FlexGlobalConfig; import com.mybatisflex.core.audit.AuditManager; import com.mybatisflex.core.dialect.DbType; import com.mybatisflex.core.keygen.KeyGenerators; +import com.mybatisflex.core.tenant.TenantFactory; +import com.mybatisflex.core.tenant.TenantManager; import jakarta.annotation.PostConstruct; import lombok.extern.slf4j.Slf4j; +import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; @@ -41,4 +45,12 @@ public class MyBatisFlexConfiguration { log.info("MyBatis-Flex 全局配置完成"); } + + @Bean + public TenantFactory tenantFactory() { + log.info("开始初始化租户工厂"); + EbikeTenantFactory factory = new EbikeTenantFactory(); + TenantManager.setTenantFactory(factory); + return factory; + } } \ No newline at end of file