SpringBoot数据库初始化（自动判断是否需要初始化）

使用方法：
源代码config\目录下，创建DataSourceInit.java单元，把下面的代码放到这个单元文件即可。在resource/db目录下，放置schema.sql，里面就是所有表的create table语句。同样，可以放置一个data.sql文件，里面是所有的insert into 插入初始化数据的SQL语句。你可以dump数据库的structure或者data出来。当然你也可以合并成一个文件，根据你的需要修改一下代码即可。

package com.company.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.Resource;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.init.DataSourceInitializer;
import org.springframework.jdbc.datasource.init.DatabasePopulator;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;

import javax.sql.DataSource;

/**
 * Automatically init the database and data if needed
 *
 * @since 1.0
 */
@Configuration
@Slf4j
public class DataSourceInit {
    @Value("classpath:db/schema.sql")
    private Resource ddl;
    @Value("classpath:db/data.sql")
    private Resource dml;

    @Bean
    public DataSourceInitializer dataSourceInitializer(final DataSource dataSource) {
        final DataSourceInitializer initializer = new DataSourceInitializer();
        // 设置数据源
        initializer.setDataSource(dataSource);
        initializer.setDatabasePopulator(databasePopulator(dataSource));
        return initializer;
    }

    private DatabasePopulator databasePopulator(DataSource dataSource) {
        final ResourceDatabasePopulator populator = new ResourceDatabasePopulator();

        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
        try {
            jdbcTemplate.queryForMap("select count(1) from 这里改成你的表名");
        } catch (DataAccessException e) {
            log.warn("初始化脚本报错:" + e.getMessage());
            // 报错,表不存在,初次导入，执行相关脚本
            populator.addScripts(ddl);
            populator.addScripts(dml);
        } catch (Exception e) {
            log.warn("初始化脚本报错:" + e.getMessage());
        }
        return populator;
    }

}