package com.example;

import cn.com.digitalhainan.mybatis.mapper.BaseMapper;
import cn.com.digitalhainan.mybatis.model.BaseDO;
import cn.com.digitalhainan.mybatis.utils.PageUtils;

/**
 * 验证 digitalhainan-mybatis 集成
 * 编译通过即表示集成成功
 */
public class MybatisIntegrationVerify {
    // BaseMapper 可用 → 集成成功
    // BaseDO 可用 → 自动填充能力就绪
    // PageUtils 可用 → 分页转换能力就绪
}

package com.example.entity;

import cn.com.digitalhainan.mybatis.model.BaseDO;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.EqualsAndHashCode;

/**
 * 示例实体类，继承 BaseDO 后自动具备：
 * - createTime: 插入时自动填充当前时间
 * - updateTime: 插入/更新时自动填充当前时间
 * - creator: 插入时从 MDC("_U_ID_") 获取当前用户 ID
 * - updator: 更新时从 MDC("_U_ID_") 获取当前用户 ID
 * - disable: 逻辑删除标记（插入时默认 false）
 * - version: 乐观锁版本号（插入时默认 1）
 *
 * 对应数据库表需包含以下字段：
 * - create_time DATETIME
 * - update_time DATETIME
 * - creator VARCHAR(64)
 * - updator VARCHAR(64)
 * - disable TINYINT(1) DEFAULT 0
 * - version INT DEFAULT 1
 */
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("t_example")
public class ExampleDO extends BaseDO {
    private Long id;
    private String name;
    private String description;
}

package com.example.mapper;

import cn.com.digitalhainan.mybatis.mapper.BaseMapper;
import com.example.entity.ExampleDO;
import org.apache.ibatis.annotations.Mapper;
import java.util.List;

/**
 * 继承 BaseMapper 后具备：
 * - MyBatis Plus 全部 CRUD 方法
 * - MyBatis Plus Join 联表查询能力（MPJBaseMapper）
 * - insertList: 批量插入所有字段
 * - selectAll: 查询所有（包括逻辑删除的记录）
 * - batchUpdate: 根据主键批量更新所有字段
 * - batchUpdateSelective: 根据主键批量更新非空字段
 */
@Mapper
public interface ExampleMapper extends BaseMapper<ExampleDO> {
    // 自定义 SQL 方法
}

package com.example.service;

import cn.com.digitalhainan.common.common.dto.PaginationInfoResDTO;
import cn.com.digitalhainan.mybatis.utils.PageUtils;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.example.entity.ExampleDO;
import com.example.vo.ExampleVO;
import com.example.converter.ExampleConverter;
import org.springframework.stereotype.Service;

/**
 * PageUtils 使用示例
 * 将 MyBatis Plus 的 IPage 或 PageHelper 的 Page 转为统一的 PaginationInfoResDTO
 */
@Service
public class ExampleService {

    // 方式一：使用 BaseConverter 转换（推荐）
    public PaginationInfoResDTO<ExampleVO> getPageWithConverter(
            IPage<ExampleDO> page, ExampleConverter converter) {
        return PageUtils.transform(page, converter);
    }

    // 方式二：使用 Function 转换
    public PaginationInfoResDTO<ExampleVO> getPageWithFunction(
            IPage<ExampleDO> page) {
        return PageUtils.transform(page, records -> {
            // 手动转换逻辑
            return records.stream().map(this::toVO).toList();
        });
    }

    // 方式三：PageHelper 分页结果
    public PaginationInfoResDTO<ExampleVO> getPageHelperResult(
            java.util.List<ExampleDO> list, ExampleConverter converter) {
        return PageUtils.transform(list, converter);
    }

    private ExampleVO toVO(ExampleDO entity) {
        ExampleVO vo = new ExampleVO();
        vo.setId(entity.getId());
        vo.setName(entity.getName());
        return vo;
    }
}

package com.example.service;

import cn.com.digitalhainan.mybatis.data.permission.annotation.DataColumn;
import cn.com.digitalhainan.mybatis.data.permission.annotation.DataPermission;
import cn.com.digitalhainan.mybatis.data.permission.constants.JoinEnum;
import com.example.mapper.ExampleMapper;
import org.springframework.stereotype.Service;

/**
 * 数据权限使用示例
 *
 * 系统内置角色 SQL 模板：
 * - ADMIN: 无限制（管理员看所有数据）
 * - SELF: creator = 当前用户ID
 * - MAIN_DEPT: dept_id = 用户主部门ID
 * - MAIN_DEPT_TREE: dept_id IN (用户主部门及子部门ID列表)
 *
 * 数据权限基于用户角色自动匹配 SQL 模板，通过 SpEL 表达式动态拼接 WHERE 条件
 */
@Service
public class ExampleService {

    private final ExampleMapper exampleMapper;

    public ExampleService(ExampleMapper exampleMapper) {
        this.exampleMapper = exampleMapper;
    }

    // 示例 1：默认数据权限（按 creator 字段过滤）
    @DataPermission
    public void listByDefault() {
        exampleMapper.selectList(null);
    }

    // 示例 2：指定数据权限字段
    @DataPermission({
        @DataColumn(key = "creator", value = "create_by"),
        @DataColumn(key = "deptId", value = "dept_id")
    })
    public void listByCustomColumn() {
        exampleMapper.selectList(null);
    }

    // 示例 3：多条件 AND 组合
    @DataPermission(
        value = {
            @DataColumn(key = "creator", value = "create_by"),
            @DataColumn(key = "deptId", value = "dept_id")
        },
        mixRoleJoin = JoinEnum.AND
    )
    public void listWithAndJoin() {
        exampleMapper.selectList(null);
    }
}

package com.example.config;

import cn.com.digitalhainan.mybatis.data.permission.conf.DataPermissionCustomizer;
import cn.com.digitalhainan.mybatis.data.permission.conf.SqlTempCodeConfiguration;
import lombok.AllArgsConstructor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.List;

/**
 * 自定义数据权限 SQL 模板
 * 实现 SqlTempCodeConfiguration 接口定义新的角色编码和 SQL 模板
 */
@Configuration
public class DataPermissionConfig {

    @Bean
    public DataPermissionCustomizer customDataPermission() {
        return sqlTemps -> {
            sqlTemps.add(new SqlTempCodeConfiguration() {
                @Override
                public String code() {
                    return "CUSTOM_ROLE";  // 角色编码，需与用户系统中的角色匹配
                }
                @Override
                public String sqlTemp() {
                    // SpEL 模板，支持 #userAccount、#creator、#deptId 等变量
                    return "#{#creator} = '#{#userAccount.getOriUserInfo().userId}'";
                }
            });
        };
    }
}

package com.example.service;

import com.baomidou.dynamic.datasource.annotation.DS;
import com.example.mapper.ExampleMapper;
import com.example.mapper.SlaveMapper;
import org.springframework.stereotype.Service;

/**
 * 多数据源切换示例
 * 使用 @DS 注解指定数据源名称（对应 application.yml 中 spring.datasource.dynamic.datasource 下的 key）
 */
@Service
public class MultiDatasourceService {

    private final ExampleMapper exampleMapper;

    public MultiDatasourceService(ExampleMapper exampleMapper) {
        this.exampleMapper = exampleMapper;
    }

    // 默认使用主数据源
    public void queryFromMaster() {
        exampleMapper.selectList(null);
    }

    // 切换到从库
    @DS("slave")
    public void queryFromSlave() {
        exampleMapper.selectList(null);
    }
}

package com.example.service;

import cn.com.digitalhainan.mybatis.datasouce.rws.context.RWSContextHolder;
import com.example.mapper.ExampleMapper;
import org.springframework.stereotype.Service;

/**
 * 读写分离场景：某些查询即使在从库可用的情况下也需要强制走主库
 */
@Service
public class RWSService {

    private final ExampleMapper exampleMapper;

    public RWSService(ExampleMapper exampleMapper) {
        this.exampleMapper = exampleMapper;
    }

    public void forceReadFromMaster() {
        // 强制走主库（调用后自动清除，仅对下一次查询生效）
        RWSContextHolder.startEnforceMaster();
        exampleMapper.selectList(null);
    }
}

package com.example.entity;

import cn.com.digitalhainan.mybatis.handlers.SM4EncryptHandle;
import cn.com.digitalhainan.mybatis.model.BaseDO;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.EqualsAndHashCode;

/**
 * SM4 字段级加解密示例
 * 数据写入时自动加密，读取时自动解密
 */
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("t_sensitive")
public class SensitiveDO extends BaseDO {
    private Long id;

    @TableField(typeHandler = SM4EncryptHandle.class)
    private String idCard;  // 身份证号，数据库中存储为 SM4 密文

    @TableField(typeHandler = SM4EncryptHandle.class)
    private String phone;   // 手机号，数据库中存储为 SM4 密文
}