Java项目初始化ES、MYSQL表结构及表数据

 一、初始化MYSQL数据

public boolean initMysql() throws Exception {
    log.info("initMysql.start");
    //获取所连接的数据库名称
  
    String database = systemMapper.getDatabase();
    if (StringUtils.isBlank(database)) {
        throw new BusinessException("连接数据库失败,数据库不存在");
    }
    //当库中没有表、则执行sql脚本
    if (systemMapper.countTable(database) == 0) {
        SqlSession sqlSession = sqlSessionFactory.openSession();
        Connection conn = sqlSession.getConnection();
        String mysqlInitPath="config/mysql/init.sql";
        ClassPathResource rc = new ClassPathResource(mysqlInitPath);
        EncodedResource er = new EncodedResource(rc, "utf-8");
        ScriptUtils.executeSqlScript(conn, er);
        log.info("initMysql.db:" + database + ".end");
    }
    log.info("initMysql.end");
    return true;
}

其中SystemMapper为

@Mapper
public interface SystemMapper {
    //获得当前数据库表的数量
    @Select("select count(*) from information_schema.TABLES where TABLE_SCHEMA= #{schema}")
    int countTable(@Param("schema")String schema);
    //===获得当前连接的数据库名称
    @Select("select database()")
    String getDatabase();
}

二、初始化ES数据

public boolean initEs() throws Exception {
    log.info("initEs.start");
   
     // 读取配置
    String artInfoMappingPath =config/es/art_info_index_mapping.json

    String artInfoMapping = ResourceUtil.readStr(artInfoMappingPath,        StandardCharsets.UTF_8);
 
     // 创建博文索引
     String esIndex="wechat"
     createIndexIfNotExist(esIndex, artInfoMapping);

     // 创建别名
     String alias="art_info";
     addAlias(alias, esIndex); 
     return true;
}

 如果索引不存在就创建es索引

    private void createIndexIfNotExist(String index, String mapping) {
        // 判断索引存不存在
        if (StrUtil.isBlank(index) || esAggregateService.indexExist(index)) {
            return;
        }
        log.info("initEs --> index: {}", index);
        // 创建索引
        CreateIndexRequest request = new CreateIndexRequest(index);
        Settings.Builder settings = Settings.builder()
                .put("max_result_window", 100000);
        request.settings(settings);
        request.mapping(mapping, XContentType.JSON);
        createIndex(request);
        }

 

public void createIndex(CreateIndexRequest request) {
    try {
        CreateIndexResponse response = elasticSearchClient.indices().create(request, RequestOptions.DEFAULT);
        log.info("create index: {}, isAcknowledged: {}", response.index(), response.isAcknowledged());
    } catch (IOException e) {
        log.error("ES 索引创建失败 --> ", e);
        throw new BusinessException(ResultCode.INTERNAL_SERVER_ERROR);
    }
}

 根据需要、添加es别名

@Override
public void addAlias(String alias, String... index) {
    // 构建请求参数
    IndicesAliasesRequest indicesAliasesRequest = new IndicesAliasesRequest();
    IndicesAliasesRequest.AliasActions aliasActions = new IndicesAliasesRequest.AliasActions(IndicesAliasesRequest.AliasActions.Type.ADD)
            .indices(index)
            .alias(alias);
    indicesAliasesRequest.addAliasAction(aliasActions);
    try {
        AcknowledgedResponse response = elasticSearchClient.indices().updateAliases(indicesAliasesRequest, RequestOptions.DEFAULT);
        log.info("add alias --> index: {}, alias: {}, isAcknowledged: {}", index, alias, response.isAcknowledged());
    } catch (IOException e) {
        log.error("ES 别名创建失败 --> ", e);
        throw new BusinessException(ResultCode.INTERNAL_SERVER_ERROR);
    }
}