MybatisPlus 多租户架构(Multi-tenancy)实现详解

select * from user t where t.name like '%tom%' and t.provider_id = 1 ;

<?xml version= "1.0" encoding= "utf-8" ?>

<project xmlns= "http://maven.apache.org/pom/4.0.0" xmlns:xsi= "http://HdhCmsTestw3.org/2001/xmlschema-instance"

      xsi:schemalocation= "http://maven.apache.org/pom/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd" >

   <modelversion> 4.0 . 0 </modelversion>

   <groupid>com.wuwenze</groupid>

   <artifactid>mybatis-plus-multi-tenancy</artifactid>

   <version> 0.0 . 1 -snapshot</version>

   <packaging>jar</packaging>

   <name>mybatis-plus-multi-tenancy</name>

   <description>demo project for spring boot</description>

   <parent>

     <groupid>org.springframework.boot</groupid>

     <artifactid>spring-boot-starter-parent</artifactid>

     <version> 2.1 . 0 .release</version>

     <relativepath/> <!-- lookup parent from repository -->

   </parent>

   <properties>

     <project.build.sourceencoding>utf- 8 </project.build.sourceencoding>

     <project.reporting.outputencoding>utf- 8 </project.reporting.outputencoding>

     <java.version> 1.8 </java.version>

   </properties>

   <dependencies>

     <dependency>

       <groupid>org.springframework.boot</groupid>

       <artifactid>spring-boot-starter</artifactid>

     </dependency>

     <dependency>

       <groupid>org.springframework.boot</groupid>

       <artifactid>spring-boot-starter-test</artifactid>

       <scope>test</scope>

     </dependency>

     <dependency>

       <groupid>org.projectlombok</groupid>

       <artifactid>lombok</artifactid>

       <scope>provided</scope>

     </dependency>

     <dependency>

       <groupid>com.google.guava</groupid>

       <artifactid>guava</artifactid>

       <version> 19.0 </version>

     </dependency>

     <dependency>

       <groupid>com.baomidou</groupid>

       <artifactid>mybatis-plus-boot-starter</artifactid>

       <version> 3.0 . 5 </version>

     </dependency>

     <dependency>

       <groupid>com.baomidou</groupid>

       <artifactid>mybatis-plus</artifactid>

       <version> 3.0 . 5 </version>

     </dependency>

     <dependency>

       <groupid>com.baomidou</groupid>

       <artifactid>mybatis-plus-generator</artifactid>

       <version> 3.0 . 5 </version>

     </dependency>

     <dependency>

       <groupid>com.h2database</groupid>

       <artifactid>h2</artifactid>

     </dependency>

   </dependencies>

   <build>

     <plugins>

       <plugin>

         <groupid>org.springframework.boot</groupid>

         <artifactid>spring-boot-maven-plugin</artifactid>

       </plugin>

     </plugins>

   </build>

</project>

spring:

  datasource:

   driver- class -name: org.h2.driver

   schema: classpath:db/schema.sql

   data: classpath:db/data.sql

   url: jdbc:h2:mem:test

   username: root

   password: test

logging:

  level:

   com.wuwenze.mybatisplusmultitenancy: debug

#schema.sql

drop table if exists user;

create table user

(

   id bigint( 20 ) not null comment '主键' ,

   provider_id bigint( 20 ) not null comment '服务商id' ,

   name varchar( 30 ) null default null comment '姓名' ,

   primary key (id)

);

#data.sql

insert into user (id, provider_id, name) values ( 1 , 1 , 'tony老师' );

insert into user (id, provider_id, name) values ( 2 , 1 , 'william老师' );

insert into user (id, provider_id, name) values ( 3 , 2 , '路人甲' );

insert into user (id, provider_id, name) values ( 4 , 2 , '路人乙' );

insert into user (id, provider_id, name) values ( 5 , 2 , '路人丙' );

insert into user (id, provider_id, name) values ( 6 , 2 , '路人丁' );

@configuration

@mapperscan ( "com.wuwenze.mybatisplusmultitenancy.mapper" )

public class mybatisplusconfig {

   private static final string system_tenant_id = "provider_id" ;

   private static final list<string> ignore_tenant_tables = lists.newarraylist( "provider" );

   @autowired

   private apicontext apicontext;

   @bean

   public paginationinterceptor paginationinterceptor() {

     paginationinterceptor paginationinterceptor = new paginationinterceptor();

     // sql解析处理拦截：增加租户处理回调。

     tenantsqlparser tenantsqlparser = new tenantsqlparser()

         .settenanthandler( new tenanthandler() {

           @override

           public expression gettenantid() {

             // 从当前系统上下文中取出当前请求的服务商id，通过解析器注入到sql中。

             long currentproviderid = apicontext.getcurrentproviderid();

             if ( null == currentproviderid) {

               throw new runtimeexception( "#1129 getcurrentproviderid error." );

             }

             return new longvalue(currentproviderid);

           }

           @override

           public string gettenantidcolumn() {

             return system_tenant_id;

           }

           @override

           public boolean dotablefilter(string tablename) {

             // 忽略掉一些表：如租户表（provider）本身不需要执行这样的处理。

             return ignore_tenant_tables.stream().anymatch((e) -> e.equalsignorecase(tablename));

           }

         });

     paginationinterceptor.setsqlparserlist(lists.newarraylist(tenantsqlparser));

     return paginationinterceptor;

   }

   @bean (name = "performanceinterceptor" )

   public performanceinterceptor performanceinterceptor() {

     return new performanceinterceptor();

   }

}

@component

public class apicontext {

   private static final string key_current_provider_id = "key_current_provider_id" ;

   private static final map<string, object> mcontext = maps.newconcurrentmap();

   public void setcurrentproviderid( long providerid) {

     mcontext.put(key_current_provider_id, providerid);

   }

   public long getcurrentproviderid() {

     return ( long ) mcontext.get(key_current_provider_id);

   }

}

@data

@tostring

@accessors (chain = true )

public class user {

   private long id;

   private long providerid;

   private string name;

}

public interface usermapper extends basemapper<user> {

}

@slf4j

@runwith (springrunner. class )

@fixmethodorder (methodsorters.jvm)

@springboottest (classes = mybatisplusmultitenancyapplication. class )

public class mybatisplusmultitenancyapplicationtests {

   @autowired

   private apicontext apicontext;

   @autowired

   private usermapper usermapper;

   @before

   public void before() {

     // 在上下文中设置当前服务商的id

     apicontext.setcurrentproviderid(1l);

   }

   @test

   public void insert() {

     user user = new user().setname( "新来的tom老师" );

     assert .asserttrue(usermapper.insert(user) > 0 );

     user = usermapper.selectbyid(user.getid());

     log.info( "#insert user={}" , user);

     // 检查插入的数据是否自动填充了租户id

     assert .assertequals(apicontext.getcurrentproviderid(), user.getproviderid());

   }

   @test

   public void selectlist() {

     usermapper.selectlist( null ).foreach((e) -> {

       log.info( "#selectlist, e={}" , e);

       // 验证查询的数据是否超出范围

       assert .assertequals(apicontext.getcurrentproviderid(), e.getproviderid());

     });

   }

}

2018 - 11 - 29 21 : 07 : 14.262 info 18688 --- [      main] .mybatisplusmultitenancyapplicationtests : started mybatisplusmultitenancyapplicationtests in 2.629 seconds (jvm running for 3.904 )

2018 - 11 - 29 21 : 07 : 14.554 debug 18688 --- [      main] c.w.m.mapper.usermapper.insert      : ==> preparing: insert into user (id, name, provider_id) values (?, ?, 1 )

2018 - 11 - 29 21 : 07 : 14.577 debug 18688 --- [      main] c.w.m.mapper.usermapper.insert      : ==> parameters: 1068129257418178562 ( long ), 新来的tom老师(string)

2018 - 11 - 29 21 : 07 : 14.577 debug 18688 --- [      main] c.w.m.mapper.usermapper.insert      : <==  updates: 1

  time： 0 ms - id：com.wuwenze.mybatisplusmultitenancy.mapper.usermapper.insert

execute sql：insert into user (id, name, provider_id) values (?, ?, 1 ) { 1 : 1068129257418178562 , 2 : stringdecode( '\u65b0\u6765\u7684tom\u8001\u5e08' )}

2018 - 11 - 29 21 : 07 : 14.585 debug 18688 --- [      main] c.w.m.mapper.usermapper.selectbyid    : ==> preparing: select id, provider_id, name from user where user.provider_id = 1 and id = ?

2018 - 11 - 29 21 : 07 : 14.595 debug 18688 --- [      main] c.w.m.mapper.usermapper.selectbyid    : ==> parameters: 1068129257418178562 ( long )

2018 - 11 - 29 21 : 07 : 14.614 debug 18688 --- [      main] c.w.m.mapper.usermapper.selectbyid    : <==   total: 1

2018 - 11 - 29 21 : 07 : 14.615 info 18688 --- [      main] .mybatisplusmultitenancyapplicationtests : #insert user=user(id= 1068129257418178562 , providerid= 1 , name=新来的tom老师)

  time： 19 ms - id：com.wuwenze.mybatisplusmultitenancy.mapper.usermapper.selectbyid

execute sql：select id, provider_id, name from user where user.provider_id = 1 and id = ? { 1 : 1068129257418178562 }

2018 - 11 - 29 21 : 07 : 14.626 debug 18688 --- [      main] c.w.m.mapper.usermapper.selectlist    : ==> preparing: select id, provider_id, name from user where user.provider_id = 1

  time： 0 ms - id：com.wuwenze.mybatisplusmultitenancy.mapper.usermapper.selectlist

execute sql：select id, provider_id, name from user where user.provider_id = 1

2018 - 11 - 29 21 : 07 : 14.629 debug 18688 --- [      main] c.w.m.mapper.usermapper.selectlist    : ==> parameters:

2018 - 11 - 29 21 : 07 : 14.630 debug 18688 --- [      main] c.w.m.mapper.usermapper.selectlist    : <==   total: 3

2018 - 11 - 29 21 : 07 : 14.632 info 18688 --- [      main] .mybatisplusmultitenancyapplicationtests : #selectlist, e=user(id= 1 , providerid= 1 , name=tony老师)

2018 - 11 - 29 21 : 07 : 14.632 info 18688 --- [      main] .mybatisplusmultitenancyapplicationtests : #selectlist, e=user(id= 2 , providerid= 1 , name=william老师)

2018 - 11 - 29 21 : 07 : 14.632 info 18688 --- [      main] .mybatisplusmultitenancyapplicationtests : #selectlist, e=user(id= 1068129257418178562 , providerid= 1 , name=新来的tom老师)