项目启动时操作数据库

对于springboot项目而言，框架提供了多种接口，在项目启动时执行自定义操作。本篇记录项目启动时操作数据库的场景,利用了spring框架帮我们封装好的JdbcDaoSupport接口，操作起来还是很简单的。

application.properties

spring.datasource.driver-class-name = com.mysql.jdbc.Driver
spring.datasource.url= jdbc:mysql://120.79.xx.yy:3306/security?useUnicode=yes&characterEncoding=UTF-8&useSSL=false
spring.datasource.username = root
spring.datasource.password = 132123

import org.springframework.jdbc.core.support.JdbcDaoSupport;

public class DefaultDaoSupport extends JdbcDaoSupport{

    public static final String CREATE_TABLE_SQL = "create table if not exists tb_student (id bigint(20) NOT NULL AUTO_INCREMENT," +
            "username varchar(64) not null, " +
            "age bigint(20) , PRIMARY KEY (id))";

    @Override
    protected void initDao() throws Exception {
        super.getJdbcTemplate().execute(CREATE_TABLE_SQL);
    }
}

@SpringBootConfiguration
public class InitConfig {
    @Autowired
    private DataSource dataSource;

    @Bean
    public DaoSupport daoSupport() {
        DefaultDaoSupport defaultDaoSupport = new DefaultDaoSupport();
        defaultDaoSupport.setDataSource(dataSource);
        return defaultDaoSupport;
    }
}

好，代码就是如上所示。项目启动时，就会执行 CREATE_TABLE_SQL 这条sql语句。原理也很简单，JdbcDaoSupport 类实现了spring的InitializingBean接口，而initDao()方法在afterPropertiesSet() 方法执行时调用。

下面是相关源码：

public abstract class DaoSupport implements InitializingBean {
    protected final Log logger = LogFactory.getLog(this.getClass());

    public DaoSupport() {
    }

    public final void afterPropertiesSet() throws IllegalArgumentException, BeanInitializationException {
        this.checkDaoConfig();

        try {
            this.initDao();
        } catch (Exception var2) {
            throw new BeanInitializationException("Initialization of DAO failed", var2);
        }
    }

    protected abstract void checkDaoConfig() throws IllegalArgumentException;

    protected void initDao() throws Exception {
    }
}

另外，配置如下所示：

@SpringBootConfiguration
public class InitConfig {
    @Autowired
    private DataSource dataSource;
    @Autowired
    private JdbcTemplate jdbcTemplate;
    @Bean
    public DaoSupport daoSupport() {
        DefaultDaoSupport defaultDaoSupport = new DefaultDaoSupport();
//        defaultDaoSupport.setDataSource(dataSource);
        defaultDaoSupport.setJdbcTemplate(jdbcTemplate);
        return defaultDaoSupport;
    }
}

项目启动时操作数据库

猜你喜欢