Tbed/src/main/java/cn/hellohao/utils/RunSqlScript.java

package cn.hellohao.utils;

import cn.hellohao.service.impl.S3Imageupload;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.sql.*;


final class RunSqlScript {
    private static Logger logger = LoggerFactory.getLogger(RunSqlScript.class);
    public static  String DBDRIVER = "com.mysql.cj.jdbc.Driver";
    public static  String DBURL ;
    //现在使用的是mysql数据库，是直接连接的，所以此处必须有用户名和密码
    public static  String USERNAME ;
    public static  String PASSWORD ;

    public static Integer RunSelectCount(String sql) {
        Integer count = 0;
        //数据库连接对象
        Connection conn = null;
        //数据库操作对象
        Statement stmt = null;
        //1、加载驱动程序
        try {
            Class.forName(DBDRIVER);
        } catch (ClassNotFoundException e) {
            logger.error("RunSqlScript[29]", e);
        }
        //2、连接数据库
        //通过连接管理器连接数据库
        try {
            //在连接的时候直接输入用户名和密码才可以连接
            conn = DriverManager.getConnection(DBURL,USERNAME,PASSWORD);
        } catch (SQLException e) {
            logger.error("RunSqlScript[36]", e);
        }
        //3、向数据库中插入一条数据
        //String sql = "select count(*) from information_schema.columns where table_name = 'config' and column_name = 'id'";
        try {
            stmt = conn.createStatement();
        } catch (SQLException e) {
            logger.error("RunSqlScript[43]", e);
        }
        //4、执行语句
        try {
            //stmt.executeUpdate(sql);
            ResultSet resultSet = stmt.executeQuery(sql);
            if(resultSet.next()){
                count = resultSet.getInt(1);
            }
        } catch (SQLException e) {
            logger.error("RunSqlScript[53]", e);
            return -1;
        }
        //5、关闭操作
        try {
            stmt.close();
            conn.close();
        } catch (SQLException e) {
            logger.error("RunSqlScript[61]", e);
        }
        return count;
    }


    //insert操作
    public static Integer RunInsert(String sql) {
        Integer count = 0;
        //数据库连接对象
        Connection conn = null;
        //数据库操作对象
        Statement stmt = null;
        //1、加载驱动程序
        try {
            Class.forName(DBDRIVER);
        } catch (ClassNotFoundException e) {
            logger.error("RunSqlScript[78]", e);
        }
        //2、连接数据库
        //通过连接管理器连接数据库
        try {
            //在连接的时候直接输入用户名和密码才可以连接
            conn = DriverManager.getConnection(DBURL,USERNAME,PASSWORD);
        } catch (SQLException e) {
            logger.error("RunSqlScript[86]", e);
        }
        try {
            stmt = conn.createStatement();
        } catch (SQLException e) {
            logger.error("RunSqlScript[91]", e);
        }
        //4、执行语句
        try {
            count = stmt.executeUpdate(sql);
        } catch (SQLException e) {
            if(e.getMessage().contains("Duplicate key name") && e.getMessage().contains("index_md5key_url")){
                logger.info("Database index already exists, skip this step.");
            }else{
                logger.error("RunSqlScript[100]", e);
            }
        }
        //5、关闭操作
        try {
            stmt.close();
            conn.close();
        } catch (SQLException e) {
            logger.error("RunSqlScript[108]", e);
        }
        return count;
    }


}