create table student
(
    id              bigint(21) unsigned auto_increment comment '主鍵id'
        primary key,
    age             int                  null comment '年齡',
    grade           int                  null comment '年級(jí)',
    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 '電話(huà)',
    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 '家庭詳細(xì)住址',
    version         varchar(200)         null comment '版本號(hào)',
    env             varchar(10)          NULL comment '數(shù)據(jù)隔離環(huán)境',
    tenant          bigint               NOT NULL default 0 comment '租戶(hù)標(biāo)識(shí)',
    gmt_created     datetime             null comment '創(chuàng)建時(shí)間',
    gmt_modified    datetime             null comment '更新時(shí)間',
    is_deleted      tinyint(2) default 0 null comment '是否邏輯刪除'
) ENGINE = InnoDB
  DEFAULT CHARSET = utf8
    COMMENT '學(xué)生信息表';

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

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

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

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

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

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

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

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

/**
 * 增刪改查中，環(huán)境和租戶(hù)隔離設(shè)置
 */
public interface IsolateSetter extends IDefaultSetter {
    /**
     * 插入的entity,如果沒(méi)有顯式設(shè)置環(huán)境和租戶(hù)，根據(jù)工具類(lèi)進(jìn)行默認(rèn)設(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());
        }
    }

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

    /**
     * 更新條件追加環(huán)境隔離和租戶(hù)隔離
     *
     * @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默認(rèn)值設(shè)置實(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ù)庫(kù)連接信息 **/
        url = url, username = "root", password = "password",
        /** Entity類(lèi)parent package路徑 **/
        basePack = basePackage,
        /** Entity代碼源目錄 **/
        srcDir = "example/many2many_demo/src/main/java",
        /** 如果表定義記錄創(chuàng)建，記錄修改，邏輯刪除字段 **/
        gmtCreated = "gmt_created", gmtModified = "gmt_modified", logicDeleted = "is_deleted",
        /** 需要生成文件的表 ( 表名稱(chēng):對(duì)應(yīng)的Entity名稱(chēng) ) **/
        tables = @Table(value = {"student"},
            entity = IsolateEntity.class,
            defaults = IsolateSetter.class)
    )
    static class Noting {
    }
}

// 標(biāo)識(shí)對(duì)應(yīng)的Entity類(lèi)需要繼承的接口
entity = IsolateEntity.class        

// 標(biāo)識(shí)對(duì)應(yīng)的Entity類(lèi)CRUD過(guò)程中需要進(jìn)行的默認(rèn)設(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()
        );
    }
}