Java IntrospectedTable.getBaseRecordType方法代码示例

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
private GeneratedJavaFile generateService(IntrospectedTable table) {
    // 获取实体类型
    FullyQualifiedJavaType entityType = new FullyQualifiedJavaType(table.getBaseRecordType());
    // 获取主键类型
    FullyQualifiedJavaType primaryType = table.getPrimaryKeyColumns().get(0).getFullyQualifiedJavaType();
    // 生成 Service 名称
    String service = targetPackage + "." + table.getFullyQualifiedTable().getDomainObjectName() + "Service";
    // 构造 Service 文件
    Interface interfaze = new Interface(new FullyQualifiedJavaType(service));
    // 设置作用域
    interfaze.setVisibility(JavaVisibility.PUBLIC);
    // import
    interfaze.addImportedType(new FullyQualifiedJavaType(baseService));
    interfaze.addImportedType(entityType);
    interfaze.addSuperInterface(new FullyQualifiedJavaType(
            baseService + "<" + entityType.getShortName() + "," + primaryType.getShortName() + ">"));
    ElementHelper.addAuthorTag(interfaze, false);
    return new GeneratedJavaFile(interfaze, targetProject, new DefaultJavaFormatter());
}
 

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
private GeneratedJavaFile generateServiceImpl(IntrospectedTable table) {
    FullyQualifiedJavaType entityType = new FullyQualifiedJavaType(table.getBaseRecordType());
    FullyQualifiedJavaType primaryType = table.getPrimaryKeyColumns().get(0).getFullyQualifiedJavaType();
    String domainObjectName = table.getFullyQualifiedTable().getDomainObjectName();
    String service = targetPackage + "." + domainObjectName + "Service";
    String serviceImpl = targetPackage + ".impl." + domainObjectName + "ServiceImpl";
    TopLevelClass clazz = new TopLevelClass(new FullyQualifiedJavaType(serviceImpl));
    clazz.addImportedType(new FullyQualifiedJavaType(baseServiceImpl));
    clazz.addImportedType(entityType);
    clazz.addImportedType(new FullyQualifiedJavaType(service));
    clazz.addImportedType(new FullyQualifiedJavaType("org.springframework.stereotype.Service"));
    clazz.addAnnotation("@Service(\"" + firstLetterLowerCase(domainObjectName + "Service") + "\")");
    clazz.setVisibility(JavaVisibility.PUBLIC);
    clazz.setSuperClass(new FullyQualifiedJavaType(
            baseServiceImpl + "<" + entityType.getShortName() + "," + primaryType.getShortName() + ">"));
    clazz.addSuperInterface(new FullyQualifiedJavaType(service));
    ElementHelper.addAuthorTag(clazz, false);
    return new GeneratedJavaFile(clazz, targetProject, new DefaultJavaFormatter());
}
 

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
/**
 * 生成的 Mapper 接口
 */
@Override
public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
    // 获取实体类
    FullyQualifiedJavaType entityType = new FullyQualifiedJavaType(introspectedTable.getBaseRecordType());
    // import 实体类
    interfaze.addImportedType(entityType);
    // import 接口
    interfaze.addImportedType(new FullyQualifiedJavaType("org.apache.ibatis.annotations.Mapper"));
    // 添加 @Mapper 注解
    interfaze.addAnnotation("@Mapper");
    for (String mapper : mappers) {
        // import mappers
        interfaze.addImportedType(new FullyQualifiedJavaType(mapper));
        // 添加父类
        interfaze.addSuperInterface(new FullyQualifiedJavaType(mapper + "<" + entityType.getShortName() + ">"));
    }
    interfaze.addJavaDocLine("/**");
    interfaze.addJavaDocLine(" * " + interfaze.getType().getShortName());
    ElementHelper.addAuthorTag(interfaze, false);
    interfaze.addJavaDocLine(" */");
    return true;
}
 

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
/**
 * 生成的Mapper接口
 *
 * @param interfaze
 * @param topLevelClass
 * @param introspectedTable
 * @return
 */
@Override
public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
    //获取实体类
    FullyQualifiedJavaType entityType = new FullyQualifiedJavaType(introspectedTable.getBaseRecordType());
    //主键
    FullyQualifiedJavaType idType = null;
    List<IntrospectedColumn> columns = introspectedTable.getPrimaryKeyColumns();
    for (IntrospectedColumn col : columns) {
        idType = javaTypeResolver.calculateJavaType(col);
        break;
    }
    //import接口
    for (String mapper : mappers) {
        interfaze.addImportedType(new FullyQualifiedJavaType(mapper));
        interfaze.addSuperInterface(
                new FullyQualifiedJavaType(mapper + "<" + entityType.getShortName() + "," + idType.getShortName() + ">"));
    }
    //import实体类
    interfaze.addImportedType(entityType);
    return true;
}
 

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
/**
 * 生成的Mapper接口
 *
 * @param interfaze
 * @param topLevelClass
 * @param introspectedTable
 * @return
 */
@Override
public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
    //获取实体类
    FullyQualifiedJavaType entityType = new FullyQualifiedJavaType(introspectedTable.getBaseRecordType());
    //import接口
    for (String mapper : mappers) {
        interfaze.addImportedType(new FullyQualifiedJavaType(mapper));
        interfaze.addSuperInterface(new FullyQualifiedJavaType(mapper + "<" + entityType.getShortName() + ">"));
    }
    //import实体类
    interfaze.addImportedType(entityType);
    return true;
}
 

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
@Override
public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
    FullyQualifiedJavaType fqjt = new FullyQualifiedJavaType("BaseMapper<" + introspectedTable.getBaseRecordType() + ">");
    FullyQualifiedJavaType imp = new FullyQualifiedJavaType("com.zhu8fei.framework.core.mybatis.mapper.BaseMapper");
    FullyQualifiedJavaType reposImp = new FullyQualifiedJavaType("org.springframework.stereotype.Repository");
    interfaze.addSuperInterface(fqjt);// 添加 extends BaseDao<User>
    interfaze.addImportedType(imp);// 添加import common.BaseDao;
    interfaze.addImportedType(reposImp);
    interfaze.addAnnotation("@Repository");
    return true;
}
 

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
/**
 * 获取Model没有BLOBs类时的类型
 * @param introspectedTable
 * @return
 */
public static FullyQualifiedJavaType getModelTypeWithoutBLOBs(IntrospectedTable introspectedTable) {
    FullyQualifiedJavaType type;
    if (introspectedTable.getRules().generateBaseRecordClass()) {
        type = new FullyQualifiedJavaType(introspectedTable.getBaseRecordType());
    } else if (introspectedTable.getRules().generatePrimaryKeyClass()) {
        type = new FullyQualifiedJavaType(introspectedTable.getPrimaryKeyType());
    } else {
        throw new RuntimeException(getString("RuntimeError.12"));
    }
    return type;
}
 

import org.mybatis.generator.api.IntrospectedTable; //导入方法依赖的package包/类
/**
 * 获取Model有BLOBs类时的类型
 * @param introspectedTable
 * @return
 */
public static FullyQualifiedJavaType getModelTypeWithBLOBs(IntrospectedTable introspectedTable) {
    FullyQualifiedJavaType type;
    if (introspectedTable.getRules().generateRecordWithBLOBsClass()) {
        type = new FullyQualifiedJavaType(introspectedTable.getRecordWithBLOBsType());
    } else {
        // the blob fields must be rolled up into the base class
        type = new FullyQualifiedJavaType(introspectedTable.getBaseRecordType());
    }
    return type;
}