create table student
(
    id              bigint(21) unsigned auto_increment comment '主鍵id'
        primary key,
    age             int                  null comment '年齡',
    grade           int                  null comment '年級',
    user_name       varchar(45)          null comment '名字',
    gender_man      tinyint(2) default 0 null comment '性別, 0:女; 1:男',
    birthday        datetime             null comment '生日',
    phone           varchar(20)          null comment '電話',
    bonus_points    bigint(21) default 0 null comment '積分',
    status          varchar(32)          null comment '狀態(tài)(字典)',
    home_county_id  bigint(21)           null comment '家庭所在區(qū)縣',
    home_address_id bigint(21)           null comment 'home_address外鍵',
    address         varchar(200)         null comment '家庭詳細住址',
    version         varchar(200)         null comment '版本號',
    env             varchar(10)          NULL comment '數(shù)據(jù)隔離環(huán)境',
    tenant          bigint               NOT NULL default 0 comment '租戶標識',
    gmt_created     datetime             null comment '創(chuàng)建時間',
    gmt_modified    datetime             null comment '更新時間',
    is_deleted      tinyint(2) default 0 null comment '是否邏輯刪除'
) ENGINE = InnoDB
  DEFAULT CHARSET = utf8
    COMMENT '學生信息表';

/**
 * 應(yīng)用部署環(huán)境工具類
 */
public class EnvUtils {
    public static String currEnv() {
        // 應(yīng)用啟動時, 讀取的機器部署環(huán)境變量, 這里簡化為返回固定值演示
        return "test1";
    }
}

/**
 * 獲取用戶所屬租戶信息工具類
 */
public class TenantUtils {
    /**
     * 租戶A
     */
    static final long A_TENANT = 111111L;
    /**
     * 租戶B
     */
    static final long B_TENANT = 222222L;

    /**
     * 租戶信息一般根據(jù)登錄用戶身份來判斷, 這里簡化為偶數(shù)用戶屬于租戶A, 奇數(shù)用戶屬于租戶B
     *
     * @return
     */
    public static long findUserTenant() {
        long userId = loginUserId();
        if (userId % 2 == 0) {
            return A_TENANT;
        } else {
            return B_TENANT;
        }
    }

    /**
     * 當前登錄的用戶id, 一般從Session中獲取
     *
     * @return
     */
    public static long loginUserId() {
        return 1L;
    }
}

/**
 * Entity類隔離屬性基類
 */
public interface IsolateEntity {
    /**
     * 返回entity env屬性值
     *
     * @return
     */
    String getEnv();

    /**
     * 設(shè)置entity env屬性值
     *
     * @param env
     * @return
     */
    IsolateEntity setEnv(String env);

    /**
     * 返回entity 租戶信息
     *
     * @return
     */
    Long getTenant();

    /**
     * 設(shè)置entity 租戶信息
     *
     * @param tenant
     * @return
     */
    IsolateEntity setTenant(Long tenant);
}

/**
 * 增刪改查中，環(huán)境和租戶隔離設(shè)置
 */
public interface IsolateSetter extends IDefaultSetter {
    /**
     * 插入的entity,如果沒有顯式設(shè)置環(huán)境和租戶，根據(jù)工具類進行默認設(shè)置
     *
     * @param entity
     */
    @Override
    default void setInsertDefault(IEntity entity) {
        IsolateEntity isolateEntity = (IsolateEntity) entity;
        if (isolateEntity.getEnv() == null) {
            isolateEntity.setEnv(EnvUtils.currEnv());
        }
        if (isolateEntity.getTenant() == null) {
            isolateEntity.setTenant(TenantUtils.findUserTenant());
        }
    }

    /**
     * 查詢條件追加環(huán)境隔離和租戶隔離
     *
     * @param query
     */
    @Override
    default void setQueryDefault(IQuery query) {
        query.where()
            .apply("env", SqlOp.EQ, EnvUtils.currEnv())
            .apply("tenant", SqlOp.EQ, TenantUtils.findUserTenant());
    }

    /**
     * 更新條件追加環(huán)境隔離和租戶隔離
     *
     * @param updater
     */
    @Override
    default void setUpdateDefault(IUpdate updater) {
        updater.where()
            .apply("env", SqlOp.EQ, EnvUtils.currEnv())
            .apply("tenant", SqlOp.EQ, TenantUtils.findUserTenant());
    }
}

public @interface FluentMybatis {
    /**
     * entity, query, updater默認值設(shè)置實現(xiàn)
     *
     * @return
     */
    Class<? extends IDefaultSetter> defaults() default IDefaultSetter.class;
}

public class FluentGenerateMain {
    static final String url = "jdbc:mysql://localhost:3306/fluent_mybatis?useSSL=false&useUnicode=true&characterEncoding=utf-8";
    /**
     * 生成代碼的package路徑
     */
    static final String basePackage = "cn.org.fluent.mybatis.many2many.demo";

    public static void main(String[] args) {
        FileGenerator.build(Noting.class);
    }

    @Tables(
        /** 數(shù)據(jù)庫連接信息 **/
        url = url, username = "root", password = "password",
        /** Entity類parent package路徑 **/
        basePack = basePackage,
        /** Entity代碼源目錄 **/
        srcDir = "example/many2many_demo/src/main/java",
        /** 如果表定義記錄創(chuàng)建，記錄修改，邏輯刪除字段 **/
        gmtCreated = "gmt_created", gmtModified = "gmt_modified", logicDeleted = "is_deleted",
        /** 需要生成文件的表 ( 表名稱:對應(yīng)的Entity名稱 ) **/
        tables = @Table(value = {"student"},
            entity = IsolateEntity.class,
            defaults = IsolateSetter.class)
    )
    static class Noting {
    }
}

// 標識對應(yīng)的Entity類需要繼承的接口
entity = IsolateEntity.class        

// 標識對應(yīng)的Entity類CRUD過程中需要進行的默認設(shè)置操作
defaults = IsolateSetter.class

@FluentMybatis(
    table = "student",
    defaults = IsolateSetter.class
)
public class StudentEntity extends RichEntity implements IsolateEntity {
    // ... 省略
}

@RunWith(SpringRunner.class)
@SpringBootTest(classes = AppMain.class)
public class InsertWithEnvDemo {
    @Autowired
    private StudentMapper mapper;

    @Test
    public void insertEntity() {
        mapper.delete(new StudentQuery());
        mapper.insert(new StudentEntity()
            .setAddress("宇宙深處")
            .setUserName("FluentMybatis")
        );
        StudentEntity student = mapper.findOne(StudentQuery.query()
            .where.userName().eq("FluentMybatis").end()
            .limit(1));
        System.out.println(student.getUserName() + ", env:" + student.getEnv() + ", tenant:" + student.getTenant());
    }
}

@RunWith(SpringRunner.class)
@SpringBootTest(classes = AppMain.class)
public class QueryWithEnvDemo {
    @Autowired
    private StudentMapper mapper;

    @Test
    public void testQueryWithEnv(){
        mapper.delete(new StudentQuery());
        mapper.insert(new StudentEntity()
            .setAddress("宇宙深處")
            .setUserName("FluentMybatis")
        );
        StudentEntity student = mapper.findOne(mapper.defaultQuery()
            .where.userName().eq("FluentMybatis").end()
            .limit(1));
        System.out.println(student.getUserName() + ", env:" + student.getEnv() + ", tenant:" + student.getTenant());
    }
}

@RunWith(SpringRunner.class)
@SpringBootTest(classes = AppMain.class)
public class UpdateWithEnvDemo {
    @Autowired
    private StudentMapper mapper;

    @Test
    public void testQueryWithEnv() {
        mapper.delete(new StudentQuery());
        mapper.insert(new StudentEntity()
            .setAddress("宇宙深處")
            .setUserName("FluentMybatis")
        );
        mapper.updateBy(StudentUpdate.defaultUpdater()
            .update.address().is("回到地球").end()
            .where.userName().eq("FluentMybatis").end()
        );
    }
}